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

栏目: 数据库 · PostgreSQL · 发布时间: 5年前

内容简介: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 包中 GetenvSetenv 方法实现。方法名称已经很明确说明了自身的功能。

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

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

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


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

C++标准库(第2版)

C++标准库(第2版)

Nicolai M. Josuttis / 侯捷 / 电子工业出版社 / 2015-6 / 186.00元

《C++标准库(第2版)》是全球C++经典权威参考书籍时隔12年,基于C++11标准的全新重大升级。标准库提供了一组公共类和接口,极大地拓展了C++语言核心功能。《C++标准库(第2版)》详细讲解了每一标准库组件,包括其设计目的和方法、复杂概念的剖析、实用而高效的编程细节、存在的陷阱、重要的类和函数,又辅以大量用C++11标准实现的实用代码范例。除覆盖全新组件、特性外,《C++标准库(第2版)》一......一起来看看 《C++标准库(第2版)》 这本书的介绍吧!

HTML 编码/解码
HTML 编码/解码

HTML 编码/解码

Base64 编码/解码
Base64 编码/解码

Base64 编码/解码