diff --git a/callback_create.go b/callback_create.go index 4734d65e..52e9369c 100644 --- a/callback_create.go +++ b/callback_create.go @@ -36,7 +36,7 @@ func Create(scope *Scope) { returningKey := "*" if scope.PrimaryKey() != "" { - returningKey = scope.PrimaryKey() + returningKey = scope.Quote(scope.PrimaryKey()) } if len(columns) == 0 { diff --git a/postgres.go b/postgres.go index 218d594d..2229e7b1 100644 --- a/postgres.go +++ b/postgres.go @@ -4,8 +4,9 @@ import ( "database/sql" "database/sql/driver" "fmt" - "github.com/lib/pq/hstore" "reflect" + + "github.com/lib/pq/hstore" ) type postgres struct { @@ -66,7 +67,7 @@ func (s *postgres) PrimaryKeyTag(value reflect.Value, size int) string { } func (s *postgres) ReturningStr(tableName, key string) string { - return fmt.Sprintf("RETURNING %v.%v", s.Quote(tableName), s.Quote(key)) + return fmt.Sprintf("RETURNING %v.%v", s.Quote(tableName), key) } func (s *postgres) SelectFromDummyTable() string {