HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improved IsColumnExist

This commit is contained in:
Lunny Xiao 2014-05-09 12:51:11 +08:00
parent adc62c0bd4
commit 8904d06558
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dialect.go
View File

@ -141,10 +141,7 @@ func (db *Base) DropTableSql(tableName string) string {
return fmt.Sprintf("DROP TABLE IF EXISTS `%s`", tableName)
}
func (db *Base) IsColumnExist(tableName string, col *Column) (bool, error) {
args := []interface{}{db.DbName, tableName, col.Name}
query := "SELECT `COLUMN_NAME` FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = ? AND `TABLE_NAME` = ? AND `COLUMN_NAME` = ?"
func (db *Base) HasRecords(query string, args ...interface{}) (bool, error) {
rows, err := db.DB().Query(query, args...)
if err != nil {
return false, err
@ -157,6 +154,12 @@ func (db *Base) IsColumnExist(tableName string, col *Column) (bool, error) {
return false, nil
}
func (db *Base) IsColumnExist(tableName string, col *Column) (bool, error) {
query := "SELECT `COLUMN_NAME` FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = ? AND `TABLE_NAME` = ? AND `COLUMN_NAME` = ?"
query = strings.Replace(query, "`", db.dialect.QuoteStr(), -1)
return db.HasRecords(query, db.DbName, tableName, col.Name)
}
func (db *Base) CreateIndexSql(tableName string, index *Index) string {
quote := db.Quote
var unique string