fix

internal/data/database.go

@@ -8,6 +8,7 @@ import (
 	"path/filepath"
 	"regexp"
 	"strings"
+	"time"
 
 	oracle "github.com/dzwvip/gorm-oracle"
 	"github.com/glebarez/sqlite"
@@ -15,6 +16,8 @@ import (
 	"gorm.io/driver/postgres"
 	"gorm.io/driver/sqlserver"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
 	"kra/internal/conf"
 )
 
@@ -99,21 +102,61 @@ func databaseDSN(c *conf.Data_Database, name string) (string, error) {
 	return "", fmt.Errorf("unsupported database driver %q", c.Driver)
 }
 
+func gormConfig(config *conf.Data_Database) *gorm.Config {
+	level := logger.Info
+	switch strings.ToLower(config.LogMode) {
+	case "silent":
+		level = logger.Silent
+	case "error":
+		level = logger.Error
+	case "warn":
+		level = logger.Warn
+	}
+	return &gorm.Config{Logger: logger.Default.LogMode(level), NamingStrategy: schema.NamingStrategy{TablePrefix: config.Prefix, SingularTable: config.Singular}}
+}
+
 func openWithDriver(driver, dsn string) (*gorm.DB, error) {
+	return openWithDriverConfig(driver, dsn, &conf.Data_Database{Driver: driver})
+}
+
+func openWithDriverConfig(driver, dsn string, config *conf.Data_Database) (*gorm.DB, error) {
+	gormConfig := gormConfig(config)
+	var db *gorm.DB
+	var err error
 	switch normalizedDriver(driver) {
 	case "mysql":
-		return gorm.Open(mysql.Open(dsn), &gorm.Config{})
+		db, err = gorm.Open(mysql.Open(dsn), gormConfig)
 	case "pgsql":
-		return gorm.Open(postgres.Open(dsn), &gorm.Config{})
+		db, err = gorm.Open(postgres.Open(dsn), gormConfig)
 	case "mssql":
-		return gorm.Open(sqlserver.Open(dsn), &gorm.Config{})
+		db, err = gorm.Open(sqlserver.Open(dsn), gormConfig)
 	case "oracle":
-		return gorm.Open(oracle.Open(dsn), &gorm.Config{})
+		db, err = gorm.Open(oracle.Open(dsn), gormConfig)
 	case "sqlite":
-		return gorm.Open(sqlite.Open(dsn), &gorm.Config{})
+		db, err = gorm.Open(sqlite.Open(dsn), gormConfig)
 	default:
 		return nil, fmt.Errorf("unsupported database driver %q", driver)
 	}
+	if err != nil {
+		return nil, err
+	}
+	sqlDB, err := db.DB()
+	if err != nil {
+		return nil, err
+	}
+	if config.MaxIdleConns > 0 {
+		sqlDB.SetMaxIdleConns(int(config.MaxIdleConns))
+	}
+	if config.MaxOpenConns > 0 {
+		sqlDB.SetMaxOpenConns(int(config.MaxOpenConns))
+	}
+	if config.ConnMaxLifetime > 0 {
+		sqlDB.SetConnMaxLifetime(time.Duration(config.ConnMaxLifetime) * time.Second)
+	}
+	if config.Engine != "" && (normalizedDriver(driver) == "mysql" || normalizedDriver(driver) == "mssql") {
+		db = db.Set("gorm:table_options", "ENGINE="+config.Engine)
+	}
+	return db, nil
 }
 
 func openDatabase(c *conf.Data_Database, create bool, template string) (*gorm.DB, error) {
@@ -129,7 +172,7 @@ func openDatabase(c *conf.Data_Database, create bool, template string) (*gorm.DB
 		if err = os.MkdirAll(filepath.Dir(dsn), 0o755); err != nil {
 			return nil, err
 		}
-		return openWithDriver(driver, dsn)
+		return openWithDriverConfig(driver, dsn, c)
 	}
 	if create && driver != "oracle" {
 		if !databaseNamePattern.MatchString(c.Name) {
@@ -146,7 +189,7 @@ func openDatabase(c *conf.Data_Database, create bool, template string) (*gorm.DB
 		if err != nil {
 			return nil, err
 		}
-		adminDB, err := openWithDriver(driver, dsn)
+		adminDB, err := openWithDriverConfig(driver, dsn, c)
 		if err != nil {
 			return nil, fmt.Errorf("connect database server: %w", err)
 		}
@@ -182,7 +225,7 @@ func openDatabase(c *conf.Data_Database, create bool, template string) (*gorm.DB
 	if err != nil {
 		return nil, err
 	}
-	return openWithDriver(driver, dsn)
+	return openWithDriverConfig(driver, dsn, c)
 }
 
 func openFallbackDatabase() (*gorm.DB, error) {