123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029 |
- // Code generated by entc, DO NOT EDIT.
- package ent
- import (
- "context"
- "database/sql/driver"
- "errors"
- "fmt"
- "math"
- "code.osinet.fr/fgm/go__ent_demo/ent/group"
- "code.osinet.fr/fgm/go__ent_demo/ent/predicate"
- "code.osinet.fr/fgm/go__ent_demo/ent/user"
- "entgo.io/ent/dialect/sql"
- "entgo.io/ent/dialect/sql/sqlgraph"
- "entgo.io/ent/schema/field"
- )
- // GroupQuery is the builder for querying Group entities.
- type GroupQuery struct {
- config
- limit *int
- offset *int
- unique *bool
- order []OrderFunc
- fields []string
- predicates []predicate.Group
- // eager-loading edges.
- withUsers *UserQuery
- // intermediate query (i.e. traversal path).
- sql *sql.Selector
- path func(context.Context) (*sql.Selector, error)
- }
- // Where adds a new predicate for the GroupQuery builder.
- func (gq *GroupQuery) Where(ps ...predicate.Group) *GroupQuery {
- gq.predicates = append(gq.predicates, ps...)
- return gq
- }
- // Limit adds a limit step to the query.
- func (gq *GroupQuery) Limit(limit int) *GroupQuery {
- gq.limit = &limit
- return gq
- }
- // Offset adds an offset step to the query.
- func (gq *GroupQuery) Offset(offset int) *GroupQuery {
- gq.offset = &offset
- return gq
- }
- // 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 (gq *GroupQuery) Unique(unique bool) *GroupQuery {
- gq.unique = &unique
- return gq
- }
- // Order adds an order step to the query.
- func (gq *GroupQuery) Order(o ...OrderFunc) *GroupQuery {
- gq.order = append(gq.order, o...)
- return gq
- }
- // QueryUsers chains the current query on the "users" edge.
- func (gq *GroupQuery) QueryUsers() *UserQuery {
- query := &UserQuery{config: gq.config}
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := gq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := gq.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(user.Table, user.FieldID),
- sqlgraph.Edge(sqlgraph.M2M, false, group.UsersTable, group.UsersPrimaryKey...),
- )
- fromU = sqlgraph.SetNeighbors(gq.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // First returns the first Group entity from the query.
- // Returns a *NotFoundError when no Group was found.
- func (gq *GroupQuery) First(ctx context.Context) (*Group, error) {
- nodes, err := gq.Limit(1).All(ctx)
- if err != nil {
- return nil, err
- }
- if len(nodes) == 0 {
- return nil, &NotFoundError{group.Label}
- }
- return nodes[0], nil
- }
- // FirstX is like First, but panics if an error occurs.
- func (gq *GroupQuery) FirstX(ctx context.Context) *Group {
- node, err := gq.First(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return node
- }
- // FirstID returns the first Group ID from the query.
- // Returns a *NotFoundError when no Group ID was found.
- func (gq *GroupQuery) FirstID(ctx context.Context) (id int, err error) {
- var ids []int
- if ids, err = gq.Limit(1).IDs(ctx); err != nil {
- return
- }
- if len(ids) == 0 {
- err = &NotFoundError{group.Label}
- return
- }
- return ids[0], nil
- }
- // FirstIDX is like FirstID, but panics if an error occurs.
- func (gq *GroupQuery) FirstIDX(ctx context.Context) int {
- id, err := gq.FirstID(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return id
- }
- // Only returns a single Group entity found by the query, ensuring it only returns one.
- // Returns a *NotSingularError when more than one Group entity is found.
- // Returns a *NotFoundError when no Group entities are found.
- func (gq *GroupQuery) Only(ctx context.Context) (*Group, error) {
- nodes, err := gq.Limit(2).All(ctx)
- if err != nil {
- return nil, err
- }
- switch len(nodes) {
- case 1:
- return nodes[0], nil
- case 0:
- return nil, &NotFoundError{group.Label}
- default:
- return nil, &NotSingularError{group.Label}
- }
- }
- // OnlyX is like Only, but panics if an error occurs.
- func (gq *GroupQuery) OnlyX(ctx context.Context) *Group {
- node, err := gq.Only(ctx)
- if err != nil {
- panic(err)
- }
- return node
- }
- // OnlyID is like Only, but returns the only Group ID in the query.
- // Returns a *NotSingularError when more than one Group ID is found.
- // Returns a *NotFoundError when no entities are found.
- func (gq *GroupQuery) OnlyID(ctx context.Context) (id int, err error) {
- var ids []int
- if ids, err = gq.Limit(2).IDs(ctx); err != nil {
- return
- }
- switch len(ids) {
- case 1:
- id = ids[0]
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = &NotSingularError{group.Label}
- }
- return
- }
- // OnlyIDX is like OnlyID, but panics if an error occurs.
- func (gq *GroupQuery) OnlyIDX(ctx context.Context) int {
- id, err := gq.OnlyID(ctx)
- if err != nil {
- panic(err)
- }
- return id
- }
- // All executes the query and returns a list of Groups.
- func (gq *GroupQuery) All(ctx context.Context) ([]*Group, error) {
- if err := gq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- return gq.sqlAll(ctx)
- }
- // AllX is like All, but panics if an error occurs.
- func (gq *GroupQuery) AllX(ctx context.Context) []*Group {
- nodes, err := gq.All(ctx)
- if err != nil {
- panic(err)
- }
- return nodes
- }
- // IDs executes the query and returns a list of Group IDs.
- func (gq *GroupQuery) IDs(ctx context.Context) ([]int, error) {
- var ids []int
- if err := gq.Select(group.FieldID).Scan(ctx, &ids); err != nil {
- return nil, err
- }
- return ids, nil
- }
- // IDsX is like IDs, but panics if an error occurs.
- func (gq *GroupQuery) IDsX(ctx context.Context) []int {
- ids, err := gq.IDs(ctx)
- if err != nil {
- panic(err)
- }
- return ids
- }
- // Count returns the count of the given query.
- func (gq *GroupQuery) Count(ctx context.Context) (int, error) {
- if err := gq.prepareQuery(ctx); err != nil {
- return 0, err
- }
- return gq.sqlCount(ctx)
- }
- // CountX is like Count, but panics if an error occurs.
- func (gq *GroupQuery) CountX(ctx context.Context) int {
- count, err := gq.Count(ctx)
- if err != nil {
- panic(err)
- }
- return count
- }
- // Exist returns true if the query has elements in the graph.
- func (gq *GroupQuery) Exist(ctx context.Context) (bool, error) {
- if err := gq.prepareQuery(ctx); err != nil {
- return false, err
- }
- return gq.sqlExist(ctx)
- }
- // ExistX is like Exist, but panics if an error occurs.
- func (gq *GroupQuery) ExistX(ctx context.Context) bool {
- exist, err := gq.Exist(ctx)
- if err != nil {
- panic(err)
- }
- return exist
- }
- // Clone returns a duplicate of the GroupQuery builder, including all associated steps. It can be
- // used to prepare common query builders and use them differently after the clone is made.
- func (gq *GroupQuery) Clone() *GroupQuery {
- if gq == nil {
- return nil
- }
- return &GroupQuery{
- config: gq.config,
- limit: gq.limit,
- offset: gq.offset,
- order: append([]OrderFunc{}, gq.order...),
- predicates: append([]predicate.Group{}, gq.predicates...),
- withUsers: gq.withUsers.Clone(),
- // clone intermediate query.
- sql: gq.sql.Clone(),
- path: gq.path,
- unique: gq.unique,
- }
- }
- // WithUsers tells the query-builder to eager-load the nodes that are connected to
- // the "users" edge. The optional arguments are used to configure the query builder of the edge.
- func (gq *GroupQuery) WithUsers(opts ...func(*UserQuery)) *GroupQuery {
- query := &UserQuery{config: gq.config}
- for _, opt := range opts {
- opt(query)
- }
- gq.withUsers = query
- return gq
- }
- // 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 {
- // Name string `json:"name,omitempty"`
- // Count int `json:"count,omitempty"`
- // }
- //
- // client.Group.Query().
- // GroupBy(group.FieldName).
- // Aggregate(ent.Count()).
- // Scan(ctx, &v)
- //
- func (gq *GroupQuery) GroupBy(field string, fields ...string) *GroupGroupBy {
- group := &GroupGroupBy{config: gq.config}
- group.fields = append([]string{field}, fields...)
- group.path = func(ctx context.Context) (prev *sql.Selector, err error) {
- if err := gq.prepareQuery(ctx); err != nil {
- return nil, err
- }
- return gq.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 {
- // Name string `json:"name,omitempty"`
- // }
- //
- // client.Group.Query().
- // Select(group.FieldName).
- // Scan(ctx, &v)
- //
- func (gq *GroupQuery) Select(fields ...string) *GroupSelect {
- gq.fields = append(gq.fields, fields...)
- return &GroupSelect{GroupQuery: gq}
- }
- func (gq *GroupQuery) prepareQuery(ctx context.Context) error {
- for _, f := range gq.fields {
- if !group.ValidColumn(f) {
- return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
- }
- }
- if gq.path != nil {
- prev, err := gq.path(ctx)
- if err != nil {
- return err
- }
- gq.sql = prev
- }
- return nil
- }
- func (gq *GroupQuery) sqlAll(ctx context.Context) ([]*Group, error) {
- var (
- nodes = []*Group{}
- _spec = gq.querySpec()
- loadedTypes = [1]bool{
- gq.withUsers != nil,
- }
- )
- _spec.ScanValues = func(columns []string) ([]interface{}, error) {
- node := &Group{config: gq.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, gq.driver, _spec); err != nil {
- return nil, err
- }
- if len(nodes) == 0 {
- return nodes, nil
- }
- if query := gq.withUsers; query != nil {
- fks := make([]driver.Value, 0, len(nodes))
- ids := make(map[int]*Group, len(nodes))
- for _, node := range nodes {
- ids[node.ID] = node
- fks = append(fks, node.ID)
- node.Edges.Users = []*User{}
- }
- var (
- edgeids []int
- edges = make(map[int][]*Group)
- )
- _spec := &sqlgraph.EdgeQuerySpec{
- Edge: &sqlgraph.EdgeSpec{
- Inverse: false,
- Table: group.UsersTable,
- Columns: group.UsersPrimaryKey,
- },
- Predicate: func(s *sql.Selector) {
- s.Where(sql.InValues(group.UsersPrimaryKey[0], fks...))
- },
- ScanValues: func() [2]interface{} {
- return [2]interface{}{new(sql.NullInt64), new(sql.NullInt64)}
- },
- Assign: func(out, in interface{}) error {
- eout, ok := out.(*sql.NullInt64)
- if !ok || eout == nil {
- return fmt.Errorf("unexpected id value for edge-out")
- }
- ein, ok := in.(*sql.NullInt64)
- if !ok || ein == nil {
- return fmt.Errorf("unexpected id value for edge-in")
- }
- outValue := int(eout.Int64)
- inValue := int(ein.Int64)
- node, ok := ids[outValue]
- if !ok {
- return fmt.Errorf("unexpected node id in edges: %v", outValue)
- }
- if _, ok := edges[inValue]; !ok {
- edgeids = append(edgeids, inValue)
- }
- edges[inValue] = append(edges[inValue], node)
- return nil
- },
- }
- if err := sqlgraph.QueryEdges(ctx, gq.driver, _spec); err != nil {
- return nil, fmt.Errorf(`query edges "users": %w`, err)
- }
- query.Where(user.IDIn(edgeids...))
- neighbors, err := query.All(ctx)
- if err != nil {
- return nil, err
- }
- for _, n := range neighbors {
- nodes, ok := edges[n.ID]
- if !ok {
- return nil, fmt.Errorf(`unexpected "users" node returned %v`, n.ID)
- }
- for i := range nodes {
- nodes[i].Edges.Users = append(nodes[i].Edges.Users, n)
- }
- }
- }
- return nodes, nil
- }
- func (gq *GroupQuery) sqlCount(ctx context.Context) (int, error) {
- _spec := gq.querySpec()
- _spec.Node.Columns = gq.fields
- if len(gq.fields) > 0 {
- _spec.Unique = gq.unique != nil && *gq.unique
- }
- return sqlgraph.CountNodes(ctx, gq.driver, _spec)
- }
- func (gq *GroupQuery) sqlExist(ctx context.Context) (bool, error) {
- n, err := gq.sqlCount(ctx)
- if err != nil {
- return false, fmt.Errorf("ent: check existence: %w", err)
- }
- return n > 0, nil
- }
- func (gq *GroupQuery) querySpec() *sqlgraph.QuerySpec {
- _spec := &sqlgraph.QuerySpec{
- Node: &sqlgraph.NodeSpec{
- Table: group.Table,
- Columns: group.Columns,
- ID: &sqlgraph.FieldSpec{
- Type: field.TypeInt,
- Column: group.FieldID,
- },
- },
- From: gq.sql,
- Unique: true,
- }
- if unique := gq.unique; unique != nil {
- _spec.Unique = *unique
- }
- if fields := gq.fields; len(fields) > 0 {
- _spec.Node.Columns = make([]string, 0, len(fields))
- _spec.Node.Columns = append(_spec.Node.Columns, group.FieldID)
- for i := range fields {
- if fields[i] != group.FieldID {
- _spec.Node.Columns = append(_spec.Node.Columns, fields[i])
- }
- }
- }
- if ps := gq.predicates; len(ps) > 0 {
- _spec.Predicate = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- if limit := gq.limit; limit != nil {
- _spec.Limit = *limit
- }
- if offset := gq.offset; offset != nil {
- _spec.Offset = *offset
- }
- if ps := gq.order; len(ps) > 0 {
- _spec.Order = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- return _spec
- }
- func (gq *GroupQuery) sqlQuery(ctx context.Context) *sql.Selector {
- builder := sql.Dialect(gq.driver.Dialect())
- t1 := builder.Table(group.Table)
- columns := gq.fields
- if len(columns) == 0 {
- columns = group.Columns
- }
- selector := builder.Select(t1.Columns(columns...)...).From(t1)
- if gq.sql != nil {
- selector = gq.sql
- selector.Select(selector.Columns(columns...)...)
- }
- if gq.unique != nil && *gq.unique {
- selector.Distinct()
- }
- for _, p := range gq.predicates {
- p(selector)
- }
- for _, p := range gq.order {
- p(selector)
- }
- if offset := gq.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 := gq.limit; limit != nil {
- selector.Limit(*limit)
- }
- return selector
- }
- // GroupGroupBy is the group-by builder for Group entities.
- type GroupGroupBy 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 (ggb *GroupGroupBy) Aggregate(fns ...AggregateFunc) *GroupGroupBy {
- ggb.fns = append(ggb.fns, fns...)
- return ggb
- }
- // Scan applies the group-by query and scans the result into the given value.
- func (ggb *GroupGroupBy) Scan(ctx context.Context, v interface{}) error {
- query, err := ggb.path(ctx)
- if err != nil {
- return err
- }
- ggb.sql = query
- return ggb.sqlScan(ctx, v)
- }
- // ScanX is like Scan, but panics if an error occurs.
- func (ggb *GroupGroupBy) ScanX(ctx context.Context, v interface{}) {
- if err := ggb.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 (ggb *GroupGroupBy) Strings(ctx context.Context) ([]string, error) {
- if len(ggb.fields) > 1 {
- return nil, errors.New("ent: GroupGroupBy.Strings is not achievable when grouping more than 1 field")
- }
- var v []string
- if err := ggb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // StringsX is like Strings, but panics if an error occurs.
- func (ggb *GroupGroupBy) StringsX(ctx context.Context) []string {
- v, err := ggb.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 (ggb *GroupGroupBy) String(ctx context.Context) (_ string, err error) {
- var v []string
- if v, err = ggb.Strings(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupGroupBy.Strings returned %d results when one was expected", len(v))
- }
- return
- }
- // StringX is like String, but panics if an error occurs.
- func (ggb *GroupGroupBy) StringX(ctx context.Context) string {
- v, err := ggb.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 (ggb *GroupGroupBy) Ints(ctx context.Context) ([]int, error) {
- if len(ggb.fields) > 1 {
- return nil, errors.New("ent: GroupGroupBy.Ints is not achievable when grouping more than 1 field")
- }
- var v []int
- if err := ggb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // IntsX is like Ints, but panics if an error occurs.
- func (ggb *GroupGroupBy) IntsX(ctx context.Context) []int {
- v, err := ggb.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 (ggb *GroupGroupBy) Int(ctx context.Context) (_ int, err error) {
- var v []int
- if v, err = ggb.Ints(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupGroupBy.Ints returned %d results when one was expected", len(v))
- }
- return
- }
- // IntX is like Int, but panics if an error occurs.
- func (ggb *GroupGroupBy) IntX(ctx context.Context) int {
- v, err := ggb.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 (ggb *GroupGroupBy) Float64s(ctx context.Context) ([]float64, error) {
- if len(ggb.fields) > 1 {
- return nil, errors.New("ent: GroupGroupBy.Float64s is not achievable when grouping more than 1 field")
- }
- var v []float64
- if err := ggb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // Float64sX is like Float64s, but panics if an error occurs.
- func (ggb *GroupGroupBy) Float64sX(ctx context.Context) []float64 {
- v, err := ggb.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 (ggb *GroupGroupBy) Float64(ctx context.Context) (_ float64, err error) {
- var v []float64
- if v, err = ggb.Float64s(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupGroupBy.Float64s returned %d results when one was expected", len(v))
- }
- return
- }
- // Float64X is like Float64, but panics if an error occurs.
- func (ggb *GroupGroupBy) Float64X(ctx context.Context) float64 {
- v, err := ggb.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 (ggb *GroupGroupBy) Bools(ctx context.Context) ([]bool, error) {
- if len(ggb.fields) > 1 {
- return nil, errors.New("ent: GroupGroupBy.Bools is not achievable when grouping more than 1 field")
- }
- var v []bool
- if err := ggb.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // BoolsX is like Bools, but panics if an error occurs.
- func (ggb *GroupGroupBy) BoolsX(ctx context.Context) []bool {
- v, err := ggb.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 (ggb *GroupGroupBy) Bool(ctx context.Context) (_ bool, err error) {
- var v []bool
- if v, err = ggb.Bools(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupGroupBy.Bools returned %d results when one was expected", len(v))
- }
- return
- }
- // BoolX is like Bool, but panics if an error occurs.
- func (ggb *GroupGroupBy) BoolX(ctx context.Context) bool {
- v, err := ggb.Bool(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- func (ggb *GroupGroupBy) sqlScan(ctx context.Context, v interface{}) error {
- for _, f := range ggb.fields {
- if !group.ValidColumn(f) {
- return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)}
- }
- }
- selector := ggb.sqlQuery()
- if err := selector.Err(); err != nil {
- return err
- }
- rows := &sql.Rows{}
- query, args := selector.Query()
- if err := ggb.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
- func (ggb *GroupGroupBy) sqlQuery() *sql.Selector {
- selector := ggb.sql.Select()
- aggregation := make([]string, 0, len(ggb.fns))
- for _, fn := range ggb.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(ggb.fields)+len(ggb.fns))
- for _, f := range ggb.fields {
- columns = append(columns, selector.C(f))
- }
- columns = append(columns, aggregation...)
- selector.Select(columns...)
- }
- return selector.GroupBy(selector.Columns(ggb.fields...)...)
- }
- // GroupSelect is the builder for selecting fields of Group entities.
- type GroupSelect struct {
- *GroupQuery
- // intermediate query (i.e. traversal path).
- sql *sql.Selector
- }
- // Scan applies the selector query and scans the result into the given value.
- func (gs *GroupSelect) Scan(ctx context.Context, v interface{}) error {
- if err := gs.prepareQuery(ctx); err != nil {
- return err
- }
- gs.sql = gs.GroupQuery.sqlQuery(ctx)
- return gs.sqlScan(ctx, v)
- }
- // ScanX is like Scan, but panics if an error occurs.
- func (gs *GroupSelect) ScanX(ctx context.Context, v interface{}) {
- if err := gs.Scan(ctx, v); err != nil {
- panic(err)
- }
- }
- // Strings returns list of strings from a selector. It is only allowed when selecting one field.
- func (gs *GroupSelect) Strings(ctx context.Context) ([]string, error) {
- if len(gs.fields) > 1 {
- return nil, errors.New("ent: GroupSelect.Strings is not achievable when selecting more than 1 field")
- }
- var v []string
- if err := gs.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // StringsX is like Strings, but panics if an error occurs.
- func (gs *GroupSelect) StringsX(ctx context.Context) []string {
- v, err := gs.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 (gs *GroupSelect) String(ctx context.Context) (_ string, err error) {
- var v []string
- if v, err = gs.Strings(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupSelect.Strings returned %d results when one was expected", len(v))
- }
- return
- }
- // StringX is like String, but panics if an error occurs.
- func (gs *GroupSelect) StringX(ctx context.Context) string {
- v, err := gs.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 (gs *GroupSelect) Ints(ctx context.Context) ([]int, error) {
- if len(gs.fields) > 1 {
- return nil, errors.New("ent: GroupSelect.Ints is not achievable when selecting more than 1 field")
- }
- var v []int
- if err := gs.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // IntsX is like Ints, but panics if an error occurs.
- func (gs *GroupSelect) IntsX(ctx context.Context) []int {
- v, err := gs.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 (gs *GroupSelect) Int(ctx context.Context) (_ int, err error) {
- var v []int
- if v, err = gs.Ints(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupSelect.Ints returned %d results when one was expected", len(v))
- }
- return
- }
- // IntX is like Int, but panics if an error occurs.
- func (gs *GroupSelect) IntX(ctx context.Context) int {
- v, err := gs.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 (gs *GroupSelect) Float64s(ctx context.Context) ([]float64, error) {
- if len(gs.fields) > 1 {
- return nil, errors.New("ent: GroupSelect.Float64s is not achievable when selecting more than 1 field")
- }
- var v []float64
- if err := gs.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // Float64sX is like Float64s, but panics if an error occurs.
- func (gs *GroupSelect) Float64sX(ctx context.Context) []float64 {
- v, err := gs.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 (gs *GroupSelect) Float64(ctx context.Context) (_ float64, err error) {
- var v []float64
- if v, err = gs.Float64s(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupSelect.Float64s returned %d results when one was expected", len(v))
- }
- return
- }
- // Float64X is like Float64, but panics if an error occurs.
- func (gs *GroupSelect) Float64X(ctx context.Context) float64 {
- v, err := gs.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 (gs *GroupSelect) Bools(ctx context.Context) ([]bool, error) {
- if len(gs.fields) > 1 {
- return nil, errors.New("ent: GroupSelect.Bools is not achievable when selecting more than 1 field")
- }
- var v []bool
- if err := gs.Scan(ctx, &v); err != nil {
- return nil, err
- }
- return v, nil
- }
- // BoolsX is like Bools, but panics if an error occurs.
- func (gs *GroupSelect) BoolsX(ctx context.Context) []bool {
- v, err := gs.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 (gs *GroupSelect) Bool(ctx context.Context) (_ bool, err error) {
- var v []bool
- if v, err = gs.Bools(ctx); err != nil {
- return
- }
- switch len(v) {
- case 1:
- return v[0], nil
- case 0:
- err = &NotFoundError{group.Label}
- default:
- err = fmt.Errorf("ent: GroupSelect.Bools returned %d results when one was expected", len(v))
- }
- return
- }
- // BoolX is like Bool, but panics if an error occurs.
- func (gs *GroupSelect) BoolX(ctx context.Context) bool {
- v, err := gs.Bool(ctx)
- if err != nil {
- panic(err)
- }
- return v
- }
- func (gs *GroupSelect) sqlScan(ctx context.Context, v interface{}) error {
- rows := &sql.Rows{}
- query, args := gs.sql.Query()
- if err := gs.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
|