Add double quote around db name

Signed-off-by: Masudur Rahman <masudjuly02@gmail.com>
masudur-rahman · Mar 2, 2024 · a15a9ce · a15a9ce
1 parent 0259c9d
commit a15a9ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sql/postgres/lib/table_sync.go b/sql/postgres/lib/table_sync.go
@@ -227,7 +227,7 @@ func createTableQuery(tableName string, fields []fieldInfo) string {
 }
 
 func generateAddColumnQuery(tableName string, missingColumns []string) string {
-	alterQuery := fmt.Sprintf("ALTER TABLE %s ", tableName)
+	alterQuery := fmt.Sprintf("ALTER TABLE \"%s\" ", tableName)
 	var addColumns []string
 	for _, col := range missingColumns {
 		addColumns = append(addColumns, fmt.Sprintf("ADD COLUMN %s", col))
@@ -312,7 +312,7 @@ func getUniqueConstraints(ctx context.Context, conn *sql.Conn, tableName string)
 }
 
 func generateDropConstraintStatement(tableName string, uqConstraints [][]string) string {
-	sql := fmt.Sprintf("ALTER TABLE %s ", tableName)
+	sql := fmt.Sprintf("ALTER TABLE \"%s\" ", tableName)
 
 	var dropConstraints []string
 	for i := range uqConstraints {
@@ -329,7 +329,7 @@ func generateDropConstraintStatement(tableName string, uqConstraints [][]string)
 func generateAddConstraintStatement(tableName string,
 	uqGroups [][]string) string {
 
-	sql := fmt.Sprintf("ALTER TABLE %s ", tableName)
+	sql := fmt.Sprintf("ALTER TABLE \"%s\" ", tableName)
 
 	var addConstraints []string
 	for i, group := range uqGroups {

diff --git a/sql/sqlite/lib/table_sync.go b/sql/sqlite/lib/table_sync.go
@@ -352,7 +352,7 @@ func tableExists(ctx context.Context, conn *sql.Conn, tableName string) (bool, e
 }
 
 func generateAddColumnQuery(tableName string, missingColumns []string) string {
-	alterQuery := fmt.Sprintf("ALTER TABLE %s ", tableName)
+	alterQuery := fmt.Sprintf("ALTER TABLE \"%s\" ", tableName)
 	var addColumns []string
 	for _, col := range missingColumns {
 		addColumns = append(addColumns, fmt.Sprintf("ADD COLUMN %s", col))