新增切片越界的检查

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(
-		"gorm",
-		zap.String("type", "sql"),
-		zap.Any("src", data[0]),
-		zap.Any("duration", data[1]),
-		zap.Any("rows", data[2]),
-		zap.Any("sql", data[3]),
-	)
+	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]))
+	}
 }