123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065 |
- // Code generated by entc, DO NOT EDIT.
- package ent
- import (
- "context"
- "database/sql/driver"
- "errors"
- "fmt"
- "math"
- "todo/ent/predicate"
- "todo/ent/todo"
- "entgo.io/ent/dialect/sql"
- "entgo.io/ent/dialect/sql/sqlgraph"
- "entgo.io/ent/schema/field"
- )
- // TodoQuery is the builder for querying Todo entities.
- type TodoQuery struct {
- config
- limit *int
- offset *int
- unique *bool
- order []OrderFunc
- fields []string
- predicates []predicate.Todo
- // eager-loading edges.
- withChildren *TodoQuery
- withParent *TodoQuery
- withFKs bool
- // intermediate query (i.e. traversal path).
- sql *sql.Selector
- path func(context.Context) (*sql.Selector, error)
- }
- // Where adds a new predicate for the TodoQuery builder.
- func (tq *TodoQuery) Where(ps ...predicate.Todo) *TodoQuery {
- tq.predicates = append(tq.predicates, ps...)
- return tq
- }
- // Limit adds a limit step to the query.
- func (tq *TodoQuery) Limit(limit int) *TodoQuery {
- tq.limit = &limit
- return tq
- }
- // Offset adds an offset step to the query.
- func (tq *TodoQuery) Offset(offset int) *TodoQuery {
- tq.offset = &offset
- return tq
- }
- // 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 (tq *TodoQuery) Unique(unique bool) *TodoQuery {
- tq.unique = &unique
- return tq
- }
- // Order adds an order step to the query.
- func (tq *TodoQuery) Order(o ...OrderFunc) *TodoQuery {
- tq.order = append(tq.order, o...)
- return tq
- }
- // QueryChildren chains the current query on the "children" edge.
- func (tq *TodoQuery) QueryChildren() *TodoQuery {
- query := &TodoQuery{config: tq.config}
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := tq.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(todo.Table, todo.FieldID, selector),
- sqlgraph.To(todo.Table, todo.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, true, todo.ChildrenTable, todo.ChildrenColumn),
- )
- fromU = sqlgraph.SetNeighbors(tq.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryParent chains the current query on the "parent" edge.
- func (tq *TodoQuery) QueryParent() *TodoQuery {
- query := &TodoQuery{config: tq.config}
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := tq.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(todo.Table, todo.FieldID, selector),
- sqlgraph.To(todo.Table, todo.FieldID),
- sqlgraph.Edge(sqlgraph.M2O, false, todo.ParentTable, todo.ParentColumn),
- )
- fromU = sqlgraph.SetNeighbors(tq.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // First returns the first Todo entity from the query.
- // Returns a *NotFoundError when no Todo was found.
- func (tq *TodoQuery) First(ctx context.Context) (*Todo, error) {
- nodes, err := tq.Limit(1).All(ctx)
- if err != nil {
- return nil, err
- }
- if len(nodes) == 0 {
- return nil, &NotFoundError{todo.Label}
- }
- return nodes[0], nil
- }
- // FirstX is like First, but panics if an error occurs.
- func (tq *TodoQuery) FirstX(ctx context.Context) *Todo {
- node, err := tq.First(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return node
- }
- // FirstID returns the first Todo ID from the query.
- // Returns a *NotFoundError when no Todo ID was found.
- func (tq *TodoQuery) FirstID(ctx context.Context) (id int, err error) {
- var ids []int
- if ids, err = tq.Limit(1).IDs(ctx); err != nil {
- return
- }
- if len(ids) == 0 {
- err = &NotFoundError{todo.Label}
- return
- }
- return ids[0], nil
- }
- // FirstIDX is like FirstID, but panics if an error occurs.
- func (tq *TodoQuery) FirstIDX(ctx context.Context) int {
- id, err := tq.FirstID(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return id
- }
- // Only returns a single Todo entity found by the query, ensuring it only returns one.
- // Returns a *NotSingularError when more than one Todo entity is found.
- // Returns a *NotFoundError when no Todo entities are found.
- func (tq *TodoQuery) Only(ctx context.Context) (*Todo, error) {
- nodes, err := tq.Limit(2).All(ctx)
- if err != nil {
- return nil, err
- }
- switch len(nodes) {
- case 1:
- return nodes[0], nil
- case 0:
- return nil, &NotFoundError{todo.Label}
- default:
- return nil, &NotSingularError{todo.Label}
- }
- }
- // OnlyX is like Only, but panics if an error occurs.
- func (tq *TodoQuery) OnlyX(ctx context.Context) *Todo {
- node, err := tq.Only(ctx)
- if err != nil {
- panic(err)
- }
- return node
- }
- // OnlyID is like Only, but returns the only Todo ID in the query.
- // Returns a *NotSingularError when more than one Todo ID is found.
- // Returns a *NotFoundError when no entities are found.
- func (tq *TodoQuery) OnlyID(ctx context.Context) (id int, err error) {
- var ids []int
- if ids, err = tq.Limit(2).IDs(ctx); err != nil {
- return
- }
- switch len(ids) {
- case 1:
- id = ids[0]
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = &NotSingularError{todo.Label}
- }
- return
- }
- // OnlyIDX is like OnlyID, but panics if an error occurs.
- func (tq *TodoQuery) OnlyIDX(ctx context.Context) int {
- id, err := tq.OnlyID(ctx)
- if err != nil {
- panic(err)
- }
- return id
- }
- // All executes the query and returns a list of Todos.
- func (tq *TodoQuery) All(ctx context.Context) ([]*Todo, error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- return tq.sqlAll(ctx)
- }
- // AllX is like All, but panics if an error occurs.
- func (tq *TodoQuery) AllX(ctx context.Context) []*Todo {
- nodes, err := tq.All(ctx)
- if err != nil {
- panic(err)
- }
- return nodes
- }
- // IDs executes the query and returns a list of Todo IDs.
- func (tq *TodoQuery) IDs(ctx context.Context) ([]int, error) {
- var ids []int
- if err := tq.Select(todo.FieldID).Scan(ctx, &ids); err != nil {
- return nil, err
- }
- return ids, nil
- }
- // IDsX is like IDs, but panics if an error occurs.
- func (tq *TodoQuery) IDsX(ctx context.Context) []int {
- ids, err := tq.IDs(ctx)
- if err != nil {
- panic(err)
- }
- return ids
- }
- // Count returns the count of the given query.
- func (tq *TodoQuery) Count(ctx context.Context) (int, error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return 0, err
- }
- return tq.sqlCount(ctx)
- }
- // CountX is like Count, but panics if an error occurs.
- func (tq *TodoQuery) CountX(ctx context.Context) int {
- count, err := tq.Count(ctx)
- if err != nil {
- panic(err)
- }
- return count
- }
- // Exist returns true if the query has elements in the graph.
- func (tq *TodoQuery) Exist(ctx context.Context) (bool, error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return false, err
- }
- return tq.sqlExist(ctx)
- }
- // ExistX is like Exist, but panics if an error occurs.
- func (tq *TodoQuery) ExistX(ctx context.Context) bool {
- exist, err := tq.Exist(ctx)
- if err != nil {
- panic(err)
- }
- return exist
- }
- // Clone returns a duplicate of the TodoQuery builder, including all associated steps. It can be
- // used to prepare common query builders and use them differently after the clone is made.
- func (tq *TodoQuery) Clone() *TodoQuery {
- if tq == nil {
- return nil
- }
- return &TodoQuery{
- config: tq.config,
- limit: tq.limit,
- offset: tq.offset,
- order: append([]OrderFunc{}, tq.order...),
- predicates: append([]predicate.Todo{}, tq.predicates...),
- withChildren: tq.withChildren.Clone(),
- withParent: tq.withParent.Clone(),
- // clone intermediate query.
- sql: tq.sql.Clone(),
- path: tq.path,
- unique: tq.unique,
- }
- }
- // WithChildren tells the query-builder to eager-load the nodes that are connected to
- // the "children" edge. The optional arguments are used to configure the query builder of the edge.
- func (tq *TodoQuery) WithChildren(opts ...func(*TodoQuery)) *TodoQuery {
- query := &TodoQuery{config: tq.config}
- for _, opt := range opts {
- opt(query)
- }
- tq.withChildren = query
- return tq
- }
- // WithParent tells the query-builder to eager-load the nodes that are connected to
- // the "parent" edge. The optional arguments are used to configure the query builder of the edge.
- func (tq *TodoQuery) WithParent(opts ...func(*TodoQuery)) *TodoQuery {
- query := &TodoQuery{config: tq.config}
- for _, opt := range opts {
- opt(query)
- }
- tq.withParent = query
- return tq
- }
- // GroupBy is used to group vertices by one or more fields/columns.
- // It is often used with aggregate functions, like: count, max, mean, min, sum.
- //
- // Example:
- //
- // var v []struct {
- // Text string `json:"text,omitempty"`
- // Count int `json:"count,omitempty"`
- // }
- //
- // client.Todo.Query().
- // GroupBy(todo.FieldText).
- // Aggregate(ent.Count()).
- // Scan(ctx, &v)
- //
- func (tq *TodoQuery) GroupBy(field string, fields ...string) *TodoGroupBy {
- group := &TodoGroupBy{config: tq.config}
- group.fields = append([]string{field}, fields...)
- group.path = func(ctx context.Context) (prev *sql.Selector, err error) {
- if err := tq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- return tq.sqlQuery(ctx), nil
- }
- return group
- }
- // Select allows the selection one or more fields/columns for the given query,
- // instead of selecting all fields in the entity.
- //
- // Example:
- //
- // var v []struct {
- // Text string `json:"text,omitempty"`
- // }
- //
- // client.Todo.Query().
- // Select(todo.FieldText).
- // Scan(ctx, &v)
- //
- func (tq *TodoQuery) Select(fields ...string) *TodoSelect {
- tq.fields = append(tq.fields, fields...)
- return &TodoSelect{TodoQuery: tq}
- }
- func (tq *TodoQuery) prepareQuery(ctx context.Context) error {
- for _, f := range tq.fields {
- if !todo.ValidColumn(f) {
- return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
- }
- }
- if tq.path != nil {
- prev, err := tq.path(ctx)
- if err != nil {
- return err
- }
- tq.sql = prev
- }
- return nil
- }
- func (tq *TodoQuery) sqlAll(ctx context.Context) ([]*Todo, error) {
- var (
- nodes = []*Todo{}
- withFKs = tq.withFKs
- _spec = tq.querySpec()
- loadedTypes = [2]bool{
- tq.withChildren != nil,
- tq.withParent != nil,
- }
- )
- if tq.withParent != nil {
- withFKs = true
- }
- if withFKs {
- _spec.Node.Columns = append(_spec.Node.Columns, todo.ForeignKeys...)
- }
- _spec.ScanValues = func(columns []string) ([]interface{}, error) {
- node := &Todo{config: tq.config}
- nodes = append(nodes, node)
- return node.scanValues(columns)
- }
- _spec.Assign = func(columns []string, values []interface{}) error {
- if len(nodes) == 0 {
- return fmt.Errorf("ent: Assign called without calling ScanValues")
- }
- node := nodes[len(nodes)-1]
- node.Edges.loadedTypes = loadedTypes
- return node.assignValues(columns, values)
- }
- if err := sqlgraph.QueryNodes(ctx, tq.driver, _spec); err != nil {
- return nil, err
- }
- if len(nodes) == 0 {
- return nodes, nil
- }
- if query := tq.withChildren; query != nil {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int]*Todo)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- nodes[i].Edges.Children = []*Todo{}
- }
- query.withFKs = true
- query.Where(predicate.Todo(func(s *sql.Selector) {
- s.Where(sql.InValues(todo.ChildrenColumn, fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return nil, err
- }
- for _, n := range neighbors {
- fk := n.todo_parent
- if fk == nil {
- return nil, fmt.Errorf(`foreign-key "todo_parent" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return nil, fmt.Errorf(`unexpected foreign-key "todo_parent" returned %v for node %v`, *fk, n.ID)
- }
- node.Edges.Children = append(node.Edges.Children, n)
- }
- }
- if query := tq.withParent; query != nil {
- ids := make([]int, 0, len(nodes))
- nodeids := make(map[int][]*Todo)
- for i := range nodes {
- if nodes[i].todo_parent == nil {
- continue
- }
- fk := *nodes[i].todo_parent
- if _, ok := nodeids[fk]; !ok {
- ids = append(ids, fk)
- }
- nodeids[fk] = append(nodeids[fk], nodes[i])
- }
- query.Where(todo.IDIn(ids...))
- neighbors, err := query.All(ctx)
- if err != nil {
- return nil, err
- }
- for _, n := range neighbors {
- nodes, ok := nodeids[n.ID]
- if !ok {
- return nil, fmt.Errorf(`unexpected foreign-key "todo_parent" returned %v`, n.ID)
- }
- for i := range nodes {
- nodes[i].Edges.Parent = n
- }
- }
- }
- return nodes, nil
- }
- func (tq *TodoQuery) sqlCount(ctx context.Context) (int, error) {
- _spec := tq.querySpec()
- _spec.Node.Columns = tq.fields
- if len(tq.fields) > 0 {
- _spec.Unique = tq.unique != nil && *tq.unique
- }
- return sqlgraph.CountNodes(ctx, tq.driver, _spec)
- }
- func (tq *TodoQuery) sqlExist(ctx context.Context) (bool, error) {
- n, err := tq.sqlCount(ctx)
- if err != nil {
- return false, fmt.Errorf("ent: check existence: %w", err)
- }
- return n > 0, nil
- }
- func (tq *TodoQuery) querySpec() *sqlgraph.QuerySpec {
- _spec := &sqlgraph.QuerySpec{
- Node: &sqlgraph.NodeSpec{
- Table: todo.Table,
- Columns: todo.Columns,
- ID: &sqlgraph.FieldSpec{
- Type: field.TypeInt,
- Column: todo.FieldID,
- },
- },
- From: tq.sql,
- Unique: true,
- }
- if unique := tq.unique; unique != nil {
- _spec.Unique = *unique
- }
- if fields := tq.fields; len(fields) > 0 {
- _spec.Node.Columns = make([]string, 0, len(fields))
- _spec.Node.Columns = append(_spec.Node.Columns, todo.FieldID)
- for i := range fields {
- if fields[i] != todo.FieldID {
- _spec.Node.Columns = append(_spec.Node.Columns, fields[i])
- }
- }
- }
- if ps := tq.predicates; len(ps) > 0 {
- _spec.Predicate = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- if limit := tq.limit; limit != nil {
- _spec.Limit = *limit
- }
- if offset := tq.offset; offset != nil {
- _spec.Offset = *offset
- }
- if ps := tq.order; len(ps) > 0 {
- _spec.Order = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- return _spec
- }
- func (tq *TodoQuery) sqlQuery(ctx context.Context) *sql.Selector {
- builder := sql.Dialect(tq.driver.Dialect())
- t1 := builder.Table(todo.Table)
- columns := tq.fields
- if len(columns) == 0 {
- columns = todo.Columns
- }
- selector := builder.Select(t1.Columns(columns...)...).From(t1)
- if tq.sql != nil {
- selector = tq.sql
- selector.Select(selector.Columns(columns...)...)
- }
- if tq.unique != nil && *tq.unique {
- selector.Distinct()
- }
- for _, p := range tq.predicates {
- p(selector)
- }
- for _, p := range tq.order {
- p(selector)
- }
- if offset := tq.offset; offset != nil {
- // limit is mandatory for offset clause. We start
- // with default value, and override it below if needed.
- selector.Offset(*offset).Limit(math.MaxInt32)
- }
- if limit := tq.limit; limit != nil {
- selector.Limit(*limit)
- }
- return selector
- }
- // TodoGroupBy is the group-by builder for Todo entities.
- type TodoGroupBy struct {
- config
- fields []string
- fns []AggregateFunc
- // intermediate query (i.e. traversal path).
- sql *sql.Selector
- path func(context.Context) (*sql.Selector, error)
- }
- // Aggregate adds the given aggregation functions to the group-by query.
- func (tgb *TodoGroupBy) Aggregate(fns ...AggregateFunc) *TodoGroupBy {
- tgb.fns = append(tgb.fns, fns...)
- return tgb
- }
- // Scan applies the group-by query and scans the result into the given value.
- func (tgb *TodoGroupBy) Scan(ctx context.Context, v interface{}) error {
- query, err := tgb.path(ctx)
- if err != nil {
- return err
- }
- tgb.sql = query
- return tgb.sqlScan(ctx, v)
- }
- // ScanX is like Scan, but panics if an error occurs.
- func (tgb *TodoGroupBy) ScanX(ctx context.Context, v interface{}) {
- if err := tgb.Scan(ctx, v); err != nil {
- panic(err)
- }
- }
- // Strings returns list of strings from group-by.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Strings(ctx context.Context) ([]string, error) {
- if len(tgb.fields) > 1 {
- return nil, errors.New("ent: TodoGroupBy.Strings is not achievable when grouping more than 1 field")
- }
- var v []string
- if err := tgb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // StringsX is like Strings, but panics if an error occurs.
- func (tgb *TodoGroupBy) StringsX(ctx context.Context) []string {
- v, err := tgb.Strings(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // String returns a single string from a group-by query.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) String(ctx context.Context) (_ string, err error) {
- var v []string
- if v, err = tgb.Strings(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoGroupBy.Strings returned %d results when one was expected", len(v))
- }
- return
- }
- // StringX is like String, but panics if an error occurs.
- func (tgb *TodoGroupBy) StringX(ctx context.Context) string {
- v, err := tgb.String(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Ints returns list of ints from group-by.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Ints(ctx context.Context) ([]int, error) {
- if len(tgb.fields) > 1 {
- return nil, errors.New("ent: TodoGroupBy.Ints is not achievable when grouping more than 1 field")
- }
- var v []int
- if err := tgb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // IntsX is like Ints, but panics if an error occurs.
- func (tgb *TodoGroupBy) IntsX(ctx context.Context) []int {
- v, err := tgb.Ints(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Int returns a single int from a group-by query.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Int(ctx context.Context) (_ int, err error) {
- var v []int
- if v, err = tgb.Ints(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoGroupBy.Ints returned %d results when one was expected", len(v))
- }
- return
- }
- // IntX is like Int, but panics if an error occurs.
- func (tgb *TodoGroupBy) IntX(ctx context.Context) int {
- v, err := tgb.Int(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Float64s returns list of float64s from group-by.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Float64s(ctx context.Context) ([]float64, error) {
- if len(tgb.fields) > 1 {
- return nil, errors.New("ent: TodoGroupBy.Float64s is not achievable when grouping more than 1 field")
- }
- var v []float64
- if err := tgb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // Float64sX is like Float64s, but panics if an error occurs.
- func (tgb *TodoGroupBy) Float64sX(ctx context.Context) []float64 {
- v, err := tgb.Float64s(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Float64 returns a single float64 from a group-by query.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Float64(ctx context.Context) (_ float64, err error) {
- var v []float64
- if v, err = tgb.Float64s(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoGroupBy.Float64s returned %d results when one was expected", len(v))
- }
- return
- }
- // Float64X is like Float64, but panics if an error occurs.
- func (tgb *TodoGroupBy) Float64X(ctx context.Context) float64 {
- v, err := tgb.Float64(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Bools returns list of bools from group-by.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Bools(ctx context.Context) ([]bool, error) {
- if len(tgb.fields) > 1 {
- return nil, errors.New("ent: TodoGroupBy.Bools is not achievable when grouping more than 1 field")
- }
- var v []bool
- if err := tgb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // BoolsX is like Bools, but panics if an error occurs.
- func (tgb *TodoGroupBy) BoolsX(ctx context.Context) []bool {
- v, err := tgb.Bools(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Bool returns a single bool from a group-by query.
- // It is only allowed when executing a group-by query with one field.
- func (tgb *TodoGroupBy) Bool(ctx context.Context) (_ bool, err error) {
- var v []bool
- if v, err = tgb.Bools(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoGroupBy.Bools returned %d results when one was expected", len(v))
- }
- return
- }
- // BoolX is like Bool, but panics if an error occurs.
- func (tgb *TodoGroupBy) BoolX(ctx context.Context) bool {
- v, err := tgb.Bool(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- func (tgb *TodoGroupBy) sqlScan(ctx context.Context, v interface{}) error {
- for _, f := range tgb.fields {
- if !todo.ValidColumn(f) {
- return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)}
- }
- }
- selector := tgb.sqlQuery()
- if err := selector.Err(); err != nil {
- return err
- }
- rows := &sql.Rows{}
- query, args := selector.Query()
- if err := tgb.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
- func (tgb *TodoGroupBy) sqlQuery() *sql.Selector {
- selector := tgb.sql.Select()
- aggregation := make([]string, 0, len(tgb.fns))
- for _, fn := range tgb.fns {
- aggregation = append(aggregation, fn(selector))
- }
- // If no columns were selected in a custom aggregation function, the default
- // selection is the fields used for "group-by", and the aggregation functions.
- if len(selector.SelectedColumns()) == 0 {
- columns := make([]string, 0, len(tgb.fields)+len(tgb.fns))
- for _, f := range tgb.fields {
- columns = append(columns, selector.C(f))
- }
- columns = append(columns, aggregation...)
- selector.Select(columns...)
- }
- return selector.GroupBy(selector.Columns(tgb.fields...)...)
- }
- // TodoSelect is the builder for selecting fields of Todo entities.
- type TodoSelect struct {
- *TodoQuery
- // intermediate query (i.e. traversal path).
- sql *sql.Selector
- }
- // Scan applies the selector query and scans the result into the given value.
- func (ts *TodoSelect) Scan(ctx context.Context, v interface{}) error {
- if err := ts.prepareQuery(ctx); err != nil {
- return err
- }
- ts.sql = ts.TodoQuery.sqlQuery(ctx)
- return ts.sqlScan(ctx, v)
- }
- // ScanX is like Scan, but panics if an error occurs.
- func (ts *TodoSelect) ScanX(ctx context.Context, v interface{}) {
- if err := ts.Scan(ctx, v); err != nil {
- panic(err)
- }
- }
- // Strings returns list of strings from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Strings(ctx context.Context) ([]string, error) {
- if len(ts.fields) > 1 {
- return nil, errors.New("ent: TodoSelect.Strings is not achievable when selecting more than 1 field")
- }
- var v []string
- if err := ts.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // StringsX is like Strings, but panics if an error occurs.
- func (ts *TodoSelect) StringsX(ctx context.Context) []string {
- v, err := ts.Strings(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // String returns a single string from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) String(ctx context.Context) (_ string, err error) {
- var v []string
- if v, err = ts.Strings(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoSelect.Strings returned %d results when one was expected", len(v))
- }
- return
- }
- // StringX is like String, but panics if an error occurs.
- func (ts *TodoSelect) StringX(ctx context.Context) string {
- v, err := ts.String(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Ints returns list of ints from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Ints(ctx context.Context) ([]int, error) {
- if len(ts.fields) > 1 {
- return nil, errors.New("ent: TodoSelect.Ints is not achievable when selecting more than 1 field")
- }
- var v []int
- if err := ts.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // IntsX is like Ints, but panics if an error occurs.
- func (ts *TodoSelect) IntsX(ctx context.Context) []int {
- v, err := ts.Ints(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Int returns a single int from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Int(ctx context.Context) (_ int, err error) {
- var v []int
- if v, err = ts.Ints(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoSelect.Ints returned %d results when one was expected", len(v))
- }
- return
- }
- // IntX is like Int, but panics if an error occurs.
- func (ts *TodoSelect) IntX(ctx context.Context) int {
- v, err := ts.Int(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Float64s returns list of float64s from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Float64s(ctx context.Context) ([]float64, error) {
- if len(ts.fields) > 1 {
- return nil, errors.New("ent: TodoSelect.Float64s is not achievable when selecting more than 1 field")
- }
- var v []float64
- if err := ts.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // Float64sX is like Float64s, but panics if an error occurs.
- func (ts *TodoSelect) Float64sX(ctx context.Context) []float64 {
- v, err := ts.Float64s(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Float64 returns a single float64 from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Float64(ctx context.Context) (_ float64, err error) {
- var v []float64
- if v, err = ts.Float64s(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoSelect.Float64s returned %d results when one was expected", len(v))
- }
- return
- }
- // Float64X is like Float64, but panics if an error occurs.
- func (ts *TodoSelect) Float64X(ctx context.Context) float64 {
- v, err := ts.Float64(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Bools returns list of bools from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Bools(ctx context.Context) ([]bool, error) {
- if len(ts.fields) > 1 {
- return nil, errors.New("ent: TodoSelect.Bools is not achievable when selecting more than 1 field")
- }
- var v []bool
- if err := ts.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // BoolsX is like Bools, but panics if an error occurs.
- func (ts *TodoSelect) BoolsX(ctx context.Context) []bool {
- v, err := ts.Bools(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- // Bool returns a single bool from a selector. It is only allowed when selecting one field.
- func (ts *TodoSelect) Bool(ctx context.Context) (_ bool, err error) {
- var v []bool
- if v, err = ts.Bools(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{todo.Label}
- default:
- err = fmt.Errorf("ent: TodoSelect.Bools returned %d results when one was expected", len(v))
- }
- return
- }
- // BoolX is like Bool, but panics if an error occurs.
- func (ts *TodoSelect) BoolX(ctx context.Context) bool {
- v, err := ts.Bool(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- func (ts *TodoSelect) sqlScan(ctx context.Context, v interface{}) error {
- rows := &sql.Rows{}
- query, args := ts.sql.Query()
- if err := ts.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
|