backend/pkg/env/env.go

38 lines
719 B
Go
Raw Permalink Normal View History

2022-01-11 00:48:03 +08:00
package env
2022-01-11 04:00:39 +08:00
import (
"fmt"
"os"
)
func init() {
AppName = GetEnvDefault("APP_NAME", "murder")
Namespace = GetEnvDefault("NAMESPACE", "brahma")
2022-01-11 04:00:39 +08:00
AppType = SAppType(GetEnvDefault("APP_TYPE", string(AppTypeAPI)))
}
var AppName string
var Namespace string
var AppType SAppType
type SAppType string
const (
AppTypeAPI = SAppType("api") // api应用类型
AppTypeService = SAppType("service") // service应用类型
)
2022-01-11 00:48:03 +08:00
func GetEnvDefault(key, defVal string) string {
val, ex := os.LookupEnv(key)
if !ex {
os.Setenv(key, defVal)
return defVal
}
return val
}
2022-01-11 04:00:39 +08:00
// GetAppNameWithType 获取带应用类型的应用名
func GetAppNameWithType() string {
return fmt.Sprintf("%s-%s", AppType, AppName)
}