go-zero 怎么区分 dev / test / prod 环境

① 准备多份配置（同结构、不同值）

② main.go 用 -f 接收配置路径

package main

import (
    "flag"
    "github.com/zeromicro/go-zero/core/conf"
    "github.com/zeromicro/go-zero/rest"
    "your_project/internal/config"
    "your_project/internal/handler"
    "your_project/internal/svc"
)

// -f 指定配置文件，默认值指向 etc 目录下的 yaml
var configFile = flag.String("f", "etc/user-api.yaml", "the config file")

func main() {
    flag.Parse()

    var c config.Config
    // MustLoad 读 -f 指向的文件；多环境只换文件、不换代码
    conf.MustLoad(*configFile, &c, conf.UseEnv())

    server := rest.MustNewServer(c.RestConf)
    defer server.Stop()

    ctx := svc.NewServiceContext(c)
    handler.RegisterHandlers(server, ctx)

    server.Start()
}

③ 三份 yaml 长啥样（只列差异项）

# user-api.dev.yaml
Name: user-api
Mode: dev
Host: 0.0.0.0
Port: 8888
Mysql:
  DataSource: root:123456@tcp(127.0.0.1:3306)/user_dev
CacheRedis:
  - Host: 127.0.0.1:6379
Trace:
  Sampler: 1.0        # dev 全采样，方便排查

# user-api.prod.yaml
Name: user-api
Mode: pro
Host: 0.0.0.0
Port: 8888
Mysql:
  DataSource: ${MYSQL_DSN}   # 密钥/地址走环境变量，不进仓库
CacheRedis:
  - Host: ${REDIS_HOST}
Trace:
  Sampler: 0.1        # prod 低采样，控成本

关键：conf.UseEnv()

go-zero 的 conf 包支持在 yaml 里写 ${VAR}，启动时替换为进程环境变量——但前提是 MustLoad 带上 conf.UseEnv() 选项（上面 main.go 已带上）。

# 一份 prod 模板 etc/user-api.prod.yaml
Mysql:
  DataSource: ${MYSQL_DSN}
CacheRedis:
  - Host: ${REDIS_HOST}
Auth:
  AccessSecret: ${JWT_SECRET}   # 私钥绝不进仓库

# 启动时注入环境变量即可（shell / Docker / K8s Secret）
MYSQL_DSN='prod_user:xxxx@tcp(db.prod:3306)/user' \
REDIS_HOST=redis.prod:6379 \
JWT_SECRET='***' \
./user-api -f etc/user-api.prod.yaml

它管什么

rest.RestConf（内嵌 service.ServiceConf）有个 Mode 字符串字段。go-zero 用它决定日志格式 / 运行时行为：

配合 Log 字段更精细

Log:
  Mode: file          # console / file / volatile
  Encoding: json     # flat(dev 友好) / json(pro 友好，便于采集)
  Path: /var/log/user-api
  Level: info

本地 / 裸机

# 开发
go run user.go -f etc/user-api.dev.yaml
# 测试
./user-api -f etc/user-api.test.yaml
# 生产
./user-api -f etc/user-api.prod.yaml

Docker

# Dockerfile
ENTRYPOINT ["/app/user-api", "-f", "/app/etc/user-api.yaml"]
# 不同环境用不同的配置文件挂载（ConfigMap / 构建参数），或传环境变量覆盖 ${ENV}

Kubernetes

# deployment.yaml —— 用 env 注入密钥，配置文件由 ConfigMap 挂到 /app/etc
env:
  - name: MYSQL_DSN
    valueFrom: { secretKeyRef: { name: user-secret, key: dsn } }
  - name: JWT_SECRET
    valueFrom: { secretKeyRef: { name: user-secret, key: jwt } }
# 容器启动参数 -f /app/etc/user-api.prod.yaml（prod 模板已含 ${MYSQL_DSN} 等）

从 config 拿 Mode

// config.go —— 把 Mode 透传到业务可访问的地方
type Config struct {
    rest.RestConf          // 内嵌了 Mode 字段
    Auth struct {
        AccessSecret string
        AccessExpire int64
    }
}

// svc 中需要时用（尽量克制）
if c.Mode == "pro" {
    // 生产才开的开关，如关闭调试接口
}

不要自己 new 一个全局 env 变量

呼应「go-zero 全局变量」那篇：环境信息只在启动时读一次、通过 config → svc 注入，别在包里写 var Env = "prod" 全局变量。需要环境相关行为时，在 svc 装配时决定并传依赖进去（比如 dev 注入 mock 客户端、pro 注入真实客户端）。