Merge pull request #5 from nicholasxuu/fix-update-problem

fix update problem
shanbay · Sep 8, 2020 · cc2389d · cc2389d
2 parents 1cd214f + 68f6ca7
commit cc2389d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dialect/sql/sqlgraph/graph.go b/dialect/sql/sqlgraph/graph.go
@@ -652,6 +652,9 @@ func (u *updater) nodes(ctx context.Context, tx dialect.ExecQuerier) (int, error
 	)
 	selector := u.builder.Select(u.Node.ID.Column).
 		From(u.builder.Table(u.Node.Table))
+
+	archivedWherePredicate := selector.Clone().P()
+
 	if pred := u.Predicate; pred != nil {
 		pred(selector)
 	}
@@ -670,7 +673,7 @@ func (u *updater) nodes(ctx context.Context, tx dialect.ExecQuerier) (int, error
 	if len(ids) == 0 {
 		return 0, nil
 	}
-	update := u.builder.Update(u.Node.Table).Where(selector.Clone().P())
+	update := u.builder.Update(u.Node.Table).Where(archivedWherePredicate)
 	if err := u.setTableColumns(update, addEdges, clearEdges); err != nil {
 		return 0, err
 	}