init

at-api/logger.go

@@ -0,0 +1,64 @@
+package main
+
+import (
+	utils "anytunnel/at-common"
+
+	"github.com/snail007/mini-logger"
+	"github.com/snail007/mini-logger/writers/console"
+	"github.com/snail007/mini-logger/writers/files"
+)
+
+var log logger.MiniLogger
+var accessLog logger.MiniLogger
+
+//initLog
+func initLog() {
+	var level uint8
+	switch cfg.GetString("log.console-level") {
+	case "debug":
+		level = logger.AllLevels
+	case "info":
+		level = logger.InfoLevel | logger.WarnLevel | logger.ErrorLevel | logger.FatalLevel
+	case "warn":
+		level = logger.WarnLevel | logger.ErrorLevel | logger.FatalLevel
+	case "error":
+		level = logger.ErrorLevel | logger.FatalLevel
+	case "fatal":
+		level = logger.FatalLevel
+	default:
+		level = 0
+	}
+	log = logger.New(false, nil)
+
+	log.AddWriter(console.NewDefault(), level)
+
+	cfgF := files.GetDefaultFileConfig()
+	cfgF.LogPath = cfg.GetString("log.dir")
+	cfgF.MaxBytes = cfg.GetInt64("log.FileMaxSize")
+	cfgF.MaxCount = cfg.GetInt("log.MaxCount")
+	cfgLevels := cfg.GetStringSlice("log.level")
+	if ok, _ := utils.InArray("debug", cfgLevels); ok {
+		cfgF.FileNameSet["debug"] = logger.AllLevels
+	}
+	if ok, _ := utils.InArray("info", cfgLevels); ok {
+		cfgF.FileNameSet["info"] = logger.InfoLevel
+	}
+	if ok, _ := utils.InArray("error", cfgLevels); ok {
+		cfgF.FileNameSet["error"] = logger.WarnLevel | logger.ErrorLevel | logger.FatalLevel
+	}
+	if cfg.GetBool("log.open") {
+		log.AddWriter(files.New(cfgF), logger.AllLevels)
+	}
+
+	accessLog = logger.New(false, nil)
+	//accessLog.AddWriter(console.NewDefault(), logger.AllLevels)
+
+	if cfg.GetBool("log.access") {
+		accessCfg := files.GetDefaultFileConfig()
+		accessCfg.LogPath = cfg.GetString("log.dir")
+		accessCfg.MaxBytes = cfg.GetInt64("log.FileMaxSize")
+		accessCfg.MaxCount = cfg.GetInt("log.MaxCount")
+		accessCfg.FileNameSet = map[string]uint8{"access": logger.InfoLevel}
+		accessLog.AddWriter(files.New(accessCfg), logger.InfoLevel)
+	}
+}