From 8ba0a125eb7302d8c51d3c85cf8ea8a4de5413db Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期六, 25 五月 2024 17:56:27 +0800
Subject: [PATCH] 修复ape类型判断

---
 config/config.go |   13 +++++++++++++
 1 files changed, 13 insertions(+), 0 deletions(-)

diff --git a/config/config.go b/config/config.go
index 9eaad86..922d680 100644
--- a/config/config.go
+++ b/config/config.go
@@ -13,10 +13,14 @@
 	Mode     string `mapstructure:"mode"`
 	Host     string `mapstructure:"host"`
 	Port     string `mapstructure:"port"`
+	Realm    string `mapstructure:"realm"`
+	Username string `mapstructure:"username"`
 	Password string `mapstructure:"password"`
+	Role     string `mapstructure:"role"` // agent 璁惧绔�, proxy 1400涓浆 涓嶅鐞哾eivce, server 鍏ㄥ姛鑳�
 }
 
 type client struct {
+	ServerId           string `mapstructure:"server-id"`
 	DeviceID           string `mapstructure:"device-id"`
 	Username           string `mapstructure:"username"`
 	Password           string `mapstructure:"password"`
@@ -46,6 +50,13 @@
 	Enable         bool   `mapstructure:"enable"`
 }
 
+// 姊帶璁惧
+type nvcs struct {
+	Model string `mapstructure:"model"` // 鍨嬪彿
+	Port  string `mapstructure:"port"`  // 绔彛
+	OSD   string `mapstructure:"osd"`
+}
+
 type rateLimitConfig struct {
 	FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"`
 	Capacity     int64 `mapstructure:"capacity" json:"capacity"`
@@ -56,6 +67,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.
@@ -84,6 +96,7 @@
 	v.UnmarshalKey("forward", ForwardConf)
 	v.UnmarshalKey("rate-limit", RateLimitConf)
 	v.UnmarshalKey("client", ClientConf)
+	v.UnmarshalKey("nvcs", NVCSConf)
 
 	if LogConf.Level == "" {
 		LogConf.Level = "info"

--
Gitblit v1.8.0