奇淼（piexlmax
4 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
22 changed files with 203 additions and 34 deletions
-
3server/config.yaml
-
1server/config/gorm.go
-
2server/core/server.go
-
6server/initialize/gorm.go
-
150server/initialize/logger.go
-
2server/initialize/router.go
-
12server/middleware/operation.go
-
3server/router/exp_customer.go
-
3server/router/sys_api.go
-
3server/router/sys_authority.go
-
3server/router/sys_casbin.go
-
3server/router/sys_dictionary.go
-
3server/router/sys_dictionary_detail.go
-
3server/router/sys_email.go
-
3server/router/sys_jwt.go
-
3server/router/sys_menu.go
-
11server/router/sys_operation_record.go
-
3server/router/sys_system.go
-
3server/router/sys_user.go
-
3server/router/sys_workflow.go
-
10server/service/sys_auto_code.go
-
2web/src/main.js
@ -0,0 +1,150 @@ |
|||
package initialize |
|||
|
|||
import ( |
|||
"context" |
|||
"fmt" |
|||
"gin-vue-admin/global" |
|||
"go.uber.org/zap" |
|||
"gorm.io/gorm/logger" |
|||
"gorm.io/gorm/utils" |
|||
"io/ioutil" |
|||
"log" |
|||
"os" |
|||
"time" |
|||
) |
|||
|
|||
var ( |
|||
Discard = New(log.New(ioutil.Discard, "", log.LstdFlags), GormConfig{}) |
|||
Default = New(log.New(os.Stdout, "\r\n", log.LstdFlags), GormConfig{ |
|||
SlowThreshold: 200 * time.Millisecond, |
|||
LogLevel: logger.Warn, |
|||
Colorful: true, |
|||
}) |
|||
Recorder = traceRecorder{Interface: Default, BeginAt: time.Now()} |
|||
) |
|||
|
|||
type traceRecorder struct { |
|||
logger.Interface |
|||
BeginAt time.Time |
|||
SQL string |
|||
RowsAffected int64 |
|||
Err error |
|||
} |
|||
|
|||
func New(writer Writer, config GormConfig) logger.Interface { |
|||
var ( |
|||
infoStr = "%s\n[info] " |
|||
warnStr = "%s\n[warn] " |
|||
errStr = "%s\n[error] " |
|||
traceStr = "%s\n[%.3fms] [rows:%v] %s" |
|||
traceWarnStr = "%s %s\n[%.3fms] [rows:%v] %s" |
|||
traceErrStr = "%s %s\n[%.3fms] [rows:%v] %s" |
|||
) |
|||
|
|||
if config.Colorful { |
|||
infoStr = logger.Green + "%s\n" + logger.Reset + logger.Green + "[info] " + logger.Reset |
|||
warnStr = logger.BlueBold + "%s\n" + logger.Reset + logger.Magenta + "[warn] " + logger.Reset |
|||
errStr = logger.Magenta + "%s\n" + logger.Reset + logger.Red + "[error] " + logger.Reset |
|||
traceStr = logger.Green + "%s\n" + logger.Reset + logger.Yellow + "[%.3fms] " + logger.BlueBold + "[rows:%v]" + logger.Reset + " %s" |
|||
traceWarnStr = logger.Green + "%s " + logger.Yellow + "%s\n" + logger.Reset + logger.RedBold + "[%.3fms] " + logger.Yellow + "[rows:%v]" + logger.Magenta + " %s" + logger.Reset |
|||
traceErrStr = logger.RedBold + "%s " + logger.MagentaBold + "%s\n" + logger.Reset + logger.Yellow + "[%.3fms] " + logger.BlueBold + "[rows:%v]" + logger.Reset + " %s" |
|||
} |
|||
|
|||
return &GormLogger{ |
|||
Writer: writer, |
|||
GormConfig: config, |
|||
infoStr: infoStr, |
|||
warnStr: warnStr, |
|||
errStr: errStr, |
|||
traceStr: traceStr, |
|||
traceWarnStr: traceWarnStr, |
|||
traceErrStr: traceErrStr, |
|||
} |
|||
} |
|||
|
|||
// Writer log writer interface
|
|||
type Writer interface { |
|||
Printf(string, ...interface{}) |
|||
} |
|||
|
|||
type GormConfig struct { |
|||
SlowThreshold time.Duration |
|||
Colorful bool |
|||
LogLevel logger.LogLevel |
|||
} |
|||
|
|||
type GormLogger struct { |
|||
Writer |
|||
GormConfig |
|||
infoStr, warnStr, errStr string |
|||
traceStr, traceErrStr, traceWarnStr string |
|||
} |
|||
|
|||
func (g *GormLogger) LogMode(level logger.LogLevel) logger.Interface { |
|||
newLogger := *g |
|||
newLogger.LogLevel = level |
|||
return &newLogger |
|||
} |
|||
|
|||
func (g *GormLogger) Info(ctx context.Context, message string, data ...interface{}) { |
|||
if g.LogLevel >= logger.Info { |
|||
g.Printf(g.infoStr+message, append([]interface{}{utils.FileWithLineNum()}, data...)...) |
|||
} |
|||
} |
|||
|
|||
func (g *GormLogger) Warn(ctx context.Context, message string, data ...interface{}) { |
|||
if g.LogLevel >= logger.Warn { |
|||
g.Printf(g.warnStr+message, append([]interface{}{utils.FileWithLineNum()}, data...)...) |
|||
} |
|||
} |
|||
|
|||
func (g *GormLogger) Error(ctx context.Context, message string, data ...interface{}) { |
|||
if g.LogLevel >= logger.Error { |
|||
g.Printf(g.errStr+message, append([]interface{}{utils.FileWithLineNum()}, data...)...) |
|||
} |
|||
} |
|||
|
|||
func (g *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) { |
|||
if g.LogLevel > 0 { |
|||
elapsed := time.Since(begin) |
|||
switch { |
|||
case err != nil && g.LogLevel >= logger.Error: |
|||
sql, rows := fc() |
|||
if rows == -1 { |
|||
g.Printf(g.traceErrStr, utils.FileWithLineNum(), err, float64(elapsed.Nanoseconds())/1e6, "-", sql) |
|||
} else { |
|||
g.Printf(g.traceErrStr, utils.FileWithLineNum(), err, float64(elapsed.Nanoseconds())/1e6, rows, sql) |
|||
} |
|||
case elapsed > g.SlowThreshold && g.SlowThreshold != 0 && g.LogLevel >= logger.Warn: |
|||
sql, rows := fc() |
|||
slowLog := fmt.Sprintf("SLOW SQL >= %v", g.SlowThreshold) |
|||
if rows == -1 { |
|||
g.Printf(g.traceWarnStr, utils.FileWithLineNum(), slowLog, float64(elapsed.Nanoseconds())/1e6, "-", sql) |
|||
} else { |
|||
g.Printf(g.traceWarnStr, utils.FileWithLineNum(), slowLog, float64(elapsed.Nanoseconds())/1e6, rows, sql) |
|||
} |
|||
case g.LogLevel >= logger.Info: |
|||
sql, rows := fc() |
|||
if rows == -1 { |
|||
g.Printf(g.traceStr, utils.FileWithLineNum(), float64(elapsed.Nanoseconds())/1e6, "-", sql) |
|||
} else { |
|||
g.Printf(g.traceStr, utils.FileWithLineNum(), float64(elapsed.Nanoseconds())/1e6, rows, sql) |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
func (g *GormLogger) Printf(message string, data ...interface{}) { |
|||
switch len(data) { |
|||
case 0: |
|||
global.GVA_LOG.Info(message) |
|||
case 1: |
|||
global.GVA_LOG.Info("gorm", zap.Any("src", data[0])) |
|||
case 2: |
|||
global.GVA_LOG.Info("gorm", zap.Any("src", data[0]), zap.Any("duration", data[1])) |
|||
case 3: |
|||
global.GVA_LOG.Info("gorm", zap.Any("src", data[0]), zap.Any("duration", data[1]), zap.Any("rows", data[2])) |
|||
case 4: |
|||
global.GVA_LOG.Info("gorm", zap.Any("src", data[0]), zap.Any("duration", data[1]), zap.Any("rows", data[2]), zap.Any("sql", data[3])) |
|||
} |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue