You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go-zero/core/proc/env.go

44 lines
510 B
Go

package proc
import (
"os"
"strconv"
"sync"
)
var (
envs = make(map[string]string)
envLock sync.RWMutex
)
func Env(name string) string {
envLock.RLock()
val, ok := envs[name]
envLock.RUnlock()
if ok {
return val
}
val = os.Getenv(name)
envLock.Lock()
envs[name] = val
envLock.Unlock()
return val
}
func EnvInt(name string) (int, bool) {
val := Env(name)
if len(val) == 0 {
return 0, false
}
n, err := strconv.Atoi(val)
if err != nil {
return 0, false
}
return n, true
}