golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen: use query ops in generated code (#4005)

Browse Source
This commit is contained in:
Ariel Mashraki
2024-04-05 15:05:56 +03:00
committed by GitHub
parent 7beda3fb2b
commit 5d43d62c65
156 changed files with 1703 additions and 1548 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
examples/o2orecur/ent/node_query.go
View File

@@ -12,6 +12,7 @@ import (
"fmt"
"math"
"entgo.io/ent"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/examples/o2orecur/ent/node"
@@ -111,7 +112,7 @@ func (nq *NodeQuery) QueryNext() *NodeQuery {
// First returns the first Node entity from the query.
// Returns a *NotFoundError when no Node was found.
func (nq *NodeQuery) First(ctx context.Context) (*Node, error) {
nodes, err := nq.Limit(1).All(setContextOp(ctx, nq.ctx, "First"))
nodes, err := nq.Limit(1).All(setContextOp(ctx, nq.ctx, ent.OpQueryFirst))
if err != nil {
return nil, err
}
@@ -134,7 +135,7 @@ func (nq *NodeQuery) FirstX(ctx context.Context) *Node {
// Returns a *NotFoundError when no Node ID was found.
func (nq *NodeQuery) FirstID(ctx context.Context) (id int, err error) {
var ids []int
if ids, err = nq.Limit(1).IDs(setContextOp(ctx, nq.ctx, "FirstID")); err != nil {
if ids, err = nq.Limit(1).IDs(setContextOp(ctx, nq.ctx, ent.OpQueryFirstID)); err != nil {
return
}
if len(ids) == 0 {
@@ -157,7 +158,7 @@ func (nq *NodeQuery) FirstIDX(ctx context.Context) int {
// Returns a *NotSingularError when more than one Node entity is found.
// Returns a *NotFoundError when no Node entities are found.
func (nq *NodeQuery) Only(ctx context.Context) (*Node, error) {
nodes, err := nq.Limit(2).All(setContextOp(ctx, nq.ctx, "Only"))
nodes, err := nq.Limit(2).All(setContextOp(ctx, nq.ctx, ent.OpQueryOnly))
if err != nil {
return nil, err
}
@@ -185,7 +186,7 @@ func (nq *NodeQuery) OnlyX(ctx context.Context) *Node {
// Returns a *NotFoundError when no entities are found.
func (nq *NodeQuery) OnlyID(ctx context.Context) (id int, err error) {
var ids []int
if ids, err = nq.Limit(2).IDs(setContextOp(ctx, nq.ctx, "OnlyID")); err != nil {
if ids, err = nq.Limit(2).IDs(setContextOp(ctx, nq.ctx, ent.OpQueryOnlyID)); err != nil {
return
}
switch len(ids) {
@@ -210,7 +211,7 @@ func (nq *NodeQuery) OnlyIDX(ctx context.Context) int {
// All executes the query and returns a list of Nodes.
func (nq *NodeQuery) All(ctx context.Context) ([]*Node, error) {
ctx = setContextOp(ctx, nq.ctx, "All")
ctx = setContextOp(ctx, nq.ctx, ent.OpQueryAll)
if err := nq.prepareQuery(ctx); err != nil {
return nil, err
}
@@ -232,7 +233,7 @@ 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")
ctx = setContextOp(ctx, nq.ctx, ent.OpQueryIDs)
if err = nq.Select(node.FieldID).Scan(ctx, &ids); err != nil {
return nil, err
}
@@ -250,7 +251,7 @@ func (nq *NodeQuery) IDsX(ctx context.Context) []int {
// Count returns the count of the given query.
func (nq *NodeQuery) Count(ctx context.Context) (int, error) {
ctx = setContextOp(ctx, nq.ctx, "Count")
ctx = setContextOp(ctx, nq.ctx, ent.OpQueryCount)
if err := nq.prepareQuery(ctx); err != nil {
return 0, err
}
@@ -268,7 +269,7 @@ func (nq *NodeQuery) CountX(ctx context.Context) int {
// Exist returns true if the query has elements in the graph.
func (nq *NodeQuery) Exist(ctx context.Context) (bool, error) {
ctx = setContextOp(ctx, nq.ctx, "Exist")
ctx = setContextOp(ctx, nq.ctx, ent.OpQueryExist)
switch _, err := nq.FirstID(ctx); {
case IsNotFound(err):
return false, nil
@@ -601,7 +602,7 @@ func (ngb *NodeGroupBy) Aggregate(fns ...AggregateFunc) *NodeGroupBy {
// Scan applies the selector query and scans the result into the given value.
func (ngb *NodeGroupBy) Scan(ctx context.Context, v any) error {
ctx = setContextOp(ctx, ngb.build.ctx, "GroupBy")
ctx = setContextOp(ctx, ngb.build.ctx, ent.OpQueryGroupBy)
if err := ngb.build.prepareQuery(ctx); err != nil {
return err
}
@@ -649,7 +650,7 @@ func (ns *NodeSelect) Aggregate(fns ...AggregateFunc) *NodeSelect {
// Scan applies the selector query and scans the result into the given value.
func (ns *NodeSelect) Scan(ctx context.Context, v any) error {
ctx = setContextOp(ctx, ns.ctx, "Select")
ctx = setContextOp(ctx, ns.ctx, ent.OpQuerySelect)
if err := ns.prepareQuery(ctx); err != nil {
return err
}