mirror of https://github.com/go-gorm/gorm.git
fix(MigrateColumn):declared different type without length (#5465)
This commit is contained in:
parent
3e6ab99043
commit
235c093bb9
|
@ -15,7 +15,6 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
regRealDataType = regexp.MustCompile(`[^\d](\d+)[^\d]?`)
|
|
||||||
regFullDataType = regexp.MustCompile(`[^\d]*(\d+)[^\d]?`)
|
regFullDataType = regexp.MustCompile(`[^\d]*(\d+)[^\d]?`)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -404,11 +403,16 @@ func (m Migrator) RenameColumn(value interface{}, oldName, newName string) error
|
||||||
// MigrateColumn migrate column
|
// MigrateColumn migrate column
|
||||||
func (m Migrator) MigrateColumn(value interface{}, field *schema.Field, columnType gorm.ColumnType) error {
|
func (m Migrator) MigrateColumn(value interface{}, field *schema.Field, columnType gorm.ColumnType) error {
|
||||||
// found, smart migrate
|
// found, smart migrate
|
||||||
fullDataType := strings.ToLower(m.DB.Migrator().FullDataTypeOf(field).SQL)
|
fullDataType := strings.TrimSpace(strings.ToLower(m.DB.Migrator().FullDataTypeOf(field).SQL))
|
||||||
realDataType := strings.ToLower(columnType.DatabaseTypeName())
|
realDataType := strings.ToLower(columnType.DatabaseTypeName())
|
||||||
|
|
||||||
alterColumn := false
|
alterColumn := false
|
||||||
|
|
||||||
|
// check type
|
||||||
|
if !field.PrimaryKey && !strings.HasPrefix(fullDataType, realDataType) {
|
||||||
|
alterColumn = true
|
||||||
|
}
|
||||||
|
|
||||||
// check size
|
// check size
|
||||||
if length, ok := columnType.Length(); length != int64(field.Size) {
|
if length, ok := columnType.Length(); length != int64(field.Size) {
|
||||||
if length > 0 && field.Size > 0 {
|
if length > 0 && field.Size > 0 {
|
||||||
|
@ -416,9 +420,8 @@ func (m Migrator) MigrateColumn(value interface{}, field *schema.Field, columnTy
|
||||||
} else {
|
} else {
|
||||||
// has size in data type and not equal
|
// has size in data type and not equal
|
||||||
// Since the following code is frequently called in the for loop, reg optimization is needed here
|
// Since the following code is frequently called in the for loop, reg optimization is needed here
|
||||||
matches := regRealDataType.FindAllStringSubmatch(realDataType, -1)
|
|
||||||
matches2 := regFullDataType.FindAllStringSubmatch(fullDataType, -1)
|
matches2 := regFullDataType.FindAllStringSubmatch(fullDataType, -1)
|
||||||
if (len(matches) == 1 && matches[0][1] != fmt.Sprint(field.Size) || !field.PrimaryKey) &&
|
if !field.PrimaryKey &&
|
||||||
(len(matches2) == 1 && matches2[0][1] != fmt.Sprint(length) && ok) {
|
(len(matches2) == 1 && matches2[0][1] != fmt.Sprint(length) && ok) {
|
||||||
alterColumn = true
|
alterColumn = true
|
||||||
}
|
}
|
||||||
|
|
|
@ -884,3 +884,42 @@ func TestInvalidCachedPlan(t *testing.T) {
|
||||||
t.Errorf("AutoMigrate err:%v", err)
|
t.Errorf("AutoMigrate err:%v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDifferentTypeWithoutDeclaredLength(t *testing.T) {
|
||||||
|
type DiffType struct {
|
||||||
|
ID uint
|
||||||
|
Name string `gorm:"type:varchar(20)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DiffType1 struct {
|
||||||
|
ID uint
|
||||||
|
Name string `gorm:"type:text"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
DB.Migrator().DropTable(&DiffType{})
|
||||||
|
|
||||||
|
err = DB.AutoMigrate(&DiffType{})
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("AutoMigrate err:%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ct, err := findColumnType(&DiffType{}, "name")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("findColumnType err:%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
AssertEqual(t, "varchar", strings.ToLower(ct.DatabaseTypeName()))
|
||||||
|
|
||||||
|
err = DB.Table("diff_types").AutoMigrate(&DiffType1{})
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("AutoMigrate err:%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ct, err = findColumnType(&DiffType{}, "name")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("findColumnType err:%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
AssertEqual(t, "text", strings.ToLower(ct.DatabaseTypeName()))
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue