Go每日一库之11:godotenv

2023-09-16 08:55:44 浏览数 (1)

简介

twelve-factor应用提倡将配置存储在环境变量中。任何从开发环境切换到生产环境时需要修改的东西都从代码抽取到环境变量里。

但是在实际开发中,如果同一台机器运行多个项目,设置环境变量容易冲突,不实用。godotenv库从.env文件中读取配置,

然后存储到程序的环境变量中。在代码中可以使用读取非常方便。godotenv源于一个 Ruby 的开源项目dotenv。

快速使用

第三方库需要先安装:

代码语言:javascript复制
$ go get github.com/joho/godotenv

后使用:

代码语言:javascript复制
package main

import (
  "fmt"
  "log"
  "os"

  "github.com/joho/godotenv"
)

func main() {
  err := godotenv.Load()
  if err != nil {
    log.Fatal(err)
  }

  fmt.Println("name: ", os.Getenv("name"))
  fmt.Println("age: ", os.Getenv("age"))
}

然后在可执行程序相同目录下,添加一个.env文件:

代码语言:javascript复制
name = dj
age = 18

运行程序,输出:

代码语言:javascript复制
name:  dj
age:  18

可见,使用非常方便。默认情况下,godotenv读取项目根目录下的.env文件,文件中使用key = value的格式,每行一个键值对。

调用godotenv.Load()即可加载,可直接调用os.Getenv("key")读取。os.Getenv是用来读取环境变量的:

代码语言:javascript复制
package main

import (
  "fmt"
  "os"
)

func main() {
  fmt.Println(os.Getenv("GOPATH"))
}

高级特性

自动加载

如果你有程序员的优良传统——懒,你可能连Load方法都不想自己调用。没关系,godotenv给你懒的权力!

导入github.com/joho/godotenv/autoload,配置会自动读取:

代码语言:javascript复制
package main

import (
  "fmt"
  "os"

  _ "github.com/joho/godotenv/autoload"
)

func main() {
  fmt.Println("name: ", os.Getenv("name"))
  fmt.Println("age: ", os.Getenv("age"))
}

注意,由于代码中没有显式用到godotenv库,需要使用空导入,即导入时包名前添加一个_

autoload包的源码,其实就是库帮你调用了Load方法:

代码语言:javascript复制
// src/github.com/joho/godotenv/autoload/autoload.go
package autoload

/*
	You can just read the .env file on import just by doing

		import _ "github.com/joho/godotenv/autoload"

	And bob's your mother's brother
*/

import "github.com/joho/godotenv"

func init() {
  godotenv.Load()
}

仔细看注释,程序员的恶趣味

0 人点赞