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

entc/gen: move sql-update-one to sqlgraph

Browse Source
This commit is contained in:
Ariel Mashraki
2019-12-18 19:25:46 +02:00
parent 1bbe460259
commit 7fde9b7642
53 changed files with 8834 additions and 7859 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -630,15 +630,28 @@ func (u *updater) setExternalEdges(ctx context.Context, ids []driver.Value, addE
// setTableColumns sets the table columns and foreign_keys used in insert.
func (u *updater) setTableColumns(update *sql.UpdateBuilder, addEdges, clearEdges map[Rel][]*EdgeSpec) error {
// Avoid multiple assignments to same column.
setEdges := make(map[string]bool)
for _, e := range addEdges[M2O] {
setEdges[e.Columns[0]] = true
}
for _, e := range addEdges[O2O] {
if e.Inverse || e.Bidi {
setEdges[e.Columns[0]] = true
}
}
for _, fi := range u.Fields.Clear {
update.SetNull(fi.Column)
}
for _, e := range clearEdges[M2O] {
update.SetNull(e.Columns[0])
if col := e.Columns[0]; !setEdges[col] {
update.SetNull(col)
}
}
for _, e := range clearEdges[O2O] {
if e.Inverse || e.Bidi {
update.SetNull(e.Columns[0])
col := e.Columns[0]
if (e.Inverse || e.Bidi) && !setEdges[col] {
update.SetNull(col)
}
}
err := setTableColumns(u.Fields.Set, addEdges, func(column string, value driver.Value) {