fixed: 修复sql日志无法加入统一zap输出的bug

This commit is contained in:
piexlMax(奇淼 2024-12-25 13:51:02 +08:00
parent cb23254152
commit 8b93ede959
2 changed files with 9 additions and 11 deletions

View File

@ -6,8 +6,6 @@ import (
"gorm.io/gorm" "gorm.io/gorm"
"gorm.io/gorm/logger" "gorm.io/gorm/logger"
"gorm.io/gorm/schema" "gorm.io/gorm/schema"
"log"
"os"
"time" "time"
) )
@ -34,7 +32,7 @@ func (g *_gorm) Config(prefix string, singular bool) *gorm.Config {
general = global.GVA_CONFIG.Mysql.GeneralDB general = global.GVA_CONFIG.Mysql.GeneralDB
} }
return &gorm.Config{ return &gorm.Config{
Logger: logger.New(NewWriter(general, log.New(os.Stdout, "\r\n", log.LstdFlags)), logger.Config{ Logger: logger.New(NewWriter(general), logger.Config{
SlowThreshold: 200 * time.Millisecond, SlowThreshold: 200 * time.Millisecond,
LogLevel: general.LogLevel(), LogLevel: general.LogLevel(),
Colorful: true, Colorful: true,

View File

@ -3,7 +3,7 @@ package internal
import ( import (
"fmt" "fmt"
"github.com/flipped-aurora/gin-vue-admin/server/config" "github.com/flipped-aurora/gin-vue-admin/server/config"
"go.uber.org/zap" "github.com/flipped-aurora/gin-vue-admin/server/global"
"gorm.io/gorm/logger" "gorm.io/gorm/logger"
) )
@ -12,8 +12,8 @@ type Writer struct {
writer logger.Writer writer logger.Writer
} }
func NewWriter(config config.GeneralDB, writer logger.Writer) *Writer { func NewWriter(config config.GeneralDB) *Writer {
return &Writer{config: config, writer: writer} return &Writer{config: config}
} }
// Printf 格式化打印日志 // Printf 格式化打印日志
@ -21,15 +21,15 @@ func (c *Writer) Printf(message string, data ...any) {
if c.config.LogZap { if c.config.LogZap {
switch c.config.LogLevel() { switch c.config.LogLevel() {
case logger.Silent: case logger.Silent:
zap.L().Debug(fmt.Sprintf(message, data...)) global.GVA_LOG.Debug(fmt.Sprintf(message, data...))
case logger.Error: case logger.Error:
zap.L().Error(fmt.Sprintf(message, data...)) global.GVA_LOG.Error(fmt.Sprintf(message, data...))
case logger.Warn: case logger.Warn:
zap.L().Warn(fmt.Sprintf(message, data...)) global.GVA_LOG.Warn(fmt.Sprintf(message, data...))
case logger.Info: case logger.Info:
zap.L().Info(fmt.Sprintf(message, data...)) global.GVA_LOG.Info(fmt.Sprintf(message, data...))
default: default:
zap.L().Info(fmt.Sprintf(message, data...)) global.GVA_LOG.Info(fmt.Sprintf(message, data...))
} }
return return
} }