gf/g/os/genv/genv.go

31 lines
702 B
Go

// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.
// 环境变量管理
package genv
import "os"
func All() []string {
return os.Environ()
}
// 获取环境变量,并可以指定当环境变量不存在时的默认值
func Get(k string, def...string) string {
v, ok := os.LookupEnv(k)
if !ok && len(def) > 0 {
return def[0]
}
return v
}
func Set(k, v string) error {
return os.Setenv(k, v)
}
func Remove(k string) error {
return os.Unsetenv(k)
}