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
|
// Parse get data type from dialector
|
||||||
func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error) {
|
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 {
|
if dest == nil {
|
||||||
return nil, fmt.Errorf("%w: %+v", ErrUnsupportedDataType, dest)
|
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)
|
modelValue := reflect.New(modelType)
|
||||||
tableName := namer.TableName(modelType.Name())
|
tableName := namer.TableName(modelType.Name())
|
||||||
|
if schemaTable != "" {
|
||||||
|
tableName = schemaTable
|
||||||
|
}
|
||||||
if tabler, ok := modelValue.Interface().(Tabler); ok {
|
if tabler, ok := modelValue.Interface().(Tabler); ok {
|
||||||
tableName = tabler.TableName()
|
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 {
|
if v, loaded := cacheStore.LoadOrStore(modelType, schema); loaded {
|
||||||
s := v.(*Schema)
|
s := v.(*Schema)
|
||||||
// Wait for the initialization of other goroutines to complete
|
// Wait for the initialization of other goroutines to complete
|
||||||
<-s.initialized
|
<-s.initialized
|
||||||
return s, s.err
|
return s, s.err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
if schema.err != nil {
|
if schema.err != nil {
|
||||||
|
|
|
@ -456,7 +456,7 @@ func (stmt *Statement) Build(clauses ...string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (stmt *Statement) Parse(value interface{}) (err error) {
|
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 {
|
if tables := strings.Split(stmt.Schema.Table, "."); len(tables) == 2 {
|
||||||
stmt.TableExpr = &clause.Expr{SQL: stmt.Quote(stmt.Schema.Table)}
|
stmt.TableExpr = &clause.Expr{SQL: stmt.Quote(stmt.Schema.Table)}
|
||||||
stmt.Table = tables[1]
|
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