Переглянути джерело

优化修改代码, 允许日志模块从其它配置文件加载日志设置

niujiuru 2 тижнів тому
батько
коміт
aaba8077e0
3 змінених файлів з 10 додано та 4 видалено
  1. 1 1
      baseapp/app.go
  2. 6 2
      baseapp/log.go
  3. 3 1
      sshd/server/server.go

+ 1 - 1
baseapp/app.go

@@ -21,7 +21,7 @@ var exitCh = make(chan struct{})
 
 func ModuleInit() {
 	InitPath()
-	InitLogger()
+	InitLogger("")
 }
 
 func SingleInstanceRun() { // 非阻塞单实例运行, 调用此函数后, 安全退出时, 需调用SafeExit()函数

+ 6 - 2
baseapp/log.go

@@ -123,8 +123,12 @@ var (
 	}
 )
 
-func InitLogger() {
-	cfgFile := filepath.Join(CFG_DIR, "config.ini")
+func InitLogger(logCfgFile string) {
+	cfgFile := logCfgFile
+	if cfgFile == "" {
+		cfgFile = filepath.Join(CFG_DIR, "config.ini")
+	}
+
 	cfgIni, err := ini.Load(cfgFile) // 从配置文件中加载相关配置项覆盖默认值
 	if err == nil && cfgIni.HasSection("Log") {
 		tmpCfgLog := *cfgLog

+ 3 - 1
sshd/server/server.go

@@ -45,7 +45,9 @@ func loadCfgServers() error {
 }
 
 func main() {
-	baseapp.ModuleInit()
+	baseapp.InitPath()
+	cfgFile := filepath.Join(baseapp.CFG_DIR, "sshd_cfg.ini")
+	baseapp.InitLogger(cfgFile)
 	baseapp.SingleInstanceRun() // 异步非阻塞
 	baseapp.Logger.Infof("程序版本: %s 构建时间: %s\n程序开始运行...",
 		Version, baseapp.BuildTime)