From e3d8312b075de213d3db89f97c1015598086846b Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Sat, 7 Mar 2020 17:40:56 +0800 Subject: [PATCH] Fix bug --- dialects/mysql.go | 2 +- dialects/postgres.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dialects/mysql.go b/dialects/mysql.go index 21347c94..3c8d3c2a 100644 --- a/dialects/mysql.go +++ b/dialects/mysql.go @@ -288,7 +288,7 @@ func (db *mysql) IndexCheckSQL(tableName, idxName string) (string, []interface{} 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`=?" - return db.HasRecords(ctx, sql) + return db.HasRecords(ctx, sql, db.uri.DBName, tableName) } func (db *mysql) AddColumnSQL(tableName string, col *schemas.Column) string { diff --git a/dialects/postgres.go b/dialects/postgres.go index 2ed219cf..69100627 100644 --- a/dialects/postgres.go +++ b/dialects/postgres.go @@ -944,10 +944,10 @@ func (db *postgres) IndexCheckSQL(tableName, idxName string) (string, []interfac func (db *postgres) IsTableExist(ctx context.Context, tableName string) (bool, error) { 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) }