21、Go应用配置管理与REST API开发实践

Go应用配置管理与REST API开发实践

1. 使用Viper进行配置管理

1.1 Viper简介

Viper是Go应用的配置管理解决方案，允许通过配置文件、环境变量和命令行标志等多种方式为应用指定配置选项。当使用Cobra生成器创建应用的样板代码时，会自动启用Viper。

1.2 初始化Viper配置

Cobra在初始化应用时通过运行 initConfig 函数来启用Viper，该函数定义在 cmd/root.go 文件中：

func initConfig() {
    if cfgFile != "" {
        // Use config file from the flag.
        viper.SetConfigFile(cfgFile)
    } else {
        // Find home directory.
        home, err := homedir.Dir()
        if err != nil {
            fmt.Println(err)
            os.Exit(1)
        }

        // Search config in home directory with name ".pScan" (without extension).
        viper.AddConfigPath(home)
        viper.SetConfigName(".pScan")
    }