diff --git a/internal/databases.go b/internal/databases.go index 97f3be6..aee5bb1 100644 --- a/internal/databases.go +++ b/internal/databases.go @@ -15,7 +15,9 @@ func DropUnusedColumns(DB *gorm.DB, values ...interface{}) { for _, dst := range values { stmt := &gorm.Statement{DB: DB} - stmt.Parse(dst) + if stmt.Parse(dst) != nil { + return + } fields := stmt.Schema.Fields columns, _ := DB.Debug().Migrator().ColumnTypes(dst) @@ -28,7 +30,9 @@ func DropUnusedColumns(DB *gorm.DB, values ...interface{}) { } } if !found { - DB.Migrator().DropColumn(dst, columns[i].Name()) + if DB.Migrator().DropColumn(dst, columns[i].Name()) != nil { + return + } } } }