快捷導(dǎo)航

Go 使用os包操作環(huán)境變量的方法

更新時間：2024年07月30日 11:08:17 作者：經(jīng)云

環(huán)境變量通常在程序啟動時就已設(shè)置好,在需要的時候隨時讀取,Go使用簡單的幾個函數(shù)就可以對環(huán)境變量進(jìn)行增刪查改,本文給大家介紹Go 使用os包操作環(huán)境變量的方法,感興趣的朋友跟隨小編一起看看吧

一、列出所有環(huán)境變量

func Environ() []string

Environ 返回一個字符串切片的副本，這些字符串代表環(huán)境變量，形式為 “key=value”。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	for _, value := range os.Environ() {
		fmt.Println(value)
	}
}

輸出如下(環(huán)境變量的一部分)：

GOBIN=d:\Program files\Go\bin"
GOFLAGS=-mod=vendor
GOMODCACHE=D:\go\pkg\mod
GOPATH=D:\go
GOPROXY=https://goproxy.io,direct
...

二、清除所有環(huán)境變量

func Clearenv()

它會在程序運(yùn)行時刪除所有環(huán)境變量，程序結(jié)束后，系統(tǒng)環(huán)境變量不會變化，放心，還在。有些地方說這個命令很危險(xiǎn)，其實(shí)沒什么。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	os.Clearenv()
	for _, value := range os.Environ() {
		fmt.Println(value)
	}
}

輸出結(jié)果為空，說明該命令起作用了。屏蔽第9行，再次運(yùn)行，系統(tǒng)環(huán)境變量再次列出，和一章節(jié)結(jié)果相同。

三、檢索環(huán)境變量

func LookupEnv(key string) (string, bool)

LookupEnv 檢索由鍵命名的環(huán)境變量的值。如果環(huán)境中有該變量，則返回值（可能為空）以及一個為真的布爾值。否則，返回的值將為空，布爾值將為假。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	goPath, bgExist := os.LookupEnv("GOPATH")
	fmt.Println("GOPATH =", goPath, ", bgExist =", bgExist)
	luaPath, blExist := os.LookupEnv("LUAPATH")
	fmt.Println("LUAPATH =", luaPath, ", blExist =", blExist)
}

輸出如下：

GOPATH = D:\go , bgExist = true
LUAPATH = , blExist = false

四、獲取環(huán)境變量

func Getenv(key string) string

Getenv 用于檢索由鍵命名的環(huán)境變量的值。它返回該值，如果變量不存在，則返回空字符串。要區(qū)分空值和未設(shè)置的值，請使用 LookupEnv。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	goPath := os.Getenv("GOPATH")
	fmt.Println("GOPATH = ", goPath)
}

輸出如下：

GOPATH = D:\go

五、設(shè)置環(huán)境變量

func Setenv(key, value string) error

Setenv 設(shè)置由鍵命名的環(huán)境變量的值。如果遇到任何錯誤，它將返回該錯誤。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	err := os.Setenv("TESTENV", "TestValue")
	if err != nil {
		os.Exit(1)
	}
	testValue := os.Getenv("TESTENV")
	fmt.Println("TESTENV = ", testValue)
}

輸出如下：

TESTENV = TestValue

六、刪除環(huán)境變量

func Unsetenv(key string) error

Unsetenv 刪除單個環(huán)境變量。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	err := os.Unsetenv("TESTENV")
	if err != nil {
		os.Exit(1)
	}
	testValue := os.Getenv("TESTENV")
	fmt.Println("TESTENV = ", testValue)
}

輸出為：

TESTENV =

七、展開環(huán)境變量

func ExpandEnv(s string) string

ExpandEnv 會根據(jù)當(dāng)前環(huán)境變量的值，將字符串中的 ${var} 或 $var 替換掉。對未定義變量的引用將被空字符串替換。相當(dāng)于不需要os.Getenv了。

示例：

package main
import (
	"fmt"
	"os"
)
func main() {
	os.Setenv("GAME_NAME", "Chronicles of Darkness")
	os.Setenv("GAME_DIR", "/usr/games")
	fmt.Println(os.ExpandEnv("The $GAME_NAME game is placed in the ${GAME_DIR} path."))
}

輸出結(jié)果為：