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

dialect/sql/sqlgraph: allow setting stmt modifiers for create-many

Browse Source
This commit is contained in:
Ariel Mashraki
2021-07-29 16:14:46 +03:00
committed by Ariel Mashraki
parent b9fcbff724
commit 7acf2234b6
2 changed files with 84 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -362,6 +362,18 @@ type (
// multiple nodes in the graph.
BatchCreateSpec struct {
Nodes []*CreateSpec
// The Modifiers option allows providing custom functions to
// modify the INSERT statement of the node before it is executed.
//
// sqlgraph.BatchCreateSpec{
// Modifiers: []func(*sql.InsertBuilder) {
// OnConflictOptions(...),
// ReturningOptions(...),
// },
// }
//
Modifiers []func(*sql.InsertBuilder)
}
)
@@ -982,8 +994,11 @@ func (c *creator) insert(ctx context.Context, tx dialect.ExecQuerier, insert *sq
return nil
}
// batchInsert inserts a batch of nodes to their table and sets their ID if it wasn't provided by the user.
// batchInsert inserts a batch of nodes to their table and sets their ID if it was not provided by the user.
func (c *creator) batchInsert(ctx context.Context, tx dialect.ExecQuerier, insert *sql.InsertBuilder) error {
for _, m := range c.BatchCreateSpec.Modifiers {
m(insert)
}
ids, err := insertLastIDs(ctx, tx, insert.Returning(c.Nodes[0].ID.Column))
if err != nil {
return err