diff --git a/database/database.go b/database/database.go index 94b052b..b819e4a 100644 --- a/database/database.go +++ b/database/database.go @@ -21,10 +21,10 @@ func EscapeColumn(dbType, col string) string { func GetNamedParameter(dbType, col string, number int) string { if dbType == "mysql" { - return fmt.Sprintf("%s=?", EscapeColumn(col)) + return fmt.Sprintf("%s=?", EscapeColumn(dbType, col)) } - return fmt.Sprintf("%s=$%d", EscapeColumn(col), number) + return fmt.Sprintf("%s=$%d", EscapeColumn(dbType, col), number) } func IsPgNumberType(value string) bool { diff --git a/database/database_test.go b/database/database_test.go index 343d675..81b9f67 100644 --- a/database/database_test.go +++ b/database/database_test.go @@ -19,7 +19,7 @@ func TestEscapeColumn(t *testing.T) { t.Fatalf("TestEscapeColumn: mysql check failed") } - if EscapeTable("postgres", "foo") != "\"foo\"" { + if EscapeColumn("postgres", "foo") != "\"foo\"" { t.Fatalf("TestEscapeColumn: postgres check failed") } }