go基础库之环境变量的获取与设置以及如何使用默认值
栏目: 数据库 · PostgreSQL · 发布时间: 6年前
内容简介：golang 基础库之go获取命令行参数，介绍了环境变量的获取与设置以及如何使用默认值
golang 基础库之 go 获取命令行参数，介绍了环境变量的获取与设置以及如何使用默认值
环境变量的获取与设置以及如何使用默认值
Golang 版本
1.12.1
前言
环境变量作为配置选项可以显著简化应用的部署，并且这些在云基础架构中也是很常见的。
通常，本地和自动构建环境的数据库连接配置是不同的。
如果配置由环境变量定义，则无需更改应用的配置文件或者程序代码。如果环境变量没有设置，则默认配置也是可以使用的。这样对于开发人员就容易多了。
实现
创建 get.go
文件，代码如下：
package main
import (
"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 main
import (
"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 main
import (
"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
。
以上就是本文的全部内容，希望本文的内容对大家的学习或者工作能带来一定的帮助，也希望大家多多支持 码农网
猜你喜欢:
- Vue项目打包后动态获取自定义变量
- 如何在iOS上获取变量中的位置(纬度和经度值)？
- 全局变量，静态全局变量，局部变量，静态局部变量
- python变量与变量作用域
- Python基础-类变量和实例变量
- python编程（类变量和实例变量）
本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。
