✨ 新增支持Gorm日志

gorm.go

@@ -0,0 +1,21 @@
+package logger
+
+import "strings"
+
+// 基于Gorm的日志实现
+type gormLogger struct{}
+
+// 打印
+func (gormLogger) Write(p []byte) (n int, err error) {
+	str := string(p)
+	// 去掉第一行
+	//str = strings.Split(str, "\n")[1]
+	str = strings.Join(strings.Split(str, "\n")[1:], " ")
+	Say.Debug(str)
+	return 0, nil
+}
+
+// NewGormLogger ...
+func NewGormLogger() *gormLogger {
+	return &gormLogger{}
+}

logger.go

@@ -10,7 +10,9 @@ var Say *zap.SugaredLogger
 
 // 避免异常，在第一次调用时初始化一个只打印到控制台的logger
 func init() {
-	InitLogger(LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false})
+	if Say == nil {
+		InitLogger(LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false})
+	}
 }
 
 // InitLogger 初始化日志工具