entc/gen/template: allow disable DISTINCT in queries (#1371)

examples/edgeindex/ent/city_query.go

@@ -26,6 +26,7 @@ type CityQuery struct {
 	config
 	limit      *int
 	offset     *int
+	unique     *bool
 	order      []OrderFunc
 	fields     []string
 	predicates []predicate.City
@@ -54,6 +55,13 @@ func (cq *CityQuery) Offset(offset int) *CityQuery {
 	return cq
 }
 
+// Unique configures the query builder to filter duplicate records on query.
+// By default, unique is set to true, and can be disabled using this method.
+func (cq *CityQuery) Unique(unique bool) *CityQuery {
+	cq.unique = &unique
+	return cq
+}
+
 // Order adds an order step to the query.
 func (cq *CityQuery) Order(o ...OrderFunc) *CityQuery {
 	cq.order = append(cq.order, o...)
@@ -428,6 +436,9 @@ func (cq *CityQuery) querySpec() *sqlgraph.QuerySpec {
 		From:   cq.sql,
 		Unique: true,
 	}
+	if unique := cq.unique; unique != nil {
+		_spec.Unique = *unique
+	}
 	if fields := cq.fields; len(fields) > 0 {
 		_spec.Node.Columns = make([]string, 0, len(fields))
 		_spec.Node.Columns = append(_spec.Node.Columns, city.FieldID)

examples/edgeindex/ent/street_query.go

@@ -25,6 +25,7 @@ type StreetQuery struct {
 	config
 	limit      *int
 	offset     *int
+	unique     *bool
 	order      []OrderFunc
 	fields     []string
 	predicates []predicate.Street
@@ -54,6 +55,13 @@ func (sq *StreetQuery) Offset(offset int) *StreetQuery {
 	return sq
 }
 
+// Unique configures the query builder to filter duplicate records on query.
+// By default, unique is set to true, and can be disabled using this method.
+func (sq *StreetQuery) Unique(unique bool) *StreetQuery {
+	sq.unique = &unique
+	return sq
+}
+
 // Order adds an order step to the query.
 func (sq *StreetQuery) Order(o ...OrderFunc) *StreetQuery {
 	sq.order = append(sq.order, o...)
@@ -435,6 +443,9 @@ func (sq *StreetQuery) querySpec() *sqlgraph.QuerySpec {
 		From:   sq.sql,
 		Unique: true,
 	}
+	if unique := sq.unique; unique != nil {
+		_spec.Unique = *unique
+	}
 	if fields := sq.fields; len(fields) > 0 {
 		_spec.Node.Columns = make([]string, 0, len(fields))
 		_spec.Node.Columns = append(_spec.Node.Columns, street.FieldID)