新增切片越界的检查

92c7ae1d · m0_50812349 · b1ae6cbe · 92c7ae1d
隐藏空白更改
内联并排

Showing with 12 addition and 8 deletion

server/initialize/logger.go server/initialize/logger.go +12 -8

未找到文件。
--- a/server/initialize/logger.go
+++ b/server/initialize/logger.go
@@ -135,12 +135,16 @@ func (g *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (stri
 }
 func (g *GormLogger) Printf(message string, data ...interface{}) {
-	global.GVA_LOG.Info(
+	switch len(data) {
-		"gorm",
+	case 0:
-		zap.String("type", "sql"),
+		global.GVA_LOG.Info(message)
-		zap.Any("src", data[0]),
+	case 1:
-		zap.Any("duration", data[1]),
+		global.GVA_LOG.Info("gorm", zap.Any("src", data[0]))
-		zap.Any("rows", data[2]),
+	case 2:
-		zap.Any("sql", data[3]),
+		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]))
+	}
 }