重命名包 client 为 edge;重命名包 server 为 gateway
This commit is contained in:
53
edge/env/env.go
vendored
Normal file
53
edge/env/env.go
vendored
Normal file
@@ -0,0 +1,53 @@
|
||||
package env
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"flag"
|
||||
"log/slog"
|
||||
)
|
||||
|
||||
var Mode = "dev"
|
||||
var Name = "dev-edge"
|
||||
|
||||
var EndpointOnline = "https://api.lanhuip.com/api/edge/online"
|
||||
var EndpointOffline = "https://api.lanhuip.com/api/edge/offline"
|
||||
|
||||
func Init() error {
|
||||
|
||||
var env = flag.String("e", "dev", "环境变量,可选值 dev 或 prod")
|
||||
var name = flag.String("n", "", "客户端唯一标识")
|
||||
var online = flag.String("online", "", "服务注册地址")
|
||||
var offline = flag.String("offline", "", "服务注销地址")
|
||||
|
||||
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 online != nil && *online != "" {
|
||||
EndpointOnline = *online
|
||||
}
|
||||
|
||||
if offline != nil && *offline != "" {
|
||||
EndpointOffline = *offline
|
||||
}
|
||||
|
||||
if Mode == "dev" {
|
||||
slog.SetLogLoggerLevel(slog.LevelDebug)
|
||||
} else {
|
||||
slog.SetLogLoggerLevel(slog.LevelWarn)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user