From 5bf9f5cc0ad2689de4e34d628d8ac3168df5142a Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期五, 23 八月 2024 17:41:41 +0800
Subject: [PATCH] 修复编译bug

---
 config/config.go |   44 +++++++++++++++++++++++++++++++++-----------
 1 files changed, 33 insertions(+), 11 deletions(-)

diff --git a/config/config.go b/config/config.go
index d113791..95a0312 100644
--- a/config/config.go
+++ b/config/config.go
@@ -9,14 +9,15 @@
 )
 
 type serve struct {
-	ID       string `mapstructure:"id"`
-	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 鍏ㄥ姛鑳�
+	ID        string `mapstructure:"id"`
+	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 鍏ㄥ姛鑳�
+	Keepalive bool   `mapstructure:"keepalive"`
 }
 
 type client struct {
@@ -31,6 +32,7 @@
 	ChannelNo          string `mapstructure:"channel-number"`     // 閫氶亾鍙�, 鍚宨d
 	HeartbeatInterval  int    `mapstructure:"heartbeat-interval"` // 蹇冭烦鍛ㄦ湡
 	HeartbeatFailCount int    `mapstructure:"heartbeat-count"`    // 蹇冭烦瓒呮椂娆℃暟
+	AddFloorToFaceId   bool   `mapstructure:"add-floor-faceId"`
 	Enable             bool   `mapstructure:"enable"`
 }
 
@@ -46,14 +48,23 @@
 	ReportServer   string `mapstructure:"report-server"`
 	ReportInterval int    `mapstructure:"report-interval"`
 	RetryInterval  int    `mapstructure:"retry-interval"`
+	FTPServer      string `mapstructure:"ftp-server"`
+	RecordServer   string `mapstructure:"record-server"`
+}
+
+type image struct {
 	CutFaceImage   bool   `mapstructure:"cut-face-image"`
-	Enable         bool   `mapstructure:"enable"`
+	Enlarge        int    `mapstructure:"enlarge"`
+	ImageUriPrefix string `mapstructure:"url-prefix"`
 }
 
 // 姊帶璁惧
 type nvcs struct {
-	Model string `mapstructure:"model"` // 鍨嬪彿
-	Port  string `mapstructure:"port"`  // 绔彛
+	Model       string `mapstructure:"model"` // 鍨嬪彿
+	Port        string `mapstructure:"port"`  // 绔彛
+	OSD         string `mapstructure:"osd"`
+	RunState    bool   `mapstructure:"run-state"`
+	WaitRunTime int    `mapstructure:"wait-run-time"`
 }
 
 type rateLimitConfig struct {
@@ -67,6 +78,7 @@
 var RateLimitConf = &rateLimitConfig{}
 var ClientConf = &client{}
 var NVCSConf = &nvcs{}
+var ImageConf = &image{}
 
 // Init is an exported method that takes the environment starts the viper
 // (external lib) and returns the configuration struct.
@@ -75,6 +87,7 @@
 	v := viper.New()
 	v.SetConfigType("yaml")
 	v.SetConfigName("gat1400")
+	v.AddConfigPath("./")
 	v.AddConfigPath("./config/")
 	v.AddConfigPath("../config/")
 	err = v.ReadInConfig()
@@ -96,6 +109,7 @@
 	v.UnmarshalKey("rate-limit", RateLimitConf)
 	v.UnmarshalKey("client", ClientConf)
 	v.UnmarshalKey("nvcs", NVCSConf)
+	v.UnmarshalKey("image", ImageConf)
 
 	if LogConf.Level == "" {
 		LogConf.Level = "info"
@@ -117,5 +131,13 @@
 		ClientConf.Proto = "http"
 	}
 
+	if ImageConf.Enlarge == 0 {
+		ImageConf.Enlarge = 100
+	}
+
+	if NVCSConf.WaitRunTime == 0 {
+		NVCSConf.WaitRunTime = 60
+	}
+
 	logger.SetLogLevel(LogConf.Level)
 }

--
Gitblit v1.8.0