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: add schema for deletion command

Browse Source
This commit is contained in:
Ariel Mashraki
2021-01-24 15:38:04 +02:00
committed by Ariel Mashraki
parent bdd8f2f624
commit 2c7228c23f
8 changed files with 73 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1040,7 +1040,13 @@ func (g *graph) clearM2MEdges(ctx context.Context, ids []driver.Value, edges Edg
}
}
}
query, args := g.builder.Delete(table).Where(sql.Or(preds...)).Query()
deleter := g.builder.Delete(table).Where(sql.Or(preds...))
if edges[0].Schema != "" {
// If the Schema field was provided to the EdgeSpec (by the
// generated code), it should be the same for all EdgeSpecs.
deleter.Schema(edges[0].Schema)
}
query, args := deleter.Query()
if err := g.tx.Exec(ctx, query, args, &res); err != nil {
return fmt.Errorf("remove m2m edge for table %s: %v", table, err)
}
@@ -1057,7 +1063,12 @@ func (g *graph) addM2MEdges(ctx context.Context, ids []driver.Value, edges EdgeS
)
for _, table := range edgeKeys(tables) {
edges := tables[table]
insert := g.builder.Insert(table).Schema(edges[0].Schema).Columns(edges[0].Columns...)
insert := g.builder.Insert(table).Columns(edges[0].Columns...)
if edges[0].Schema != "" {
// If the Schema field was provided to the EdgeSpec (by the
// generated code), it should be the same for all EdgeSpecs.
insert.Schema(edges[0].Schema)
}
for _, edge := range edges {
pk1, pk2 := ids, edge.Target.Nodes
if edge.Inverse {
@@ -1085,7 +1096,12 @@ func (g *graph) batchAddM2M(ctx context.Context, spec *BatchCreateSpec) error {
for t, edges := range edges.GroupTable() {
insert, ok := tables[t]
if !ok {
insert = g.builder.Insert(t).Schema(edges[0].Schema).Columns(edges[0].Columns...)
insert = g.builder.Insert(t).Columns(edges[0].Columns...)
if edges[0].Schema != "" {
// If the Schema field was provided to the EdgeSpec (by the
// generated code), it should be the same for all EdgeSpecs.
insert.Schema(edges[0].Schema)
}
}
tables[t] = insert
if len(edges) != 1 {