读取启动参数#

结合 pflag 读取启动参数，验证不同配置来源的优先级与端口校验。

本系列使用 Viper v1.21.0。每章都是独立实验，示例命令在项目目录中执行；开始新一章时，使用该章的完整程序和配置，不累积上一章的临时修改。

尚未准备练习项目时，先完成第 1 章的项目准备。

初始示例#

初始配置 config/config.yaml：

1
app:
2
  name: viper-demo
3
  port: 8080

初始完整 main.go：

1
package main
2

3
import (
4
  "fmt"
5
  "github.com/spf13/pflag"
6
  "github.com/spf13/viper"
7
  "log"
8
)
9

10
func main() {
11
  pflag.Int("port", 4000, "应用端口")
12
  pflag.Parse()
13

14
  v := viper.New()
15
  v.SetConfigFile("config/config.yaml")
16
  v.SetDefault("app.port", 3000)
17
  if err := v.BindEnv("app.port", "APP_PORT"); err != nil {
18
    log.Fatal(err)
19
  }
20
  if err := v.BindPFlag("app.port", pflag.Lookup("port")); err != nil {
21
    log.Fatal(err)
22
  }
23
  if err := v.ReadInConfig(); err != nil {
24
    log.Fatal(err)
25
  }
26

27
  fmt.Println("监听端口:", v.GetInt("app.port"))
28
}

pflag.Int 注册参数名、默认值和帮助说明；Parse 解析启动参数；BindPFlag 把命令行 port 对应到 Viper 的 app.port。

分别执行：

1
env -u APP_PORT go run main.go
2
env APP_PORT=9090 go run main.go
3
env APP_PORT=9090 go run main.go --port=7070
4
go run main.go --help

前三条预期端口依次为 8080、9090、7070，最后一条显示帮助并退出。未显式传入 --port 时，flag 默认值 4000 不会压过文件中的 8080。

第一次修改#

在获取端口之前增加：

1
v.Set("app.port", 6060)

重新运行带 --port=7070 的命令，预期为 6060。本次涉及的覆盖关系为 Set > 显式命令行参数 > 环境变量 > 文件 > SetDefault，由来源决定，不是最后调用哪个方法就用哪个来源。

第二次修改#

删除上一处 Set，在 main 外增加：

1
type Config struct {
2
    App struct {
3
        Name string `mapstructure:"name"`
4
        Port int    `mapstructure:"port"`
5
    } `mapstructure:"app"`
6
}

将最后的端口打印语句替换为以下完整块：

1
var cfg Config
2
if err := v.UnmarshalExact(&cfg); err != nil {
3
    log.Fatal(err)
4
}
5
if cfg.App.Port < 1 || cfg.App.Port > 65535 {
6
    log.Fatal("app.port 必须在 1 到 65535 之间")
7
}
8
fmt.Println("监听端口:", cfg.App.Port)

运行 go run main.go --port=7070 应成功；运行 go run main.go --port=70000 应校验失败。注册参数和绑定不变，只是最终获取方式从 Getter 换成结构体解码。

HikariTish

读取启动参数#

初始示例#

第一次修改#

第二次修改#

参考资料#

Viper 学习记录

继续阅读

换条路线

HikariTish

读取启动参数#

初始示例#

第一次修改#

第二次修改#

参考资料#

分享文章

沿着主题读

换条路线