add MustDropTable method for dialect

This commit is contained in:
Lunny Xiao 2015-03-03 15:06:48 +08:00
parent 8d883536e3
commit 076f7bbf08
1 changed files with 7 additions and 6 deletions

View File

@ -56,21 +56,17 @@ type Dialect interface {
IsColumnExist(tableName string, col *Column) (bool, error) IsColumnExist(tableName string, col *Column) (bool, error)
CreateTableSql(table *Table, tableName, storeEngine, charset string) string CreateTableSql(table *Table, tableName, storeEngine, charset string) string
DropTableSql(tableName string) string //DropTableSql(tableName string) string
CreateIndexSql(tableName string, index *Index) string CreateIndexSql(tableName string, index *Index) string
DropIndexSql(tableName string, index *Index) string DropIndexSql(tableName string, index *Index) string
ModifyColumnSql(tableName string, col *Column) string ModifyColumnSql(tableName string, col *Column) string
MustDropTable(tableName string) error
GetColumns(tableName string) ([]string, map[string]*Column, error) GetColumns(tableName string) ([]string, map[string]*Column, error)
GetTables() ([]*Table, error) GetTables() ([]*Table, error)
GetIndexes(tableName string) (map[string]*Index, error) GetIndexes(tableName string) (map[string]*Index, error)
// Get data from db cell to a struct's field
//GetData(col *Column, fieldValue *reflect.Value, cellData interface{}) error
// Set field data to db
//SetData(col *Column, fieldValue *refelct.Value) (interface{}, error)
Filters() []Filter Filters() []Filter
} }
@ -145,6 +141,11 @@ func (db *Base) DropTableSql(tableName string) string {
return fmt.Sprintf("DROP TABLE IF EXISTS `%s`", tableName) return fmt.Sprintf("DROP TABLE IF EXISTS `%s`", tableName)
} }
func (db *Base) MustDropTable(tableName string) error {
_, err := db.db.Exec(db.DropTableSql(tableName))
return err
}
func (db *Base) HasRecords(query string, args ...interface{}) (bool, error) { func (db *Base) HasRecords(query string, args ...interface{}) (bool, error) {
rows, err := db.DB().Query(query, args...) rows, err := db.DB().Query(query, args...)
if db.Logger != nil { if db.Logger != nil {