为什么加载环境变量主函数golang

Question

在制作 Golang 应用程序时

关于Mysql的读取方式，最好用main函数读取环境变量，传给参数。

原因导致环境变量是一种全局变量情况，在测试过程中从模块外部查看时，你不知道为什么DB包连接到一个名为XXX的主机。我收到了一个解释，有人怀疑可能会写入一个固定值

当前代码

mysql.go

func Init() *gorm.DB {
    path := strings.Join([]string{os.Getenv("MYSQL_USER"), ":", os.Getenv("MYSQL_PASSWORD"), "@tcp(", os.Getenv("DB_HOST"), ":", os.Getenv("DB_PORT"), ")/", os.Getenv("MYSQL_DATABASE"), "?charset=utf8&parseTime=True&loc=Local"}, "")

    db, err := gorm.Open("mysql", path)
    if err != nil {
        panic(err)
    }

    err = db.DB().Ping()
    if err != nil {
        panic(err)
    }

    DB = db

    return DB
}

main.go

func main {

    db := mysql.Init()
    defer db.Close()

}

如果你能告诉我你是否知道最好的方法，我会很高兴。

谢谢

Answer 1

理想情况下，您应该有一个单独的结构来保存所有配置值。然后，您可以拥有从各种来源（环境变量、文件等）填充配置结构的函数

此外，您可以使用fmt.Sprintf 来改进路径构建。

这是一个粗略的例子。

func Init() *gorm.DB {
    cfg := LoadConfigFromEnvironment()
    path := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local",
        cfg.DBUser,
        cfg.DBPassword,
        cfg.DBHost,
        cfg.DBPort,
        cfg.DBName,
    )

    db, err := gorm.Open("mysql", path)
    if err != nil {
        panic(err)
    }

    err = db.DB().Ping()
    if err != nil {
        panic(err)
    }

    return db
}

注意还有我是如何删除DB = db 分配的。不确定那是什么，但您不应该从该函数分配给全局变量。

正如其他人所提到的，理想情况下，您应该将配置加载与初始化分开，因此Init 的定义最好如下所示。

func Init(cfg Config) *gorm.DB {
   ...
}