restructure

providers/database/mysql/config.go

@@ -0,0 +1,50 @@
+package mysql
+
+import (
+	"fmt"
+)
+
+// MySQL database config
+type Config struct {
+	Host         string
+	Port         uint
+	Database     string
+	Username     string
+	Password     string
+	Prefix       string // 表前缀
+	Singular     bool   // 是否开启全局禁用复数，true表示开启
+	MaxIdleConns int    // 空闲中的最大连接数
+	MaxOpenConns int    // 打开到数据库的最大连接数
+	Engine       string // 数据库引擎，默认InnoDB
+}
+
+func (m *Config) CreateDatabaseSql() string {
+	return fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;", m.Database)
+}
+
+func (m *Config) EmptyDsn() string {
+	dsnTpl := "%s@tcp(%s:%d)/"
+
+	authString := func() string {
+		if len(m.Password) > 0 {
+			return m.Username + ":" + m.Password
+		}
+		return m.Username
+	}
+
+	return fmt.Sprintf(dsnTpl, authString(), m.Host, m.Port)
+}
+
+// DSN connection dsn
+func (m *Config) DSN() string {
+	dsnTpl := "%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local"
+
+	authString := func() string {
+		if len(m.Password) > 0 {
+			return m.Username + ":" + m.Password
+		}
+		return m.Username
+	}
+
+	return fmt.Sprintf(dsnTpl, authString(), m.Host, m.Port, m.Database)
+}

providers/database/mysql/mysql.go

@@ -0,0 +1,77 @@
+package mysql
+
+import (
+	"database/sql"
+
+	"github.com/rogeecn/atom/container"
+	"github.com/rogeecn/atom/providers/log"
+	"go.uber.org/dig"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
+)
+
+func Provide(conf *Config, opts ...dig.ProvideOption) error {
+	return container.Container.Provide(func() (*gorm.DB, error) {
+		if err := createMySQLDatabase(conf.EmptyDsn(), "mysql", conf.CreateDatabaseSql()); err != nil {
+			return nil, err
+		}
+
+		mysqlConfig := mysql.Config{
+			DSN:                       conf.DSN(), // DSN data source name
+			DefaultStringSize:         191,        // string 类型字段的默认长度
+			SkipInitializeWithVersion: false,      // 根据版本自动配置
+		}
+
+		gormConfig := gorm.Config{
+			NamingStrategy: schema.NamingStrategy{
+				TablePrefix:   conf.Prefix,
+				SingularTable: conf.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,
+		// })
+		// conf.Logger = _default.LogMode(logger.Warn)
+
+		db, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
+		if err != nil {
+			return nil, err
+		}
+
+		// config instance
+		db.InstanceSet("gorm:table_options", "ENGINE="+conf.Engine)
+
+		sqlDB, _ := db.DB()
+		sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
+		sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
+
+		return db, err
+	}, opts...)
+}
+
+// createDatabase 创建数据库
+func createMySQLDatabase(dsn, driver, 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 {
+			log.Error(err)
+		}
+	}(db)
+	err = db.Ping()
+	if err != nil {
+		return err
+	}
+	_, err = db.Exec(createSql)
+	return err
+}