From 38a06bfdaae5dde7519faad908373493cec20211 Mon Sep 17 00:00:00 2001
From: Jinzhu <wosmvp@gmail.com>
Date: Mon, 19 May 2014 12:08:31 +0800
Subject: [PATCH] quote table name

---
 callback_create.go |  4 ++--
 callback_delete.go |  4 ++--
 callback_update.go |  2 +-
 scope_private.go   | 16 ++++++++--------
 4 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/callback_create.go b/callback_create.go
index 2f641b3d..7ad40f4b 100644
--- a/callback_create.go
+++ b/callback_create.go
@@ -35,13 +35,13 @@ func Create(scope *Scope) {
 
 		if len(columns) == 0 {
 			scope.Raw(fmt.Sprintf("INSERT INTO %v DEFAULT VALUES %v",
-				scope.TableName(),
+				scope.Quote(scope.TableName()),
 				scope.Dialect().ReturningStr(scope.PrimaryKey()),
 			))
 		} else {
 			scope.Raw(fmt.Sprintf(
 				"INSERT INTO %v (%v) VALUES (%v) %v",
-				scope.TableName(),
+				scope.Quote(scope.TableName()),
 				strings.Join(columns, ","),
 				strings.Join(sqls, ","),
 				scope.Dialect().ReturningStr(scope.PrimaryKey()),
diff --git a/callback_delete.go b/callback_delete.go
index 3bf5b7b0..9539b4ca 100644
--- a/callback_delete.go
+++ b/callback_delete.go
@@ -14,12 +14,12 @@ func Delete(scope *Scope) {
 		if !scope.Search.Unscope && scope.HasColumn("DeletedAt") {
 			scope.Raw(
 				fmt.Sprintf("UPDATE %v SET deleted_at=%v %v",
-					scope.TableName(),
+					scope.Quote(scope.TableName()),
 					scope.AddToVars(time.Now()),
 					scope.CombinedConditionSql(),
 				))
 		} else {
-			scope.Raw(fmt.Sprintf("DELETE FROM %v %v", scope.TableName(), scope.CombinedConditionSql()))
+			scope.Raw(fmt.Sprintf("DELETE FROM %v %v", scope.Quote(scope.TableName()), scope.CombinedConditionSql()))
 		}
 
 		scope.Exec()
diff --git a/callback_update.go b/callback_update.go
index 095bbf48..138953a0 100644
--- a/callback_update.go
+++ b/callback_update.go
@@ -59,7 +59,7 @@ func Update(scope *Scope) {
 
 		scope.Raw(fmt.Sprintf(
 			"UPDATE %v SET %v %v",
-			scope.TableName(),
+			scope.Quote(scope.TableName()),
 			strings.Join(sqls, ", "),
 			scope.CombinedConditionSql(),
 		))
diff --git a/scope_private.go b/scope_private.go
index 67f08bda..2b39dddb 100644
--- a/scope_private.go
+++ b/scope_private.go
@@ -237,7 +237,7 @@ func (scope *Scope) prepareQuerySql() {
 	if scope.Search.Raw {
 		scope.Raw(strings.TrimLeft(scope.CombinedConditionSql(), "WHERE "))
 	} else {
-		scope.Raw(fmt.Sprintf("SELECT %v FROM %v %v", scope.selectSql(), scope.TableName(), scope.CombinedConditionSql()))
+		scope.Raw(fmt.Sprintf("SELECT %v FROM %v %v", scope.selectSql(), scope.Quote(scope.TableName()), scope.CombinedConditionSql()))
 	}
 	return
 }
@@ -414,21 +414,21 @@ func (scope *Scope) createTable() *Scope {
 			sqls = append(sqls, scope.Quote(field.DBName)+" "+field.SqlTag)
 		}
 	}
-	scope.Raw(fmt.Sprintf("CREATE TABLE %v (%v)", scope.TableName(), strings.Join(sqls, ","))).Exec()
+	scope.Raw(fmt.Sprintf("CREATE TABLE %v (%v)", scope.Quote(scope.TableName()), strings.Join(sqls, ","))).Exec()
 	return scope
 }
 
 func (scope *Scope) dropTable() *Scope {
-	scope.Raw(fmt.Sprintf("DROP TABLE %v", scope.TableName())).Exec()
+	scope.Raw(fmt.Sprintf("DROP TABLE %v", scope.Quote(scope.TableName()))).Exec()
 	return scope
 }
 
 func (scope *Scope) modifyColumn(column string, typ string) {
-	scope.Raw(fmt.Sprintf("ALTER TABLE %v MODIFY %v %v", scope.TableName(), scope.Quote(column), typ)).Exec()
+	scope.Raw(fmt.Sprintf("ALTER TABLE %v MODIFY %v %v", scope.Quote(scope.TableName()), scope.Quote(column), typ)).Exec()
 }
 
 func (scope *Scope) dropColumn(column string) {
-	scope.Raw(fmt.Sprintf("ALTER TABLE %v DROP COLUMN %v", scope.TableName(), scope.Quote(column))).Exec()
+	scope.Raw(fmt.Sprintf("ALTER TABLE %v DROP COLUMN %v", scope.Quote(scope.TableName()), scope.Quote(column))).Exec()
 }
 
 func (scope *Scope) addIndex(column string, names ...string) {
@@ -439,11 +439,11 @@ func (scope *Scope) addIndex(column string, names ...string) {
 		indexName = fmt.Sprintf("index_%v_on_%v", scope.TableName(), column)
 	}
 
-	scope.Raw(fmt.Sprintf("CREATE INDEX %v ON %v(%v);", indexName, scope.TableName(), scope.Quote(column))).Exec()
+	scope.Raw(fmt.Sprintf("CREATE INDEX %v ON %v(%v);", indexName, scope.Quote(scope.TableName()), scope.Quote(column))).Exec()
 }
 
 func (scope *Scope) removeIndex(indexName string) {
-	scope.Raw(fmt.Sprintf("DROP INDEX %v ON %v", indexName, scope.TableName())).Exec()
+	scope.Raw(fmt.Sprintf("DROP INDEX %v ON %v", indexName, scope.Quote(scope.TableName()))).Exec()
 }
 
 func (scope *Scope) autoMigrate() *Scope {
@@ -453,7 +453,7 @@ func (scope *Scope) autoMigrate() *Scope {
 		for _, field := range scope.Fields() {
 			if !scope.Dialect().HasColumn(scope, scope.TableName(), field.DBName) {
 				if len(field.SqlTag) > 0 && !field.IsIgnored {
-					scope.Raw(fmt.Sprintf("ALTER TABLE %v ADD %v %v;", scope.TableName(), field.DBName, field.SqlTag)).Exec()
+					scope.Raw(fmt.Sprintf("ALTER TABLE %v ADD %v %v;", scope.Quote(scope.TableName()), field.DBName, field.SqlTag)).Exec()
 				}
 			}
 		}