tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add AddColumn method

Browse Source
This commit is contained in:
Maxim Lanin 2017-02-16 18:58:35 +03:00
parent 5050a58b45
commit 3b699fdf3e
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.go
View File

@ -552,6 +552,13 @@ func (s *DB) AutoMigrate(values ...interface{}) *DB {
return db return db
} }
// AddColumn add column with type
func (s *DB) AddColumn(column string, typ string) *DB {
scope := s.clone().NewScope(s.Value)
scope.addColumn(column, typ)
return scope.db
}
// ModifyColumn modify column to type // ModifyColumn modify column to type
func (s *DB) ModifyColumn(column string, typ string) *DB { func (s *DB) ModifyColumn(column string, typ string) *DB {
scope := s.clone().NewScope(s.Value) scope := s.clone().NewScope(s.Value)

4
scope.go
View File

@ -1124,6 +1124,10 @@ func (scope *Scope) dropTable() *Scope {
return scope return scope
} }
func (scope *Scope) addColumn(column string, typ string) {
scope.Raw(fmt.Sprintf("ALTER TABLE %v ADD %v %v", scope.QuotedTableName(), scope.Quote(column), typ)).Exec()
}
func (scope *Scope) modifyColumn(column string, typ string) { func (scope *Scope) modifyColumn(column string, typ string) {
scope.Raw(fmt.Sprintf("ALTER TABLE %v MODIFY %v %v", scope.QuotedTableName(), scope.Quote(column), typ)).Exec() scope.Raw(fmt.Sprintf("ALTER TABLE %v MODIFY %v %v", scope.QuotedTableName(), scope.Quote(column), typ)).Exec()
} }