调整数据库迁移方案

dbConfig.go

@@ -2,6 +2,7 @@ package data
 
 import (
 	"fmt"
+
 	"github.com/farseer-go/fs/container"
 	"gorm.io/gorm"
 )
@@ -14,6 +15,8 @@ type dbConfig struct {
 	PoolMinSize      int
 	ConnectionString string
 	databaseName     string // 数据库名称
+	Migrate          string // code first
+	migrated         bool   // 是否包含自动创建数据库
 }
 
 // GetDriver 获取对应驱动

internalContext.go

@@ -36,8 +36,8 @@ type IGetInternalContext interface {
 type internalContext struct {
 	dbConfig       *dbConfig          // 数据库配置
 	IsolationLevel sql.IsolationLevel // 事务等级
-	dbName         string             // 库名
-	nameReplacer   *strings.Replacer  // 替换dbName、tableName
+	//dbName         string             // 库名
+	nameReplacer *strings.Replacer // 替换dbName、tableName
 }
 
 // RegisterInternalContext 注册内部上下文
@@ -56,6 +56,7 @@ func RegisterInternalContext(key string, configString string) {
 	}
 	config.DataType = strings.ToLower(config.DataType)
 	config.keyName = key
+	config.migrated = strings.Contains(configString, "Migrate=")
 
 	// 获取数据库名称
 	switch config.DataType {
@@ -96,7 +97,7 @@ func RegisterInternalContext(key string, configString string) {
 
 	// 注册上下文
 	ins := &internalContext{dbConfig: &config}
-	ins.dbName = config.databaseName
+	//ins.dbName = config.databaseName
 	ins.nameReplacer = strings.NewReplacer("{database}", config.databaseName)
 	container.RegisterInstance[core.ITransaction](ins, key)
 

tableSet.go

@@ -59,9 +59,7 @@ func (receiver *TableSet[Table]) Init(dbContext *internalContext, param map[stri
 		//_ = db.Statement.Parse(t) // db.Statement.Model
 		//tableName := db.Statement.Schema.Table
 		tableName := reflect.TypeOf(t).Name()
-		if strings.HasSuffix(tableName, "PO") {
-			tableName = tableName[:len(tableName)-2]
-		}
+		tableName = strings.TrimSuffix(tableName, "PO")
 		tableName = schema.NamingStrategy{IdentifierMaxLength: 64}.ColumnName("", tableName)
 		//tableName = snakeString(tableName)
 		receiver.tableName = tableName
@@ -73,8 +71,12 @@ func (receiver *TableSet[Table]) Init(dbContext *internalContext, param map[stri
 	ts.dbName = receiver.dbName
 	ts.nameReplacer = receiver.nameReplacer
 
-	// 自动创建表
-	if migrate, exists := param["migrate"]; exists {
+	migrate, exists := param["migrate"]
+	if !exists {
+		exists = receiver.dbContext.dbConfig.migrated
+		migrate = receiver.dbContext.dbConfig.Migrate
+	}
+	if exists {
 		// 创建表
 		ts.CreateTable(migrate)
 		// 创建索引