补充财务查询索引和慢 SQL 日志

backend/internal/database/mysql.go

@@ -1,14 +1,20 @@
 package database
 
 import (
+	"log"
+	"os"
+	"strings"
 	"time"
 
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 )
 
-func OpenMySQL(dsn string) (*gorm.DB, error) {
-	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
+func OpenMySQL(dsn string, logLevel string) (*gorm.DB, error) {
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+		Logger: newGormLogger(logLevel),
+	})
 	if err != nil {
 		return nil, err
 	}
@@ -25,3 +31,19 @@ func OpenMySQL(dsn string) (*gorm.DB, error) {
 
 	return db, nil
 }
+
+func newGormLogger(logLevel string) logger.Interface {
+	level := logger.Warn
+	if strings.EqualFold(strings.TrimSpace(logLevel), "debug") {
+		level = logger.Info
+	}
+	return logger.New(
+		log.New(os.Stdout, "\r\n", log.LstdFlags),
+		logger.Config{
+			SlowThreshold:             500 * time.Millisecond,
+			LogLevel:                  level,
+			IgnoreRecordNotFoundError: true,
+			Colorful:                  false,
+		},
+	)
+}