entc/gen: <T>Query.IDs should not return duplicates (#3304)

This commit is contained in:
Ariel Mashraki
2023-02-06 21:19:05 +02:00
committed by GitHub
parent 8022271f12
commit 138bd2bc1d
137 changed files with 683 additions and 408 deletions

View File

@@ -228,10 +228,12 @@ func (nq *NodeQuery) AllX(ctx context.Context) []*Node {
}
// IDs executes the query and returns a list of Node IDs.
func (nq *NodeQuery) IDs(ctx context.Context) ([]int, error) {
var ids []int
func (nq *NodeQuery) IDs(ctx context.Context) (ids []int, err error) {
if nq.ctx.Unique == nil && nq.path != nil {
nq.Unique(true)
}
ctx = setContextOp(ctx, nq.ctx, "IDs")
if err := nq.Select(node.FieldID).Scan(ctx, &ids); err != nil {
if err = nq.Select(node.FieldID).Scan(ctx, &ids); err != nil {
return nil, err
}
return ids, nil