golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-04-30 06:30:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/sqlgraph: fix postgres lastinsert scanning for non-integer types (#987)

Browse Source 
Fixed #985
This commit is contained in:
Ariel Mashraki
2020-11-26 12:12:02 +02:00
committed by GitHub
parent 550081ea33
commit ac8e8dac83
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dialect/sql/sqlgraph/graph.go
View File

@@ -1180,7 +1180,7 @@ func insertLastID(ctx context.Context, tx dialect.ExecQuerier, insert *sql.Inser
}
// insertLastIDs invokes the batch insert query on the transaction and returns the LastInsertID of all entities.
func insertLastIDs(ctx context.Context, tx dialect.ExecQuerier, insert *sql.InsertBuilder) (ids []int64, err error) {
func insertLastIDs(ctx context.Context, tx dialect.ExecQuerier, insert *sql.InsertBuilder) (ids []driver.Value, err error) {
query, args := insert.Query()
// PostgreSQL does not support the LastInsertId() method of sql.Result
// on Exec, and should be extracted manually using the `RETURNING` clause.
@@ -1205,7 +1205,7 @@ func insertLastIDs(ctx context.Context, tx dialect.ExecQuerier, insert *sql.Inse
if err != nil {
return nil, err
}
ids = make([]int64, 0, affected)
ids = make([]driver.Value, 0, affected)
switch insert.Dialect() {
case dialect.SQLite:
id -= affected - 1