可以查询配置文件

bootstraps/bootstrap.go

@@ -185,6 +185,8 @@ func (b *ServiceBootstrapper) runWebService() {
 	logger.Debug("测试日志是否写入ES.")
 
 	log.Printf("服务开始运行...")
+	defer b.CleanupDatabase()
+
 	if err := b.WebService.Run(); err != nil {
 		log.Fatal("服务运行失败:", err)
 	}

config/config.go

@@ -23,6 +23,14 @@ type IConfig interface {
 	IsRedisConfigured() bool
 	SetServiceName(name string)
 	SetServiceVersion(version string)
+	SetEnv(name string)
+}
+
+// 根据启动参数设置环境变量
+func (c *Config) SetEnv(name string) {
+	if c.GetService().Env == "" {
+		c.GetService().Env = name
+	}
 }
 
 // 设置微服务名称。。如果配置文件设置了，就按配置文件

config/loader.go

@@ -10,33 +10,39 @@ import (
 	"gopkg.in/yaml.v2"
 )
 
-// LoadConfig 加载配置到注册表
+// LoadConfig 从文件加载配置到注册表（保持原有接口不变）
 func LoadConfig() error {
-	// 1. 设置所有注册配置的默认值
-	for _, config := range subconfigs.GetAllConfigs() {
-		config.SetDefaults()
-	}
-
-	// 2. 查找配置文件
+	// 1. 查找配置文件
 	configFile, err := findConfigFile()
 	if err != nil {
 		return err
 	}
 
-	// 3. 读取文件
+	// 2. 读取并解析文件
 	data, err := os.ReadFile(configFile)
 	if err != nil {
 		return fmt.Errorf("read config file error: %v", err)
 	}
 
-	// 4. 解析为map
 	var rawConfig map[string]interface{}
 	err = yaml.Unmarshal(data, &rawConfig)
 	if err != nil {
 		return fmt.Errorf("parse yaml error: %v", err)
 	}
 
-	// 5. 循环注册表，为每个配置加载数据
+	// 3. 从map加载配置
+	return LoadConfigFromMap(rawConfig)
+}
+
+// LoadConfigFromMap 从map[string]interface{}加载配置到注册表
+// 新增方法，供外部调用（比如从数据库加载后使用）
+func LoadConfigFromMap(rawConfig map[string]interface{}) error {
+	// 1. 设置所有注册配置的默认值
+	for _, config := range subconfigs.GetAllConfigs() {
+		config.SetDefaults()
+	}
+
+	// 2. 循环注册表，为每个配置加载数据
 	for name, config := range subconfigs.GetAllConfigs() {
 		if configData, ok := rawConfig[name].(map[interface{}]interface{}); ok {
 			// 转换为 map[string]interface{}
@@ -47,17 +53,10 @@ func LoadConfig() error {
 		}
 	}
 
-	// // 6. 验证所有配置
-	// for name, config := range subconfigs.GetAllConfigs() {
-	// 	if err := config.Validate(); err != nil {
-	// 		return fmt.Errorf("validate config %s error: %v", name, err)
-	// 	}
-	// }
-
 	return nil
 }
 
-// convertMap 转换map类型
+// convertMap 转换map类型（内部函数保持不变）
 func convertMap(input map[interface{}]interface{}) map[string]interface{} {
 	output := make(map[string]interface{})
 	for k, v := range input {
@@ -68,7 +67,7 @@ func convertMap(input map[interface{}]interface{}) map[string]interface{} {
 	return output
 }
 
-// findConfigFile 查找配置文件
+// findConfigFile 查找配置文件（内部函数保持不变）
 func findConfigFile() (string, error) {
 	exePath, _ := os.Executable()
 	exeDir := filepath.Dir(exePath)
@@ -91,8 +90,6 @@ func findConfigFile() (string, error) {
 		}
 	}
 
-	//return "", fmt.Errorf("no configuration file found")
-
 	return "", fmt.Errorf(`no configuration file found
 
 Searched locations:

config/subconfigs/service_config.go

@@ -14,6 +14,7 @@ type ServiceConfig struct {
 	IdleTimeout    int    `yaml:"idle_timeout"`
 	TrustedProxies string `yaml:"trusted_proxies"`
 	InstanceName   string `yaml:"instance-name"`
+	Env            string `yaml:"env"`
 }
 
 func NewServiceConfig() *ServiceConfig {

middleware/authMiddleware.go

@@ -84,7 +84,7 @@ func sendAuthError(w http.ResponseWriter, message string, format ResponseFormat)
 		w.Write([]byte("unauthorized," + message + "\n"))
 	default:
 		w.Header().Set("Content-Type", "application/json")
-		json.NewEncoder(w).Encode(&types.QueryResult{
+		json.NewEncoder(w).Encode(&types.QueryResult[map[string]interface{}]{
 			Success: false,
 			Error:   message,
 			Time:    time.Now().Format(time.RFC3339),

myservice/service.go

@@ -7,6 +7,7 @@ import (
 
 	"git.x2erp.com/qdy/go-base/config"
 	"github.com/go-micro/plugins/v4/registry/consul"
+	"github.com/urfave/cli/v2"
 	"go-micro.dev/v4/registry"
 	"go-micro.dev/v4/web"
 )
@@ -30,6 +31,14 @@ func StartWithRegistry(cfg config.IConfig) web.Service {
 		web.Registry(consulRegistry),
 		web.RegisterTTL(30*time.Second),      // Health check interval
 		web.RegisterInterval(20*time.Second), // Registration interval
+		web.Flags(&cli.StringFlag{
+			Name:  "env",
+			Value: "dev",
+			Usage: "环境参数"}),
+		web.Action(func(c *cli.Context) {
+			cfg.SetEnv(c.String("env"))
+			log.Printf("环境参数: %s\n", serviceConfig.Env)
+		}),
 	)
 
 	// 3. Initialize service

types/types.go

@@ -105,9 +105,9 @@ type QueryRequest struct {
 }
 
 // QueryResult 查询结果
-type QueryResult struct {
+type QueryResult[T any] struct {
 	Success    bool                `json:"success"`
-	Data       interface{}         `json:"data,omitempty"`
+	Data       T                   `json:"data,omitempty"`
 	Error      string              `json:"error,omitempty"`
 	Count      int                 `json:"count,omitempty"`
 	Time       string              `json:"time,omitempty"`
@@ -118,14 +118,6 @@ type QueryResult struct {
 	Metadata   *ctx.RequestContext `json:"metadata,omitempty"`
 }
 
-// PageResult 分页结果
-type PageResult struct {
-	QueryResult
-	Page     int `json:"page"`
-	PageSize int `json:"pageSize"`
-	Total    int `json:"total"`
-}
-
 // HealthCheck 健康检查
 type HealthCheck struct {
 	Status    string    `json:"status"`