fix(MigrateColumn):declared different type without length
This commit is contained in:
		
							parent
							
								
									a70af2a4c0
								
							
						
					
					
						commit
						3f5b0bc2f2
					
				@ -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…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user