tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added support to schema name in returning clause

Browse Source
This commit is contained in:
Marcelo Monaco 2015-09-07 17:47:15 -03:00
parent 447885d98f
commit 51213b0f69
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
postgres.go
View File

@ -5,6 +5,7 @@ import (
"database/sql/driver" "database/sql/driver"
"fmt" "fmt"
"reflect" "reflect"
"strings"
"time" "time"
"github.com/lib/pq/hstore" "github.com/lib/pq/hstore"
@ -60,7 +61,12 @@ func (postgres) SqlTag(value reflect.Value, size int, autoIncrease bool) string
} }
func (s postgres) ReturningStr(tableName, key string) string { func (s postgres) ReturningStr(tableName, key string) string {
return fmt.Sprintf("RETURNING %v.%v", s.Quote(tableName), key) fullName := strings.Split(tableName,".")
if len(fullName) == 1 {
return fmt.Sprintf("RETURNING %v.%v", s.Quote(tableName), key)
} else {
return fmt.Sprintf("RETURNING %v.%v.%v", s.Quote(fullName[0]), s.Quote(fullName[1]), key)
}
} }
func (s postgres) HasTable(scope *Scope, tableName string) bool { func (s postgres) HasTable(scope *Scope, tableName string) bool {