forked from mirror/gorm
fix: automigrate error caused by indexes while using dynamic table name
This commit is contained in:
parent
9a5ba37604
commit
a3bd9c3ea2
|
@ -73,6 +73,15 @@ type Tabler interface {
|
|||
|
||||
// Parse get data type from dialector
|
||||
func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error) {
|
||||
return parse(dest, cacheStore, namer, "")
|
||||
}
|
||||
|
||||
// ParseWithSchemaTable get data type from dialector with extra schema table
|
||||
func ParseWithSchemaTable(dest interface{}, cacheStore *sync.Map, namer Namer, schemaTable string) (*Schema, error) {
|
||||
return parse(dest, cacheStore, namer, schemaTable)
|
||||
}
|
||||
|
||||
func parse(dest interface{}, cacheStore *sync.Map, namer Namer, schemaTable string) (*Schema, error) {
|
||||
if dest == nil {
|
||||
return nil, fmt.Errorf("%w: %+v", ErrUnsupportedDataType, dest)
|
||||
}
|
||||
|
@ -107,6 +116,9 @@ func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error)
|
|||
|
||||
modelValue := reflect.New(modelType)
|
||||
tableName := namer.TableName(modelType.Name())
|
||||
if schemaTable != "" {
|
||||
tableName = schemaTable
|
||||
}
|
||||
if tabler, ok := modelValue.Interface().(Tabler); ok {
|
||||
tableName = tabler.TableName()
|
||||
}
|
||||
|
@ -235,12 +247,14 @@ func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error)
|
|||
}
|
||||
}
|
||||
|
||||
if schemaTable == "" {
|
||||
if v, loaded := cacheStore.LoadOrStore(modelType, schema); loaded {
|
||||
s := v.(*Schema)
|
||||
// Wait for the initialization of other goroutines to complete
|
||||
<-s.initialized
|
||||
return s, s.err
|
||||
}
|
||||
}
|
||||
|
||||
defer func() {
|
||||
if schema.err != nil {
|
||||
|
|
|
@ -456,7 +456,7 @@ func (stmt *Statement) Build(clauses ...string) {
|
|||
}
|
||||
|
||||
func (stmt *Statement) Parse(value interface{}) (err error) {
|
||||
if stmt.Schema, err = schema.Parse(value, stmt.DB.cacheStore, stmt.DB.NamingStrategy); err == nil && stmt.Table == "" {
|
||||
if stmt.Schema, err = schema.ParseWithSchemaTable(value, stmt.DB.cacheStore, stmt.DB.NamingStrategy, stmt.DB.Statement.Table); err == nil && stmt.Table == "" {
|
||||
if tables := strings.Split(stmt.Schema.Table, "."); len(tables) == 2 {
|
||||
stmt.TableExpr = &clause.Expr{SQL: stmt.Quote(stmt.Schema.Table)}
|
||||
stmt.Table = tables[1]
|
||||
|
|
|
@ -381,3 +381,33 @@ func TestMigrateConstraint(t *testing.T) {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
type MigrateUser struct {
|
||||
gorm.Model
|
||||
Name string `gorm:"index"`
|
||||
}
|
||||
|
||||
// https://github.com/go-gorm/gorm/issues/4752
|
||||
func TestMigrateIndexesWithDynamicTableName(t *testing.T) {
|
||||
tableNameSuffixes := []string{"01", "02", "03"}
|
||||
for _, v := range tableNameSuffixes {
|
||||
tableName := "migrate_user_" + v
|
||||
m := DB.Scopes(func(db *gorm.DB) *gorm.DB {
|
||||
return db.Table(tableName)
|
||||
}).Migrator()
|
||||
|
||||
if err := m.AutoMigrate(&MigrateUser{}); err != nil {
|
||||
t.Fatalf("Failed to create table for %#v", tableName)
|
||||
}
|
||||
|
||||
if !m.HasTable(tableName) {
|
||||
t.Fatalf("Failed to create table for %#v", tableName)
|
||||
}
|
||||
if !m.HasIndex(&MigrateUser{}, "Name") {
|
||||
t.Fatalf("Should find index for %s's name after AutoMigrate", tableName)
|
||||
}
|
||||
if !m.HasIndex(&MigrateUser{}, "DeletedAt") {
|
||||
t.Fatalf("Should find index for %s's deleted_at after AutoMigrate", tableName)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue