Skip to content

Commit

Permalink
fix: backtick only works with mysql
Browse files Browse the repository at this point in the history
  • Loading branch information
aldy505 committed Jun 30, 2021
1 parent 03e6bdb commit 8d6de64
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
10 changes: 5 additions & 5 deletions create.go
Original file line number Diff line number Diff line change
Expand Up @@ -78,15 +78,15 @@ func (d *createData) ToSQL() (sqlStr string, args []interface{}, err error) {
sql.WriteString("CREATE TABLE ")

if d.Schema != "" {
sql.WriteString("`" + d.Schema + "`.")
sql.WriteString("\"" + d.Schema + "\".")
}

sql.WriteString("`" + d.TableName + "`")
sql.WriteString("\"" + d.TableName + "\"")
sql.WriteString(" ")

var columnTypes []string
for i := 0; i < len(d.Columns); i++ {
columnTypes = append(columnTypes, "`"+d.Columns[i]+"` "+d.Types[i])
columnTypes = append(columnTypes, "\""+d.Columns[i]+"\" "+d.Types[i])
}

sql.WriteString("(")
Expand All @@ -99,7 +99,7 @@ func (d *createData) ToSQL() (sqlStr string, args []interface{}, err error) {
return
}
sql.WriteString(" ")
sql.WriteString("ALTER TABLE `" + d.TableName + "` ADD PRIMARY KEY (`" + d.Primary + "`);")
sql.WriteString("ALTER TABLE \"" + d.TableName + "\" ADD PRIMARY KEY (\"" + d.Primary + "\");")
}

if d.Unique != "" {
Expand All @@ -108,7 +108,7 @@ func (d *createData) ToSQL() (sqlStr string, args []interface{}, err error) {
return
}
sql.WriteString(" ")
sql.WriteString("ALTER TABLE `" + d.TableName + "` ADD UNIQUE (`" + d.Unique + "`);")
sql.WriteString("ALTER TABLE \"" + d.TableName + "\" ADD UNIQUE (\"" + d.Unique + "\");")
}
sqlStr = sql.String()
return
Expand Down
6 changes: 3 additions & 3 deletions create_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ func TestCreate(t *testing.T) {
if err != nil {
t.Fatal(err.Error())
}
result := "CREATE TABLE `users` (`name` varchar(255), `password` text, `date` date);"
result := "CREATE TABLE \"users\" (\"name\" varchar(255), \"password\" text, \"date\" date);"
if sql != result {
t.Fatal("sql is not equal to result:", sql)
}
Expand All @@ -28,7 +28,7 @@ func TestCreate(t *testing.T) {
if err != nil {
t.Fatal(err.Error())
}
result := "CREATE TABLE `users` (`id` uuid, `name` varchar(255), `email` varchar(255), `password` text, `date` date); ALTER TABLE `users` ADD PRIMARY KEY (`id`); ALTER TABLE `users` ADD UNIQUE (`email`);"
result := "CREATE TABLE \"users\" (\"id\" uuid, \"name\" varchar(255), \"email\" varchar(255), \"password\" text, \"date\" date); ALTER TABLE \"users\" ADD PRIMARY KEY (\"id\"); ALTER TABLE \"users\" ADD UNIQUE (\"email\");"
if sql != result {
t.Fatal("sql is not equal to result:", sql)
}
Expand All @@ -39,7 +39,7 @@ func TestCreate(t *testing.T) {
if err != nil {
t.Fatal(err.Error())
}
result := "CREATE TABLE `private`.`users` (`name` varchar(255), `password` text, `date` date);"
result := "CREATE TABLE \"private\".\"users\" (\"name\" varchar(255), \"password\" text, \"date\" date);"
if sql != result {
t.Fatal("sql is not equal to result:", sql)
}
Expand Down

0 comments on commit 8d6de64

Please sign in to comment.