自动创建数据库

service/syncdb.go

@@ -1,13 +1,22 @@
 package service
 
 import (
+	"database/sql"
+	"fmt"
 	"moredoc/conf"
 	"moredoc/model"
 
+	"github.com/go-sql-driver/mysql"
 	"go.uber.org/zap"
 )
 
 func SyncDB(cfg *conf.Config, logger *zap.Logger) {
+	err := checkAndCreateDatabase(cfg.Database.DSN, logger)
+	if err != nil {
+		logger.Fatal("checkAndCreateDatabase", zap.Error(err))
+		return
+	}
+
 	lg := logger.Named("syncdb")
 	lg.Info("start syncdb")
 	dbModel, err := model.NewDBModel(&cfg.Database, logger)
@@ -15,6 +24,7 @@ func SyncDB(cfg *conf.Config, logger *zap.Logger) {
 		lg.Fatal("NewDBModel", zap.Error(err))
 		return
 	}
+
 	err = dbModel.SyncDB()
 	if err != nil {
 		lg.Fatal("SyncDB", zap.Error(err))
@@ -22,3 +32,31 @@ func SyncDB(cfg *conf.Config, logger *zap.Logger) {
 	}
 	lg.Info("syncdb success")
 }
+
+func checkAndCreateDatabase(dsn string, loggger *zap.Logger) (err error) {
+	cfg, err := mysql.ParseDSN(dsn)
+	if err != nil {
+		loggger.Error("ParseDSN", zap.Error(err))
+		return
+	}
+
+	dbName := cfg.DBName
+	if dbName == "" {
+		loggger.Error("ParseDSN", zap.String("database", "数据库名称不能为空"))
+		return
+	}
+
+	conn := fmt.Sprintf("%s:%s@tcp(%s)/", cfg.User, cfg.Passwd, cfg.Addr)
+	db, err := sql.Open("mysql", conn)
+	if err != nil {
+		loggger.Error("sql.Open", zap.Error(err))
+		return
+	}
+
+	createDB := fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARSET utf8mb4 COLLATE utf8mb4_unicode_ci", cfg.DBName)
+	_, err = db.Exec(createDB)
+	if err != nil {
+		loggger.Error("db.Exec", zap.Error(err))
+	}
+	return
+}