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…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user
	 Cr
						Cr