一篇文章告诉你 golang 环境变量的所有基础操作

大家好，我是老麦，我将每天 早上9点 为你分享一篇好文章。

一篇文章告诉你 golang 环境变量的所有基础操作

原文链接：https://tangx.in/posts/2021/09/06/golang-os-env-operation/

golang 中的环境变量操作都在 os 包下面，只有很少的几个方法，而且字面意思也很明确。

所有环境变量操作对象都是 字符串string ，因此对于 int， bool 类型需要自己实现转换。
golang 程序执行的时候，是在 linux 系统中 fork 的一种子进程中
1. golang程序在 复制了开始运行的那一瞬间的当前环境所有变量，之后的父进程中的变量变化不再影响 golang 程序。
2. golang 程序对环境变量的所有操作，都是在自身的子进程中，因此 只会影响 golang 程序本身。
3. go 语言中没有类似 bash 中的 export 的操作。但是可以通过 os.Setenv 设置环境变量。

image-20210907001155092

环境变量操作方法

1. `os.Setenv("key","val")`

创建一个环境变量

2. `os.Unsetenv("key")`

取消一个变量

3. `val=os.Getenv("key")`

返回一个变量的值。如果变量不存在， val 为空字符串。

len(val)==0

这种方法无法区分变量不存在或者变量值为空字符串。

4. `val,ok=os.LookupEnv("key")`

返回一个变量的值与变量是否存在的 bool 结果。

如果变量存在， val 为值， ok 为 true
如果变量不存在， val 为空字符串， ok 为 false

注意， 变量不存在 (ok=false), 和 变量值为空 (ok=true) 不一样

5. `os.Clearenv()`

清空所有变量。危险慎用。

6. `envs=os.Environ()`

返回包含所有变量的 []string 切片副本。

由于 os.Environ() 返回的是一个 []string 切片，在某些场景下，如果要进行 传递并检索 的时候，并不是很方便，因此会有需求转换成 map[string]string 。

在这里 需要额外小心，如果想通过使用了 strings.Split 分割 **等号=**以获取变量名和值，但是变量值没有使用 strings.Join 合并时，可能会造成数据丢失。有点拗口，看例子

因为一下语句时合法的

代码语言：javascript复制

VAR=key1=val1,key2=val2

这句话赋值语句表示将 字符串 key1=val1,key2=val2 赋值给 变量 VAR

例如下面这段代码, envutils - fix: lost value when trans env string slice into map[1]

代码语言：javascript复制

 _ = os.Setenv("VAR", "key=val1,key2=val2")
 m := make(map[string]string)
 envs := os.Environ()

 for _, pair := range envs {
  kv := strings.Split(pair, "=")
    // m[kv[0]] = kv[1:] // wrong: VAR=key1 与实际情况不符合
  m[kv[0]] = strings.Join(kv[1:], "=")  // 注意这里要使用 Join
 }

7. `os.ExpandEnv("string")` or `os.Expand("string",os.Getenv)`

如果 string 中包含 key 或者 {key} 的占位符，则将被替换为实际的值。就是 bash 中的变量用法。

代码语言：javascript复制

func Test_ExpandEnv(t *testing.T) {
  
 _ = os.Setenv("MY_Age", "18")
 _ = os.Setenv("MY_Name", "Zhangsan")

 // MY_age 大小写不匹配。
 result := os.ExpandEnv("my name is ${MY_Name}, i'm $MY_age years old")
 fmt.Println(result) // my name is Zhangsan, i'm  years old
}

参考资料

[1]

envutils - fix: lost value when trans env string slice into map: https://github.com/tangx/envutils/commit/ca10e1c057193283ef308ae708ef421de3d1ec1b

go 编程算法 bash bash指令

0 人点赞

一篇文章告诉你 golang 环境变量的所有基础操作