go基础库之环境变量的获取与设置以及如何使用默认值

golang 基础库之 go 获取命令行参数，介绍了环境变量的获取与设置以及如何使用默认值

环境变量的获取与设置以及如何使用默认值

Golang 版本

1.12.1

前言

环境变量作为配置选项可以显著简化应用的部署，并且这些在云基础架构中也是很常见的。

通常，本地和自动构建环境的数据库连接配置是不同的。

如果配置由环境变量定义，则无需更改应用的配置文件或者程序代码。如果环境变量没有设置，则默认配置也是可以使用的。这样对于开发人员就容易多了。

实现

创建 get.go 文件，代码如下：

package mainimport ("log""os"
)func main(){conStr := os.Getenv("DB_CONN")log.Printf("连接：%s\n",conStr)
}

运行命令 export DB_CONN=db:/user@example && go run get.go

$ export  DB_CONN=db:/user@example && go run get.go
2019/05/18 21:33:39 连接：db:/user@example

创建 lookup.go 文件，代码如下：

package mainimport ("fmt""log""os"
)func main() {key := "DB_CONN"connStr, ex := os.LookupEnv(key)if !ex {log.Printf("环境变量 %s 没有设置\n", key)}fmt.Println(connStr)
}

运行命令 unset DB_CONN && go run lookup.go

$ unset DB_CONN && go run lookup.go
2019/05/18 21:39:33 环境变量 DB_CONN 没有设置

创建 main.go 文件，代码如下：

package mainimport ("log""os"
)func main() {key := "DB_CONN"// 设置环境变量os.Setenv(key, "postgres://as:as@example.com/pg?sslmode=verify-full")val := GetEnvDefault(key, "postgres://as:as@localhost/pg?sslmode=verify-full")log.Println("值是 :" + val)os.Unsetenv(key)val = GetEnvDefault(key, "postgres://as:as@127.0.0.1/pg?sslmode=verify-full")log.Println("默认值是 :" + val)}func GetEnvDefault(key, defVal string) string {val, ex := os.LookupEnv(key)if !ex {return defVal}return val
}

运行命令 go run main.go

$ go run main.go
2019/05/18 21:42:58 值是 :postgres://as:as@example.com/pg?sslmode=verify-full
2019/05/18 21:42:58 默认值是 :postgres://as:as@127.0.0.1/pg?sslmode=verify-full

原理

环境变量的获取和设置分别可以通过 os 包中 Getenv 和 Setenv 方法实现。方法名称已经很明确说明了自身的功能。

但 Getenv 方法有个缺点，即使在未设置环境变量的情况下，它也返回一个空的字符串。

os 包中还有一个很有用的方法， LookupEnv ，该方法返回两个值，一个是变量的值，另一个则是变量在环境中是否设置的布尔值。而 LookupEnv 方法则可以克服上面 Getenv 的缺点。

当我们要判断是否设置了环境变量以及实现一个默认环境变量的方法，都应该使用 LookupEnv 。因为如果未设置环境变量，则第二个值会返回 false