添加webapi, 优化模块启动

zhangzengfei

2024-05-17 9ee887fce2f87f7a79d0b94640cf1d341a254319

 config/config.go

@@ -20,6 +20,7 @@
}

type client struct {
   ServerId           string `mapstructure:"server-id"`
   DeviceID           string `mapstructure:"device-id"`
   Username           string `mapstructure:"username"`
   Password           string `mapstructure:"password"`
@@ -49,6 +50,12 @@
   Enable         bool   `mapstructure:"enable"`
}

// 梯控设备
type nvcs struct {
   Model string // 型号
   Port  string // 端口
}

type rateLimitConfig struct {
   FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"`
   Capacity     int64 `mapstructure:"capacity" json:"capacity"`
@@ -59,6 +66,7 @@
var ForwardConf = &forward{}
var RateLimitConf = &rateLimitConfig{}
var ClientConf = &client{}
var NVCSConf = &nvcs{}

// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -87,6 +95,7 @@
   v.UnmarshalKey("forward", ForwardConf)
   v.UnmarshalKey("rate-limit", RateLimitConf)
   v.UnmarshalKey("client", ClientConf)
   v.UnmarshalKey("nvcs", NVCSConf)

   if LogConf.Level == "" {
      LogConf.Level = "info"