Skip to content

Commit

Permalink
feat: migrator support type aliases (#5627)
Browse files Browse the repository at this point in the history
* feat: migrator support type aliases

* perf: check type
  • Loading branch information
a631807682 committed Sep 22, 2022
1 parent 101a7c7 commit 73bc53f
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 3 deletions.
1 change: 1 addition & 0 deletions migrator.go
Expand Up @@ -68,6 +68,7 @@ type Migrator interface {
// Database
CurrentDatabase() string
FullDataTypeOf(*schema.Field) clause.Expr
GetTypeAliases(databaseTypeName string) []string

// Tables
CreateTable(dst ...interface{}) error
Expand Down
29 changes: 26 additions & 3 deletions migrator/migrator.go
Expand Up @@ -408,9 +408,27 @@ func (m Migrator) MigrateColumn(value interface{}, field *schema.Field, columnTy

alterColumn := false

// check type
if !field.PrimaryKey && !strings.HasPrefix(fullDataType, realDataType) {
alterColumn = true
if !field.PrimaryKey {
// check type
var isSameType bool
if strings.HasPrefix(fullDataType, realDataType) {
isSameType = true
}

// check type aliases
if !isSameType {
aliases := m.DB.Migrator().GetTypeAliases(realDataType)
for _, alias := range aliases {
if strings.HasPrefix(fullDataType, alias) {
isSameType = true
break
}
}
}

if !isSameType {
alterColumn = true
}
}

// check size
Expand Down Expand Up @@ -863,3 +881,8 @@ func (m Migrator) CurrentTable(stmt *gorm.Statement) interface{} {
func (m Migrator) GetIndexes(dst interface{}) ([]gorm.Index, error) {
return nil, errors.New("not support")
}

// GetTypeAliases return database type aliases
func (m Migrator) GetTypeAliases(databaseTypeName string) []string {
return nil
}

0 comments on commit 73bc53f

Please sign in to comment.