From e5be7a7ed63d4b4e4267057d84b77c4bd3c16781 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 16 五月 2024 13:09:28 +0800
Subject: [PATCH] 添加订阅修改接口
---
config/config.go | 46 ++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 46 insertions(+), 0 deletions(-)
diff --git a/config/config.go b/config/config.go
index c5b686c..84b926e 100644
--- a/config/config.go
+++ b/config/config.go
@@ -2,6 +2,7 @@
import (
"fmt"
+ "gat1400Exchange/pkg/logger"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
@@ -12,7 +13,24 @@
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 {
+ DeviceID string `mapstructure:"device-id"`
+ Username string `mapstructure:"username"`
+ Password string `mapstructure:"password"`
+ ServerAddr string `mapstructure:"server-addr"`
+ ServerPort string `mapstructure:"server-port"`
+ Proto string `mapstructure:"proto" ` // http, https
+ UploadType string `mapstructure:"upload-type"` // binary, url
+ ChannelNo string `mapstructure:"channel-number"` // 閫氶亾鍙�, 鍚宨d
+ HeartbeatInterval int `mapstructure:"heartbeat-interval"` // 蹇冭烦鍛ㄦ湡
+ HeartbeatFailCount int `mapstructure:"heartbeat-count"` // 蹇冭烦瓒呮椂娆℃暟
+ Enable bool `mapstructure:"enable"`
}
type logConfig struct {
@@ -26,6 +44,9 @@
SyncServer string `mapstructure:"sync-server"`
ReportServer string `mapstructure:"report-server"`
ReportInterval int `mapstructure:"report-interval"`
+ RetryInterval int `mapstructure:"retry-interval"`
+ CutFaceImage bool `mapstructure:"cut-face-image"`
+ Enable bool `mapstructure:"enable"`
}
type rateLimitConfig struct {
@@ -37,6 +58,7 @@
var LogConf = &logConfig{}
var ForwardConf = &forward{}
var RateLimitConf = &rateLimitConfig{}
+var ClientConf = &client{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -46,6 +68,7 @@
v.SetConfigType("yaml")
v.SetConfigName("gat1400")
v.AddConfigPath("./config/")
+ v.AddConfigPath("../config/")
err = v.ReadInConfig()
if err != nil {
fmt.Println("error on parsing configuration file", err)
@@ -63,4 +86,27 @@
v.UnmarshalKey("log", LogConf)
v.UnmarshalKey("forward", ForwardConf)
v.UnmarshalKey("rate-limit", RateLimitConf)
+ v.UnmarshalKey("client", ClientConf)
+
+ if LogConf.Level == "" {
+ LogConf.Level = "info"
+ }
+
+ if ForwardConf.ReportInterval == 0 {
+ ForwardConf.ReportInterval = 10
+ }
+
+ if ForwardConf.RetryInterval == 0 {
+ ForwardConf.RetryInterval = 5
+ }
+
+ if ClientConf.HeartbeatInterval == 0 {
+ ClientConf.HeartbeatInterval = 30
+ }
+
+ if ClientConf.Proto == "" {
+ ClientConf.Proto = "http"
+ }
+
+ logger.SetLogLevel(LogConf.Level)
}
--
Gitblit v1.8.0