【总结】几种常见的golang动态设置方式

时间：2023-04-11 12:06

在 Go 语言中，我们经常需要动态地设置一些配置，例如读取配置文件、从环境变量中读取配置等。本文将介绍几种常见的动态设置方式。

读取配置文件

读取配置文件是一种常见的动态设置方式，在 Go 语言中，我们可以使用 [viper](https://github.com/spf13/viper) 库来方便地读取配置文件。以下是一个简单的示例：

package mainimport (    "fmt"    "github.com/spf13/viper")func main() {    viper.SetConfigName("config") // 配置文件名    viper.SetConfigType("yaml")   // 配置文件类型    viper.AddConfigPath(".")      // 配置文件路径    err := viper.ReadInConfig()   // 读取配置文件    if err != nil {        fmt.Printf("Read config file error: %v
", err)        return    }    fmt.Println(viper.GetString("name")) // 从配置文件中读取配置项}

在上述代码中，我们首先设置了配置文件的名字、类型和路径，然后调用 viper.ReadInConfig() 方法来读取配置文件，最后通过 viper.GetString() 方法来获取配置项的值。

从环境变量中读取配置

在实际开发中，有时我们需要从环境变量中读取一些配置信息。在 Go 语言中，我们可以使用 os 包的 Getenv() 方法来获取环境变量的值。以下是一个示例：

package mainimport (    "fmt"    "os")func main() {    name := os.Getenv("APP_NAME")    if name == "" {        name = "unknown"    }    fmt.Printf("Application name: %s
", name)}

在上述代码中，我们首先通过 os.Getenv() 方法获取环境变量 APP_NAME 的值，如果该环境变量不存在，则使用默认值 unknown。

通过命令行参数设置配置项

有时我们需要通过命令行参数来指定一些配置信息。在 Go 语言中，我们可以使用 flag 包来解析命令行参数。以下是一个示例：

package mainimport (    "flag"    "fmt")func main() {    var name string    flag.StringVar(&name, "name", "unknown", "Application name")    flag.Parse()    fmt.Printf("Application name: %s
", name)}

在上述代码中，我们首先定义了一个 name 变量，然后调用 flag.StringVar() 方法来解析命令行参数 -name 并设置默认值为 unknown，最后通过 flag.Parse() 方法来解析命令行参数并获取其值。