auto create database

providers/database/mysql.go

@@ -4,10 +4,12 @@ import (
 	"atom/container"
 	"atom/providers/config"
 	"database/sql"
+	"fmt"
 	"log"
 
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
 )
 
 func init() {
@@ -17,14 +19,62 @@ func init() {
 }
 
 func NewDatabase(config *config.Config) (*gorm.DB, error) {
-	sqlDB, err := sql.Open("mysql", config.Database.MySQL.DSN())
+	mysqlConfig := mysql.Config{
+		DSN:                       config.Database.MySQL.DSN(), // DSN data source name
+		DefaultStringSize:         191,                         // string 类型字段的默认长度
+		SkipInitializeWithVersion: false,                       // 根据版本自动配置
+	}
+
+	gormConfig := gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   config.Database.MySQL.Prefix,
+			SingularTable: config.Database.MySQL.Singular,
+		},
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
+
+	// TODO: config logger
+	// _default := logger.New(NewWriter(log.New(os.Stdout, "\r\n", log.LstdFlags)), logger.Config{
+	// 	SlowThreshold: 200 * time.Millisecond,
+	// 	LogLevel:      logger.Warn,
+	// 	Colorful:      true,
+	// })
+	// config.Logger = _default.LogMode(logger.Warn)
+
+	db, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
 	if err != nil {
 		return nil, err
 	}
 
-	gormDB, err := gorm.Open(mysql.New(mysql.Config{
-		Conn: sqlDB,
-	}), &gorm.Config{})
+	// config instance
+	db.InstanceSet("gorm:table_options", "ENGINE="+config.Database.MySQL.Engine)
 
-	return gormDB, err
+	sqlDB, _ := db.DB()
+	sqlDB.SetMaxIdleConns(config.Database.MySQL.MaxIdleConns)
+	sqlDB.SetMaxOpenConns(config.Database.MySQL.MaxOpenConns)
+
+	if err := createDatabase(config.Database.MySQL.EmptyDsn(), "mysql", config.Database.MySQL.CreateDatabaseSql()); err != nil {
+		return nil, err
+	}
+
+	return db, err
+}
+
+// createDatabase 创建数据库（ EnsureDB() 中调用 ）
+func createDatabase(dsn string, driver string, createSql string) error {
+	db, err := sql.Open(driver, dsn)
+	if err != nil {
+		return err
+	}
+	defer func(db *sql.DB) {
+		err = db.Close()
+		if err != nil {
+			fmt.Println(err)
+		}
+	}(db)
+	if err = db.Ping(); err != nil {
+		return err
+	}
+	_, err = db.Exec(createSql)
+	return err
 }