re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure identifiers are quoted for `RETURNING` and `DROP INDEX` statements.

This commit is contained in:
Jay Taylor 2015-01-13 15:27:35 -08:00
parent 016e91db20
commit 2166649409
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
postgres.go
View File

@ -66,7 +66,7 @@ func (s *postgres) PrimaryKeyTag(value reflect.Value, size int) string {
} }
func (s *postgres) ReturningStr(tableName, key string) string { func (s *postgres) ReturningStr(tableName, key string) string {
return fmt.Sprintf("RETURNING \"%v\".%v", tableName, key) return fmt.Sprintf("RETURNING %v.%v", s.Quote(tableName), s.Quote(key))
} }
func (s *postgres) SelectFromDummyTable() string { func (s *postgres) SelectFromDummyTable() string {
@ -97,7 +97,7 @@ func (s *postgres) HasColumn(scope *Scope, tableName string, columnName string)
} }
func (s *postgres) RemoveIndex(scope *Scope, indexName string) { func (s *postgres) RemoveIndex(scope *Scope, indexName string) {
scope.Raw(fmt.Sprintf("DROP INDEX %v", indexName)).Exec() scope.Raw(fmt.Sprintf("DROP INDEX %v", s.Quote(indexName))).Exec()
} }
var hstoreType = reflect.TypeOf(Hstore{}) var hstoreType = reflect.TypeOf(Hstore{})