48 lines
867 B
Go
48 lines
867 B
Go
package env
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"log/slog"
|
|
)
|
|
|
|
var Mode = "dev"
|
|
var Name = "dev-edge"
|
|
|
|
var EndpointAssign = "https://api.lanhuip.com/api/edge/assign"
|
|
|
|
func Init() error {
|
|
|
|
var env = flag.String("e", "dev", "环境变量，可选值 dev 或 prod")
|
|
var name = flag.String("n", "", "节点唯一标识")
|
|
var assign = flag.String("assign", "", "服务发现地址")
|
|
|
|
flag.Parse()
|
|
|
|
if env != nil && *env != "" {
|
|
if *env == "dev" || *env == "prod" {
|
|
Mode = *env
|
|
} else {
|
|
return errors.New("环境变量只能为 dev 或 prod")
|
|
}
|
|
}
|
|
|
|
if name != nil && *name != "" {
|
|
Name = *name
|
|
} else {
|
|
return errors.New("节点唯一标识不能为空")
|
|
}
|
|
|
|
if assign != nil && *assign != "" {
|
|
EndpointAssign = *assign
|
|
}
|
|
|
|
if Mode == "dev" {
|
|
slog.SetLogLoggerLevel(slog.LevelDebug)
|
|
} else {
|
|
slog.SetLogLoggerLevel(slog.LevelWarn)
|
|
}
|
|
|
|
return nil
|
|
}
|