From 5f31d07ff92d87dd1016e9279df84d8f3641feb5 Mon Sep 17 00:00:00 2001 From: zhangzengfei <zhangzengfei@smartai.com> Date: 星期日, 25 八月 2024 22:26:29 +0800 Subject: [PATCH] 添加系统校时功能 --- config/config.go | 22 ++++++++++++++++++---- 1 files changed, 18 insertions(+), 4 deletions(-) diff --git a/config/config.go b/config/config.go index 95a0312..f4609ee 100644 --- a/config/config.go +++ b/config/config.go @@ -36,7 +36,7 @@ Enable bool `mapstructure:"enable"` } -type logConfig struct { +type log struct { LogWay string `mapstructure:"log-way"` // 鏃ュ織杈撳嚭 Path string `mapstructure:"path"` // 鏃ュ織瀛樺偍璺緞 Level string `mapstructure:"level"` // 鏃ュ織绛夌骇 @@ -67,18 +67,24 @@ WaitRunTime int `mapstructure:"wait-run-time"` } -type rateLimitConfig struct { +type rateLimit struct { FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"` Capacity int64 `mapstructure:"capacity" json:"capacity"` } +type sysTime struct { + NTPServer string `mapstructure:"ntp-server" json:"ntp-server"` + SyncInterval int `mapstructure:"sync-interval" json:"sync-interval"` +} + var ServeConf = &serve{} -var LogConf = &logConfig{} +var LogConf = &log{} var ForwardConf = &forward{} -var RateLimitConf = &rateLimitConfig{} +var RateLimitConf = &rateLimit{} var ClientConf = &client{} var NVCSConf = &nvcs{} var ImageConf = &image{} +var SysTimeConf = &sysTime{} // Init is an exported method that takes the environment starts the viper // (external lib) and returns the configuration struct. @@ -139,5 +145,13 @@ NVCSConf.WaitRunTime = 60 } + if SysTimeConf.SyncInterval == 0 { + SysTimeConf.SyncInterval = 10 + } + + if SysTimeConf.NTPServer == "" { + SysTimeConf.NTPServer = "ntp.aliyun.com" + } + logger.SetLogLevel(LogConf.Level) } -- Gitblit v1.8.0