Merge pull request #1 from sljeff/master

fix for kingshard
shanbay · Jul 9, 2020 · b1d16e4 · b1d16e4
2 parents 7207664 + d28d294
commit b1d16e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dialect/sql/sqlgraph/graph.go b/dialect/sql/sqlgraph/graph.go
@@ -570,7 +570,7 @@ func (q *query) count(ctx context.Context, drv dialect.Driver) (int, error) {
 	selector.Count(selector.C(q.Node.ID.Column))
 	if q.Unique {
 		selector.SetDistinct(false)
-		selector.Count(sql.Distinct(selector.C(q.Node.ID.Column)))
+		selector.Count()
 	}
 	query, args := selector.Query()
 	if err := drv.Query(ctx, query, args, rows); err != nil {
@@ -670,7 +670,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(matchID(u.Node.ID.Column, ids))
+	update := u.builder.Update(u.Node.Table).Where(selector.Clone().P())
 	if err := u.setTableColumns(update, addEdges, clearEdges); err != nil {
 		return 0, err
 	}