首页 > 数据库 > Go 每日一库之 godotenv
2020
02-13

Go 每日一库之 godotenv

Go 每日一库之 godotenv
简介
twelve-factor应用提倡将配置存储在环境变量中。任何从开发环境切换到生产环境时需要修改的东西都从代码抽取到环境变量里。 但是在实际开发中,如果同一台机器运行多个项目,设置环境变量容易冲突,不实用。godotenv库从.env文件中读取配置, 然后存储到程序的环境变量中。在代码中可以使用读取非常方便。godotenv源于一个 Ruby 的开源项目dotenv。

快速使用
第三方库需要先安装:

$ go get github.com/joho/godotenv

复制代码
后使用:
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文件:

name = dj
age = 18
复制代码
运行程序,输出:

name:  dj
age:  18
复制代码
可见,使用非常方便。默认情况下,godotenv读取项目根目录下的.env文件,文件中使用key = value的格式,每行一个键值对。 调用godotenv.Load()即可加载,可直接调用os.Getenv("key")读取。os.Getenv是用来读取环境变量的:

package main

import (
  "fmt"
  "os"
)

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

复制代码
高级特性
自动加载
如果你有程序员的优良传统——懒,你可能连Load方法都不想自己调用。没关系,godotenv给你懒的权力!

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

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方法:

// 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()
}

复制代码
仔细看注释,程序员的恶趣味

扫码芷若 获取免费视频学习资料

编程学习

查 看2019高级编程视频教程免费获取