From e6720a590f9196dc5b1ef6475e7fd5d2d8709e63 Mon Sep 17 00:00:00 2001
From: lixxxww <941403820@qq.com>
Date: Tue, 23 Jan 2024 11:36:21 +0000
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4kit/logger?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: lixxxww <941403820@qq.com>
---
 kit/logger/logger.go | 64 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 kit/logger/logger.go

diff --git a/kit/logger/logger.go b/kit/logger/logger.go
new file mode 100644
index 0000000..dbee893
--- /dev/null
+++ b/kit/logger/logger.go
@@ -0,0 +1,64 @@
+package logger
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/sirupsen/logrus"
+)
+
+var Log *logrus.Logger
+
+func InitLog(fileName, level string) *logrus.Logger {
+	Log = logrus.New()
+	Log.SetFormatter(new(LogFormatter))
+	Log.SetReportCaller(true)
+
+	// 根据配置文件设置日志级别
+	if level != "" {
+		l, err := logrus.ParseLevel(level)
+		if err != nil {
+			panic(any(fmt.Sprintf("日志级别不存在: %s", level)))
+		}
+		Log.SetLevel(l)
+	} else {
+		Log.SetLevel(logrus.DebugLevel)
+	}
+	if fileName != "" {
+		file, err := os.OpenFile(fileName, os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModeAppend|0666)
+		if err != nil {
+			panic(any(fmt.Sprintf("创建日志文件失败: %s", err.Error())))
+		}
+		Log.Out = file
+	}
+
+	return Log
+}
+
+type LogFormatter struct{}
+
+func (l *LogFormatter) Format(entry *logrus.Entry) ([]byte, error) {
+	timestamp := time.Now().Local().Format("2006-01-02 15:04:05.000")
+	level := entry.Level
+	logMsg := fmt.Sprintf("%s [%s]", timestamp, strings.ToUpper(level.String()))
+	// 如果存在调用信息，且为error级别以上记录文件及行号
+	if caller := entry.Caller; caller != nil {
+		var fp string
+		// 全路径切割，只获取项目相关路径，
+		// 即/Users/hml/Desktop/project/go/pandax/test.go只获取/test.go
+		ps := strings.Split(caller.File, "pandax/")
+		if len(ps) >= 2 {
+			fp = ps[1]
+		} else {
+			fp = ps[0]
+		}
+		logMsg = logMsg + fmt.Sprintf(" [%s:%d]", fp, caller.Line)
+	}
+	for k, v := range entry.Data {
+		logMsg = logMsg + fmt.Sprintf(" [%s=%v]", k, v)
+	}
+	logMsg = logMsg + fmt.Sprintf(" : %s\n", entry.Message)
+	return []byte(logMsg), nil
+}