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

Fix bug

This commit is contained in:
Lunny Xiao 2020-03-07 17:40:56 +08:00
parent 284443565d
commit e3d8312b07
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dialects/mysql.go
View File

@ -288,7 +288,7 @@ func (db *mysql) IndexCheckSQL(tableName, idxName string) (string, []interface{}
func (db *mysql) IsTableExist(ctx context.Context, tableName string) (bool, error) { func (db *mysql) IsTableExist(ctx context.Context, tableName string) (bool, error) {
sql := "SELECT `TABLE_NAME` from `INFORMATION_SCHEMA`.`TABLES` WHERE `TABLE_SCHEMA`=? and `TABLE_NAME`=?" sql := "SELECT `TABLE_NAME` from `INFORMATION_SCHEMA`.`TABLES` WHERE `TABLE_SCHEMA`=? and `TABLE_NAME`=?"
return db.HasRecords(ctx, sql) return db.HasRecords(ctx, sql, db.uri.DBName, tableName)
} }
func (db *mysql) AddColumnSQL(tableName string, col *schemas.Column) string { func (db *mysql) AddColumnSQL(tableName string, col *schemas.Column) string {

4
dialects/postgres.go
View File

@ -944,10 +944,10 @@ func (db *postgres) IndexCheckSQL(tableName, idxName string) (string, []interfac
func (db *postgres) IsTableExist(ctx context.Context, tableName string) (bool, error) { func (db *postgres) IsTableExist(ctx context.Context, tableName string) (bool, error) {
if len(db.uri.Schema) == 0 { if len(db.uri.Schema) == 0 {
return db.HasRecords(ctx, `SELECT tablename FROM pg_tables WHERE tablename = ?`, tableName) return db.HasRecords(ctx, `SELECT tablename FROM pg_tables WHERE tablename = $1`, tableName)
} }
return db.HasRecords(ctx, `SELECT tablename FROM pg_tables WHERE schemaname = ? AND tablename = ?`, return db.HasRecords(ctx, `SELECT tablename FROM pg_tables WHERE schemaname = $1 AND tablename = $2`,
db.uri.Schema, tableName) db.uri.Schema, tableName)
} }