| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232 |
- // Code generated by ent, DO NOT EDIT.
- package ent
- import (
- "context"
- "database/sql/driver"
- "fmt"
- "math"
- "entgo.io/ent"
- "entgo.io/ent/dialect"
- "entgo.io/ent/dialect/sql"
- "entgo.io/ent/dialect/sql/sqlgraph"
- "entgo.io/ent/schema/field"
- "github.com/Wei-Shaw/sub2api/ent/account"
- "github.com/Wei-Shaw/sub2api/ent/accountgroup"
- "github.com/Wei-Shaw/sub2api/ent/apikey"
- "github.com/Wei-Shaw/sub2api/ent/group"
- "github.com/Wei-Shaw/sub2api/ent/predicate"
- "github.com/Wei-Shaw/sub2api/ent/redeemcode"
- "github.com/Wei-Shaw/sub2api/ent/usagelog"
- "github.com/Wei-Shaw/sub2api/ent/user"
- "github.com/Wei-Shaw/sub2api/ent/userallowedgroup"
- "github.com/Wei-Shaw/sub2api/ent/usersubscription"
- )
- // GroupQuery is the builder for querying Group entities.
- type GroupQuery struct {
- config
- ctx *QueryContext
- order []group.OrderOption
- inters []Interceptor
- predicates []predicate.Group
- withAPIKeys *APIKeyQuery
- withRedeemCodes *RedeemCodeQuery
- withSubscriptions *UserSubscriptionQuery
- withUsageLogs *UsageLogQuery
- withAccounts *AccountQuery
- withAllowedUsers *UserQuery
- withAccountGroups *AccountGroupQuery
- withUserAllowedGroups *UserAllowedGroupQuery
- modifiers []func(*sql.Selector)
- // 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 (_q *GroupQuery) Where(ps ...predicate.Group) *GroupQuery {
- _q.predicates = append(_q.predicates, ps...)
- return _q
- }
- // Limit the number of records to be returned by this query.
- func (_q *GroupQuery) Limit(limit int) *GroupQuery {
- _q.ctx.Limit = &limit
- return _q
- }
- // Offset to start from.
- func (_q *GroupQuery) Offset(offset int) *GroupQuery {
- _q.ctx.Offset = &offset
- return _q
- }
- // 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 (_q *GroupQuery) Unique(unique bool) *GroupQuery {
- _q.ctx.Unique = &unique
- return _q
- }
- // Order specifies how the records should be ordered.
- func (_q *GroupQuery) Order(o ...group.OrderOption) *GroupQuery {
- _q.order = append(_q.order, o...)
- return _q
- }
- // QueryAPIKeys chains the current query on the "api_keys" edge.
- func (_q *GroupQuery) QueryAPIKeys() *APIKeyQuery {
- query := (&APIKeyClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(apikey.Table, apikey.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, false, group.APIKeysTable, group.APIKeysColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryRedeemCodes chains the current query on the "redeem_codes" edge.
- func (_q *GroupQuery) QueryRedeemCodes() *RedeemCodeQuery {
- query := (&RedeemCodeClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(redeemcode.Table, redeemcode.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, false, group.RedeemCodesTable, group.RedeemCodesColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QuerySubscriptions chains the current query on the "subscriptions" edge.
- func (_q *GroupQuery) QuerySubscriptions() *UserSubscriptionQuery {
- query := (&UserSubscriptionClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(usersubscription.Table, usersubscription.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, false, group.SubscriptionsTable, group.SubscriptionsColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryUsageLogs chains the current query on the "usage_logs" edge.
- func (_q *GroupQuery) QueryUsageLogs() *UsageLogQuery {
- query := (&UsageLogClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(usagelog.Table, usagelog.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, false, group.UsageLogsTable, group.UsageLogsColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryAccounts chains the current query on the "accounts" edge.
- func (_q *GroupQuery) QueryAccounts() *AccountQuery {
- query := (&AccountClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(account.Table, account.FieldID),
- sqlgraph.Edge(sqlgraph.M2M, true, group.AccountsTable, group.AccountsPrimaryKey...),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryAllowedUsers chains the current query on the "allowed_users" edge.
- func (_q *GroupQuery) QueryAllowedUsers() *UserQuery {
- query := (&UserClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.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, true, group.AllowedUsersTable, group.AllowedUsersPrimaryKey...),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryAccountGroups chains the current query on the "account_groups" edge.
- func (_q *GroupQuery) QueryAccountGroups() *AccountGroupQuery {
- query := (&AccountGroupClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(accountgroup.Table, accountgroup.GroupColumn),
- sqlgraph.Edge(sqlgraph.O2M, true, group.AccountGroupsTable, group.AccountGroupsColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryUserAllowedGroups chains the current query on the "user_allowed_groups" edge.
- func (_q *GroupQuery) QueryUserAllowedGroups() *UserAllowedGroupQuery {
- query := (&UserAllowedGroupClient{config: _q.config}).Query()
- query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- selector := _q.sqlQuery(ctx)
- if err := selector.Err(); err != nil {
- return nil, err
- }
- step := sqlgraph.NewStep(
- sqlgraph.From(group.Table, group.FieldID, selector),
- sqlgraph.To(userallowedgroup.Table, userallowedgroup.GroupColumn),
- sqlgraph.Edge(sqlgraph.O2M, true, group.UserAllowedGroupsTable, group.UserAllowedGroupsColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.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 (_q *GroupQuery) First(ctx context.Context) (*Group, error) {
- nodes, err := _q.Limit(1).All(setContextOp(ctx, _q.ctx, ent.OpQueryFirst))
- 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 (_q *GroupQuery) FirstX(ctx context.Context) *Group {
- node, err := _q.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 (_q *GroupQuery) FirstID(ctx context.Context) (id int64, err error) {
- var ids []int64
- if ids, err = _q.Limit(1).IDs(setContextOp(ctx, _q.ctx, ent.OpQueryFirstID)); 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 (_q *GroupQuery) FirstIDX(ctx context.Context) int64 {
- id, err := _q.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 (_q *GroupQuery) Only(ctx context.Context) (*Group, error) {
- nodes, err := _q.Limit(2).All(setContextOp(ctx, _q.ctx, ent.OpQueryOnly))
- 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 (_q *GroupQuery) OnlyX(ctx context.Context) *Group {
- node, err := _q.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 (_q *GroupQuery) OnlyID(ctx context.Context) (id int64, err error) {
- var ids []int64
- if ids, err = _q.Limit(2).IDs(setContextOp(ctx, _q.ctx, ent.OpQueryOnlyID)); 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 (_q *GroupQuery) OnlyIDX(ctx context.Context) int64 {
- id, err := _q.OnlyID(ctx)
- if err != nil {
- panic(err)
- }
- return id
- }
- // All executes the query and returns a list of Groups.
- func (_q *GroupQuery) All(ctx context.Context) ([]*Group, error) {
- ctx = setContextOp(ctx, _q.ctx, ent.OpQueryAll)
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- qr := querierAll[[]*Group, *GroupQuery]()
- return withInterceptors[[]*Group](ctx, _q, qr, _q.inters)
- }
- // AllX is like All, but panics if an error occurs.
- func (_q *GroupQuery) AllX(ctx context.Context) []*Group {
- nodes, err := _q.All(ctx)
- if err != nil {
- panic(err)
- }
- return nodes
- }
- // IDs executes the query and returns a list of Group IDs.
- func (_q *GroupQuery) IDs(ctx context.Context) (ids []int64, err error) {
- if _q.ctx.Unique == nil && _q.path != nil {
- _q.Unique(true)
- }
- ctx = setContextOp(ctx, _q.ctx, ent.OpQueryIDs)
- if err = _q.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 (_q *GroupQuery) IDsX(ctx context.Context) []int64 {
- ids, err := _q.IDs(ctx)
- if err != nil {
- panic(err)
- }
- return ids
- }
- // Count returns the count of the given query.
- func (_q *GroupQuery) Count(ctx context.Context) (int, error) {
- ctx = setContextOp(ctx, _q.ctx, ent.OpQueryCount)
- if err := _q.prepareQuery(ctx); err != nil {
- return 0, err
- }
- return withInterceptors[int](ctx, _q, querierCount[*GroupQuery](), _q.inters)
- }
- // CountX is like Count, but panics if an error occurs.
- func (_q *GroupQuery) CountX(ctx context.Context) int {
- count, err := _q.Count(ctx)
- if err != nil {
- panic(err)
- }
- return count
- }
- // Exist returns true if the query has elements in the graph.
- func (_q *GroupQuery) Exist(ctx context.Context) (bool, error) {
- ctx = setContextOp(ctx, _q.ctx, ent.OpQueryExist)
- switch _, err := _q.FirstID(ctx); {
- case IsNotFound(err):
- return false, nil
- case err != nil:
- return false, fmt.Errorf("ent: check existence: %w", err)
- default:
- return true, nil
- }
- }
- // ExistX is like Exist, but panics if an error occurs.
- func (_q *GroupQuery) ExistX(ctx context.Context) bool {
- exist, err := _q.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 (_q *GroupQuery) Clone() *GroupQuery {
- if _q == nil {
- return nil
- }
- return &GroupQuery{
- config: _q.config,
- ctx: _q.ctx.Clone(),
- order: append([]group.OrderOption{}, _q.order...),
- inters: append([]Interceptor{}, _q.inters...),
- predicates: append([]predicate.Group{}, _q.predicates...),
- withAPIKeys: _q.withAPIKeys.Clone(),
- withRedeemCodes: _q.withRedeemCodes.Clone(),
- withSubscriptions: _q.withSubscriptions.Clone(),
- withUsageLogs: _q.withUsageLogs.Clone(),
- withAccounts: _q.withAccounts.Clone(),
- withAllowedUsers: _q.withAllowedUsers.Clone(),
- withAccountGroups: _q.withAccountGroups.Clone(),
- withUserAllowedGroups: _q.withUserAllowedGroups.Clone(),
- // clone intermediate query.
- sql: _q.sql.Clone(),
- path: _q.path,
- }
- }
- // WithAPIKeys tells the query-builder to eager-load the nodes that are connected to
- // the "api_keys" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithAPIKeys(opts ...func(*APIKeyQuery)) *GroupQuery {
- query := (&APIKeyClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withAPIKeys = query
- return _q
- }
- // WithRedeemCodes tells the query-builder to eager-load the nodes that are connected to
- // the "redeem_codes" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithRedeemCodes(opts ...func(*RedeemCodeQuery)) *GroupQuery {
- query := (&RedeemCodeClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withRedeemCodes = query
- return _q
- }
- // WithSubscriptions tells the query-builder to eager-load the nodes that are connected to
- // the "subscriptions" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithSubscriptions(opts ...func(*UserSubscriptionQuery)) *GroupQuery {
- query := (&UserSubscriptionClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withSubscriptions = query
- return _q
- }
- // WithUsageLogs tells the query-builder to eager-load the nodes that are connected to
- // the "usage_logs" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithUsageLogs(opts ...func(*UsageLogQuery)) *GroupQuery {
- query := (&UsageLogClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withUsageLogs = query
- return _q
- }
- // WithAccounts tells the query-builder to eager-load the nodes that are connected to
- // the "accounts" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithAccounts(opts ...func(*AccountQuery)) *GroupQuery {
- query := (&AccountClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withAccounts = query
- return _q
- }
- // WithAllowedUsers tells the query-builder to eager-load the nodes that are connected to
- // the "allowed_users" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithAllowedUsers(opts ...func(*UserQuery)) *GroupQuery {
- query := (&UserClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withAllowedUsers = query
- return _q
- }
- // WithAccountGroups tells the query-builder to eager-load the nodes that are connected to
- // the "account_groups" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithAccountGroups(opts ...func(*AccountGroupQuery)) *GroupQuery {
- query := (&AccountGroupClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withAccountGroups = query
- return _q
- }
- // WithUserAllowedGroups tells the query-builder to eager-load the nodes that are connected to
- // the "user_allowed_groups" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *GroupQuery) WithUserAllowedGroups(opts ...func(*UserAllowedGroupQuery)) *GroupQuery {
- query := (&UserAllowedGroupClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withUserAllowedGroups = query
- return _q
- }
- // 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 {
- // CreatedAt time.Time `json:"created_at,omitempty"`
- // Count int `json:"count,omitempty"`
- // }
- //
- // client.Group.Query().
- // GroupBy(group.FieldCreatedAt).
- // Aggregate(ent.Count()).
- // Scan(ctx, &v)
- func (_q *GroupQuery) GroupBy(field string, fields ...string) *GroupGroupBy {
- _q.ctx.Fields = append([]string{field}, fields...)
- grbuild := &GroupGroupBy{build: _q}
- grbuild.flds = &_q.ctx.Fields
- grbuild.label = group.Label
- grbuild.scan = grbuild.Scan
- return grbuild
- }
- // 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 {
- // CreatedAt time.Time `json:"created_at,omitempty"`
- // }
- //
- // client.Group.Query().
- // Select(group.FieldCreatedAt).
- // Scan(ctx, &v)
- func (_q *GroupQuery) Select(fields ...string) *GroupSelect {
- _q.ctx.Fields = append(_q.ctx.Fields, fields...)
- sbuild := &GroupSelect{GroupQuery: _q}
- sbuild.label = group.Label
- sbuild.flds, sbuild.scan = &_q.ctx.Fields, sbuild.Scan
- return sbuild
- }
- // Aggregate returns a GroupSelect configured with the given aggregations.
- func (_q *GroupQuery) Aggregate(fns ...AggregateFunc) *GroupSelect {
- return _q.Select().Aggregate(fns...)
- }
- func (_q *GroupQuery) prepareQuery(ctx context.Context) error {
- for _, inter := range _q.inters {
- if inter == nil {
- return fmt.Errorf("ent: uninitialized interceptor (forgotten import ent/runtime?)")
- }
- if trv, ok := inter.(Traverser); ok {
- if err := trv.Traverse(ctx, _q); err != nil {
- return err
- }
- }
- }
- for _, f := range _q.ctx.Fields {
- if !group.ValidColumn(f) {
- return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
- }
- }
- if _q.path != nil {
- prev, err := _q.path(ctx)
- if err != nil {
- return err
- }
- _q.sql = prev
- }
- return nil
- }
- func (_q *GroupQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Group, error) {
- var (
- nodes = []*Group{}
- _spec = _q.querySpec()
- loadedTypes = [8]bool{
- _q.withAPIKeys != nil,
- _q.withRedeemCodes != nil,
- _q.withSubscriptions != nil,
- _q.withUsageLogs != nil,
- _q.withAccounts != nil,
- _q.withAllowedUsers != nil,
- _q.withAccountGroups != nil,
- _q.withUserAllowedGroups != nil,
- }
- )
- _spec.ScanValues = func(columns []string) ([]any, error) {
- return (*Group).scanValues(nil, columns)
- }
- _spec.Assign = func(columns []string, values []any) error {
- node := &Group{config: _q.config}
- nodes = append(nodes, node)
- node.Edges.loadedTypes = loadedTypes
- return node.assignValues(columns, values)
- }
- if len(_q.modifiers) > 0 {
- _spec.Modifiers = _q.modifiers
- }
- for i := range hooks {
- hooks[i](ctx, _spec)
- }
- if err := sqlgraph.QueryNodes(ctx, _q.driver, _spec); err != nil {
- return nil, err
- }
- if len(nodes) == 0 {
- return nodes, nil
- }
- if query := _q.withAPIKeys; query != nil {
- if err := _q.loadAPIKeys(ctx, query, nodes,
- func(n *Group) { n.Edges.APIKeys = []*APIKey{} },
- func(n *Group, e *APIKey) { n.Edges.APIKeys = append(n.Edges.APIKeys, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withRedeemCodes; query != nil {
- if err := _q.loadRedeemCodes(ctx, query, nodes,
- func(n *Group) { n.Edges.RedeemCodes = []*RedeemCode{} },
- func(n *Group, e *RedeemCode) { n.Edges.RedeemCodes = append(n.Edges.RedeemCodes, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withSubscriptions; query != nil {
- if err := _q.loadSubscriptions(ctx, query, nodes,
- func(n *Group) { n.Edges.Subscriptions = []*UserSubscription{} },
- func(n *Group, e *UserSubscription) { n.Edges.Subscriptions = append(n.Edges.Subscriptions, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withUsageLogs; query != nil {
- if err := _q.loadUsageLogs(ctx, query, nodes,
- func(n *Group) { n.Edges.UsageLogs = []*UsageLog{} },
- func(n *Group, e *UsageLog) { n.Edges.UsageLogs = append(n.Edges.UsageLogs, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withAccounts; query != nil {
- if err := _q.loadAccounts(ctx, query, nodes,
- func(n *Group) { n.Edges.Accounts = []*Account{} },
- func(n *Group, e *Account) { n.Edges.Accounts = append(n.Edges.Accounts, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withAllowedUsers; query != nil {
- if err := _q.loadAllowedUsers(ctx, query, nodes,
- func(n *Group) { n.Edges.AllowedUsers = []*User{} },
- func(n *Group, e *User) { n.Edges.AllowedUsers = append(n.Edges.AllowedUsers, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withAccountGroups; query != nil {
- if err := _q.loadAccountGroups(ctx, query, nodes,
- func(n *Group) { n.Edges.AccountGroups = []*AccountGroup{} },
- func(n *Group, e *AccountGroup) { n.Edges.AccountGroups = append(n.Edges.AccountGroups, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withUserAllowedGroups; query != nil {
- if err := _q.loadUserAllowedGroups(ctx, query, nodes,
- func(n *Group) { n.Edges.UserAllowedGroups = []*UserAllowedGroup{} },
- func(n *Group, e *UserAllowedGroup) { n.Edges.UserAllowedGroups = append(n.Edges.UserAllowedGroups, e) }); err != nil {
- return nil, err
- }
- }
- return nodes, nil
- }
- func (_q *GroupQuery) loadAPIKeys(ctx context.Context, query *APIKeyQuery, nodes []*Group, init func(*Group), assign func(*Group, *APIKey)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(apikey.FieldGroupID)
- }
- query.Where(predicate.APIKey(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.APIKeysColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- if fk == nil {
- return fmt.Errorf(`foreign-key "group_id" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, *fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) loadRedeemCodes(ctx context.Context, query *RedeemCodeQuery, nodes []*Group, init func(*Group), assign func(*Group, *RedeemCode)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(redeemcode.FieldGroupID)
- }
- query.Where(predicate.RedeemCode(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.RedeemCodesColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- if fk == nil {
- return fmt.Errorf(`foreign-key "group_id" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, *fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) loadSubscriptions(ctx context.Context, query *UserSubscriptionQuery, nodes []*Group, init func(*Group), assign func(*Group, *UserSubscription)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(usersubscription.FieldGroupID)
- }
- query.Where(predicate.UserSubscription(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.SubscriptionsColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- node, ok := nodeids[fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) loadUsageLogs(ctx context.Context, query *UsageLogQuery, nodes []*Group, init func(*Group), assign func(*Group, *UsageLog)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(usagelog.FieldGroupID)
- }
- query.Where(predicate.UsageLog(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.UsageLogsColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- if fk == nil {
- return fmt.Errorf(`foreign-key "group_id" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, *fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) loadAccounts(ctx context.Context, query *AccountQuery, nodes []*Group, init func(*Group), assign func(*Group, *Account)) error {
- edgeIDs := make([]driver.Value, len(nodes))
- byID := make(map[int64]*Group)
- nids := make(map[int64]map[*Group]struct{})
- for i, node := range nodes {
- edgeIDs[i] = node.ID
- byID[node.ID] = node
- if init != nil {
- init(node)
- }
- }
- query.Where(func(s *sql.Selector) {
- joinT := sql.Table(group.AccountsTable)
- s.Join(joinT).On(s.C(account.FieldID), joinT.C(group.AccountsPrimaryKey[0]))
- s.Where(sql.InValues(joinT.C(group.AccountsPrimaryKey[1]), edgeIDs...))
- columns := s.SelectedColumns()
- s.Select(joinT.C(group.AccountsPrimaryKey[1]))
- s.AppendSelect(columns...)
- s.SetDistinct(false)
- })
- if err := query.prepareQuery(ctx); err != nil {
- return err
- }
- qr := QuerierFunc(func(ctx context.Context, q Query) (Value, error) {
- return query.sqlAll(ctx, func(_ context.Context, spec *sqlgraph.QuerySpec) {
- assign := spec.Assign
- values := spec.ScanValues
- spec.ScanValues = func(columns []string) ([]any, error) {
- values, err := values(columns[1:])
- if err != nil {
- return nil, err
- }
- return append([]any{new(sql.NullInt64)}, values...), nil
- }
- spec.Assign = func(columns []string, values []any) error {
- outValue := values[0].(*sql.NullInt64).Int64
- inValue := values[1].(*sql.NullInt64).Int64
- if nids[inValue] == nil {
- nids[inValue] = map[*Group]struct{}{byID[outValue]: {}}
- return assign(columns[1:], values[1:])
- }
- nids[inValue][byID[outValue]] = struct{}{}
- return nil
- }
- })
- })
- neighbors, err := withInterceptors[[]*Account](ctx, query, qr, query.inters)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- nodes, ok := nids[n.ID]
- if !ok {
- return fmt.Errorf(`unexpected "accounts" node returned %v`, n.ID)
- }
- for kn := range nodes {
- assign(kn, n)
- }
- }
- return nil
- }
- func (_q *GroupQuery) loadAllowedUsers(ctx context.Context, query *UserQuery, nodes []*Group, init func(*Group), assign func(*Group, *User)) error {
- edgeIDs := make([]driver.Value, len(nodes))
- byID := make(map[int64]*Group)
- nids := make(map[int64]map[*Group]struct{})
- for i, node := range nodes {
- edgeIDs[i] = node.ID
- byID[node.ID] = node
- if init != nil {
- init(node)
- }
- }
- query.Where(func(s *sql.Selector) {
- joinT := sql.Table(group.AllowedUsersTable)
- s.Join(joinT).On(s.C(user.FieldID), joinT.C(group.AllowedUsersPrimaryKey[0]))
- s.Where(sql.InValues(joinT.C(group.AllowedUsersPrimaryKey[1]), edgeIDs...))
- columns := s.SelectedColumns()
- s.Select(joinT.C(group.AllowedUsersPrimaryKey[1]))
- s.AppendSelect(columns...)
- s.SetDistinct(false)
- })
- if err := query.prepareQuery(ctx); err != nil {
- return err
- }
- qr := QuerierFunc(func(ctx context.Context, q Query) (Value, error) {
- return query.sqlAll(ctx, func(_ context.Context, spec *sqlgraph.QuerySpec) {
- assign := spec.Assign
- values := spec.ScanValues
- spec.ScanValues = func(columns []string) ([]any, error) {
- values, err := values(columns[1:])
- if err != nil {
- return nil, err
- }
- return append([]any{new(sql.NullInt64)}, values...), nil
- }
- spec.Assign = func(columns []string, values []any) error {
- outValue := values[0].(*sql.NullInt64).Int64
- inValue := values[1].(*sql.NullInt64).Int64
- if nids[inValue] == nil {
- nids[inValue] = map[*Group]struct{}{byID[outValue]: {}}
- return assign(columns[1:], values[1:])
- }
- nids[inValue][byID[outValue]] = struct{}{}
- return nil
- }
- })
- })
- neighbors, err := withInterceptors[[]*User](ctx, query, qr, query.inters)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- nodes, ok := nids[n.ID]
- if !ok {
- return fmt.Errorf(`unexpected "allowed_users" node returned %v`, n.ID)
- }
- for kn := range nodes {
- assign(kn, n)
- }
- }
- return nil
- }
- func (_q *GroupQuery) loadAccountGroups(ctx context.Context, query *AccountGroupQuery, nodes []*Group, init func(*Group), assign func(*Group, *AccountGroup)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(accountgroup.FieldGroupID)
- }
- query.Where(predicate.AccountGroup(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.AccountGroupsColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- node, ok := nodeids[fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, fk, n)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) loadUserAllowedGroups(ctx context.Context, query *UserAllowedGroupQuery, nodes []*Group, init func(*Group), assign func(*Group, *UserAllowedGroup)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Group)
- for i := range nodes {
- fks = append(fks, nodes[i].ID)
- nodeids[nodes[i].ID] = nodes[i]
- if init != nil {
- init(nodes[i])
- }
- }
- if len(query.ctx.Fields) > 0 {
- query.ctx.AppendFieldOnce(userallowedgroup.FieldGroupID)
- }
- query.Where(predicate.UserAllowedGroup(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(group.UserAllowedGroupsColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.GroupID
- node, ok := nodeids[fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "group_id" returned %v for node %v`, fk, n)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *GroupQuery) sqlCount(ctx context.Context) (int, error) {
- _spec := _q.querySpec()
- if len(_q.modifiers) > 0 {
- _spec.Modifiers = _q.modifiers
- }
- _spec.Node.Columns = _q.ctx.Fields
- if len(_q.ctx.Fields) > 0 {
- _spec.Unique = _q.ctx.Unique != nil && *_q.ctx.Unique
- }
- return sqlgraph.CountNodes(ctx, _q.driver, _spec)
- }
- func (_q *GroupQuery) querySpec() *sqlgraph.QuerySpec {
- _spec := sqlgraph.NewQuerySpec(group.Table, group.Columns, sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64))
- _spec.From = _q.sql
- if unique := _q.ctx.Unique; unique != nil {
- _spec.Unique = *unique
- } else if _q.path != nil {
- _spec.Unique = true
- }
- if fields := _q.ctx.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 := _q.predicates; len(ps) > 0 {
- _spec.Predicate = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- if limit := _q.ctx.Limit; limit != nil {
- _spec.Limit = *limit
- }
- if offset := _q.ctx.Offset; offset != nil {
- _spec.Offset = *offset
- }
- if ps := _q.order; len(ps) > 0 {
- _spec.Order = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- return _spec
- }
- func (_q *GroupQuery) sqlQuery(ctx context.Context) *sql.Selector {
- builder := sql.Dialect(_q.driver.Dialect())
- t1 := builder.Table(group.Table)
- columns := _q.ctx.Fields
- if len(columns) == 0 {
- columns = group.Columns
- }
- selector := builder.Select(t1.Columns(columns...)...).From(t1)
- if _q.sql != nil {
- selector = _q.sql
- selector.Select(selector.Columns(columns...)...)
- }
- if _q.ctx.Unique != nil && *_q.ctx.Unique {
- selector.Distinct()
- }
- for _, m := range _q.modifiers {
- m(selector)
- }
- for _, p := range _q.predicates {
- p(selector)
- }
- for _, p := range _q.order {
- p(selector)
- }
- if offset := _q.ctx.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 := _q.ctx.Limit; limit != nil {
- selector.Limit(*limit)
- }
- return selector
- }
- // ForUpdate locks the selected rows against concurrent updates, and prevent them from being
- // updated, deleted or "selected ... for update" by other sessions, until the transaction is
- // either committed or rolled-back.
- func (_q *GroupQuery) ForUpdate(opts ...sql.LockOption) *GroupQuery {
- if _q.driver.Dialect() == dialect.Postgres {
- _q.Unique(false)
- }
- _q.modifiers = append(_q.modifiers, func(s *sql.Selector) {
- s.ForUpdate(opts...)
- })
- return _q
- }
- // ForShare behaves similarly to ForUpdate, except that it acquires a shared mode lock
- // on any rows that are read. Other sessions can read the rows, but cannot modify them
- // until your transaction commits.
- func (_q *GroupQuery) ForShare(opts ...sql.LockOption) *GroupQuery {
- if _q.driver.Dialect() == dialect.Postgres {
- _q.Unique(false)
- }
- _q.modifiers = append(_q.modifiers, func(s *sql.Selector) {
- s.ForShare(opts...)
- })
- return _q
- }
- // GroupGroupBy is the group-by builder for Group entities.
- type GroupGroupBy struct {
- selector
- build *GroupQuery
- }
- // Aggregate adds the given aggregation functions to the group-by query.
- func (_g *GroupGroupBy) Aggregate(fns ...AggregateFunc) *GroupGroupBy {
- _g.fns = append(_g.fns, fns...)
- return _g
- }
- // Scan applies the selector query and scans the result into the given value.
- func (_g *GroupGroupBy) Scan(ctx context.Context, v any) error {
- ctx = setContextOp(ctx, _g.build.ctx, ent.OpQueryGroupBy)
- if err := _g.build.prepareQuery(ctx); err != nil {
- return err
- }
- return scanWithInterceptors[*GroupQuery, *GroupGroupBy](ctx, _g.build, _g, _g.build.inters, v)
- }
- func (_g *GroupGroupBy) sqlScan(ctx context.Context, root *GroupQuery, v any) error {
- selector := root.sqlQuery(ctx).Select()
- aggregation := make([]string, 0, len(_g.fns))
- for _, fn := range _g.fns {
- aggregation = append(aggregation, fn(selector))
- }
- if len(selector.SelectedColumns()) == 0 {
- columns := make([]string, 0, len(*_g.flds)+len(_g.fns))
- for _, f := range *_g.flds {
- columns = append(columns, selector.C(f))
- }
- columns = append(columns, aggregation...)
- selector.Select(columns...)
- }
- selector.GroupBy(selector.Columns(*_g.flds...)...)
- if err := selector.Err(); err != nil {
- return err
- }
- rows := &sql.Rows{}
- query, args := selector.Query()
- if err := _g.build.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
- // GroupSelect is the builder for selecting fields of Group entities.
- type GroupSelect struct {
- *GroupQuery
- selector
- }
- // Aggregate adds the given aggregation functions to the selector query.
- func (_s *GroupSelect) Aggregate(fns ...AggregateFunc) *GroupSelect {
- _s.fns = append(_s.fns, fns...)
- return _s
- }
- // Scan applies the selector query and scans the result into the given value.
- func (_s *GroupSelect) Scan(ctx context.Context, v any) error {
- ctx = setContextOp(ctx, _s.ctx, ent.OpQuerySelect)
- if err := _s.prepareQuery(ctx); err != nil {
- return err
- }
- return scanWithInterceptors[*GroupQuery, *GroupSelect](ctx, _s.GroupQuery, _s, _s.inters, v)
- }
- func (_s *GroupSelect) sqlScan(ctx context.Context, root *GroupQuery, v any) error {
- selector := root.sqlQuery(ctx)
- aggregation := make([]string, 0, len(_s.fns))
- for _, fn := range _s.fns {
- aggregation = append(aggregation, fn(selector))
- }
- switch n := len(*_s.selector.flds); {
- case n == 0 && len(aggregation) > 0:
- selector.Select(aggregation...)
- case n != 0 && len(aggregation) > 0:
- selector.AppendSelect(aggregation...)
- }
- rows := &sql.Rows{}
- query, args := selector.Query()
- if err := _s.driver.Query(ctx, query, args, rows); err != nil {
- return err
- }
- defer rows.Close()
- return sql.ScanSlice(rows, v)
- }
|