区分环境

apps/gateway/gateway.go

@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"os"
 
 	"tianyuan-api/apps/gateway/internal/config"
 	"tianyuan-api/apps/gateway/internal/handler"
@@ -12,9 +13,23 @@ import (
 	"github.com/zeromicro/go-zero/rest"
 )
 
-var configFile = flag.String("f", "etc/gateway-api.yaml", "the config file")
-
 func main() {
+	// 读取环境变量 ENV，默认为 "prod"
+	env := os.Getenv("ENV")
+	if env == "" {
+		env = "production"
+	}
+
+	// 根据 ENV 加载不同的配置文件
+	var defaultConfigFile string
+	if env == "development" {
+		defaultConfigFile = "etc/gateway-api.dev.yaml" // 开发环境配置
+	} else {
+		defaultConfigFile = "etc/gateway-api.yaml" // 生产环境配置
+	}
+
+	// 允许通过命令行参数覆盖配置文件路径
+	configFile := flag.String("f", defaultConfigFile, "the config file")
 	flag.Parse()
 
 	var c config.Config