From 5cf6309c132fdf1e110fb23f419e6971c63f533c Mon Sep 17 00:00:00 2001
From: gongshangguo <gongshangguo@admin.com>
Date: 星期五, 25 二月 2022 18:51:32 +0800
Subject: [PATCH] logger

---
 logger.go |   26 ++++++++++++++++++++++++--
 1 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/logger.go b/logger.go
index aa3b124..685626b 100644
--- a/logger.go
+++ b/logger.go
@@ -37,17 +37,38 @@
 	FatalLevel
 )
 
-func InitLogger(logPath string, logLevel int,maxSize int, maxBackups int, maxAge int) {
+func InitLogger(logPath string, logLevel int,maxSize int, maxBackups int, maxAge int) *zap.SugaredLogger {
 	logdir := "./logger/"
+	logName := ""
 	if index := strings.LastIndex(logPath, "/"); index != -1 {
 		logdir = logPath[0:index] + "/"
+		if index < len(logPath)-1 {
+			logName = logPath[index+1:]
+		}
+	} else {
+		logName = logPath
+	}
+	if logName == "" {
+		logName = "log.log"
 	}
 	fi,err := os.Stat(logdir)
 	if !((err == nil || os.IsExist(err)) && fi.IsDir()) {
 		os.MkdirAll(logdir, os.ModePerm)
 	}
+	if logLevel < DebugLevel || logLevel > FatalLevel {
+		logLevel = DebugLevel
+	}
+	if maxSize <=0 {
+		maxSize = 128
+	}
+	if maxBackups <=0 {
+		maxBackups = 30
+	}
+	if maxAge <= 0 {
+		maxAge = 15
+	}
 	hook := lumberjack.Logger {
-		Filename: logPath,   //鏃ュ織鏂囦欢鐨勪綅缃�
+		Filename: logdir+logName,   //鏃ュ織鏂囦欢鐨勪綅缃�
 		MaxSize: maxSize,   	 //鍦ㄨ繘琛屽垏鍓蹭箣鍓嶏紝鏃ュ織鏂囦欢鐨勬渶澶уぇ灏忥紙浠B涓哄崟浣嶏級
 		MaxBackups: maxBackups, 	 //淇濈暀鏃ф枃浠剁殑鏈�澶т釜鏁�
 		MaxAge: maxAge, 			 //淇濈暀鏃ф枃浠剁殑鏈�澶уぉ鏁�
@@ -78,6 +99,7 @@
 		logger: log.Sugar(),
 	}
 	blog.logger.Info("init logger success")
+	return blog.logger
 }
 
 func getZapLevel(logLevel int) zapcore.Level {

--
Gitblit v1.8.0