util: add env var util functions

This commit is contained in:
c9s 2021-06-08 00:07:56 +08:00
parent ec6c10a96a
commit ecf888dfd6

38
pkg/util/envvars.go Normal file
View File

@ -0,0 +1,38 @@
package util
import (
"os"
"strconv"
"github.com/sirupsen/logrus"
)
func GetEnvVarInt(n string) (int, bool) {
str, ok := os.LookupEnv(n)
if !ok {
return 0, false
}
num, err := strconv.Atoi(str)
if err != nil {
logrus.WithError(err).Errorf("can not parse env var %q as int, incorrect format", str)
return 0, false
}
return num, true
}
func GetEnvVarBool(n string) (bool, bool) {
str, ok := os.LookupEnv(n)
if !ok {
return false, false
}
num, err := strconv.ParseBool(str)
if err != nil {
logrus.WithError(err).Errorf("can not parse env var %q as bool, incorrect format", str)
return false, false
}
return num, true
}