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

dialect/sql/sqlgraph: support batch-upadte for edge-schemas (#2897)

Browse Source
This commit is contained in:
Ariel Mashraki
2022-09-02 15:33:41 +03:00
committed by GitHub
parent 3b1b75b9d7
commit 2969796669
3 changed files with 45 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -741,10 +741,27 @@ func (u *updater) nodes(ctx context.Context, drv dialect.Driver) (int, error) {
clearEdges = EdgeSpecs(u.Edges.Clear).GroupRel()
multiple = hasExternalEdges(addEdges, clearEdges)
update = u.builder.Update(u.Node.Table).Schema(u.Node.Schema)
selector = u.builder.Select(u.Node.ID.Column).
selector = u.builder.Select().
From(u.builder.Table(u.Node.Table).Schema(u.Node.Schema)).
WithContext(ctx)
)
switch {
// In case it is not an edge schema, the id holds the PK of
// the returned nodes are used for updating external tables.
case u.Node.ID != nil:
selector.Select(u.Node.ID.Column)
case len(u.Node.CompositeID) == 2:
// Other edge-schemas (M2M tables) cannot be updated by this operation.
// Also, in case there is a need to update an external foreign-key, it must
// be a single value and the user should use the "update by id" API instead.
if multiple {
return 0, fmt.Errorf("sql/sqlgraph: update edge schema table %q cannot update external tables", u.Node.Table)
}
case len(u.Node.CompositeID) != 2:
return 0, fmt.Errorf("sql/sqlgraph: invalid composite id for update table %q", u.Node.Table)
default:
return 0, fmt.Errorf("sql/sqlgraph: missing node id for update table %q", u.Node.Table)
}
if err := u.setTableColumns(update, addEdges, clearEdges); err != nil {
return 0, err
}