| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799 |
- // 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/predicate"
- "github.com/Wei-Shaw/sub2api/ent/proxy"
- )
- // ProxyQuery is the builder for querying Proxy entities.
- type ProxyQuery struct {
- config
- ctx *QueryContext
- order []proxy.OrderOption
- inters []Interceptor
- predicates []predicate.Proxy
- withAccounts *AccountQuery
- withPrimaryProxies *ProxyQuery
- withBackupProxy *ProxyQuery
- 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 ProxyQuery builder.
- func (_q *ProxyQuery) Where(ps ...predicate.Proxy) *ProxyQuery {
- _q.predicates = append(_q.predicates, ps...)
- return _q
- }
- // Limit the number of records to be returned by this query.
- func (_q *ProxyQuery) Limit(limit int) *ProxyQuery {
- _q.ctx.Limit = &limit
- return _q
- }
- // Offset to start from.
- func (_q *ProxyQuery) Offset(offset int) *ProxyQuery {
- _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 *ProxyQuery) Unique(unique bool) *ProxyQuery {
- _q.ctx.Unique = &unique
- return _q
- }
- // Order specifies how the records should be ordered.
- func (_q *ProxyQuery) Order(o ...proxy.OrderOption) *ProxyQuery {
- _q.order = append(_q.order, o...)
- return _q
- }
- // QueryAccounts chains the current query on the "accounts" edge.
- func (_q *ProxyQuery) 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(proxy.Table, proxy.FieldID, selector),
- sqlgraph.To(account.Table, account.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, true, proxy.AccountsTable, proxy.AccountsColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryPrimaryProxies chains the current query on the "primary_proxies" edge.
- func (_q *ProxyQuery) QueryPrimaryProxies() *ProxyQuery {
- query := (&ProxyClient{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(proxy.Table, proxy.FieldID, selector),
- sqlgraph.To(proxy.Table, proxy.FieldID),
- sqlgraph.Edge(sqlgraph.O2M, true, proxy.PrimaryProxiesTable, proxy.PrimaryProxiesColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // QueryBackupProxy chains the current query on the "backup_proxy" edge.
- func (_q *ProxyQuery) QueryBackupProxy() *ProxyQuery {
- query := (&ProxyClient{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(proxy.Table, proxy.FieldID, selector),
- sqlgraph.To(proxy.Table, proxy.FieldID),
- sqlgraph.Edge(sqlgraph.M2O, false, proxy.BackupProxyTable, proxy.BackupProxyColumn),
- )
- fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
- return fromU, nil
- }
- return query
- }
- // First returns the first Proxy entity from the query.
- // Returns a *NotFoundError when no Proxy was found.
- func (_q *ProxyQuery) First(ctx context.Context) (*Proxy, 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{proxy.Label}
- }
- return nodes[0], nil
- }
- // FirstX is like First, but panics if an error occurs.
- func (_q *ProxyQuery) FirstX(ctx context.Context) *Proxy {
- node, err := _q.First(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return node
- }
- // FirstID returns the first Proxy ID from the query.
- // Returns a *NotFoundError when no Proxy ID was found.
- func (_q *ProxyQuery) 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{proxy.Label}
- return
- }
- return ids[0], nil
- }
- // FirstIDX is like FirstID, but panics if an error occurs.
- func (_q *ProxyQuery) FirstIDX(ctx context.Context) int64 {
- id, err := _q.FirstID(ctx)
- if err != nil && !IsNotFound(err) {
- panic(err)
- }
- return id
- }
- // Only returns a single Proxy entity found by the query, ensuring it only returns one.
- // Returns a *NotSingularError when more than one Proxy entity is found.
- // Returns a *NotFoundError when no Proxy entities are found.
- func (_q *ProxyQuery) Only(ctx context.Context) (*Proxy, 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{proxy.Label}
- default:
- return nil, &NotSingularError{proxy.Label}
- }
- }
- // OnlyX is like Only, but panics if an error occurs.
- func (_q *ProxyQuery) OnlyX(ctx context.Context) *Proxy {
- node, err := _q.Only(ctx)
- if err != nil {
- panic(err)
- }
- return node
- }
- // OnlyID is like Only, but returns the only Proxy ID in the query.
- // Returns a *NotSingularError when more than one Proxy ID is found.
- // Returns a *NotFoundError when no entities are found.
- func (_q *ProxyQuery) 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{proxy.Label}
- default:
- err = &NotSingularError{proxy.Label}
- }
- return
- }
- // OnlyIDX is like OnlyID, but panics if an error occurs.
- func (_q *ProxyQuery) 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 Proxies.
- func (_q *ProxyQuery) All(ctx context.Context) ([]*Proxy, error) {
- ctx = setContextOp(ctx, _q.ctx, ent.OpQueryAll)
- if err := _q.prepareQuery(ctx); err != nil {
- return nil, err
- }
- qr := querierAll[[]*Proxy, *ProxyQuery]()
- return withInterceptors[[]*Proxy](ctx, _q, qr, _q.inters)
- }
- // AllX is like All, but panics if an error occurs.
- func (_q *ProxyQuery) AllX(ctx context.Context) []*Proxy {
- nodes, err := _q.All(ctx)
- if err != nil {
- panic(err)
- }
- return nodes
- }
- // IDs executes the query and returns a list of Proxy IDs.
- func (_q *ProxyQuery) 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(proxy.FieldID).Scan(ctx, &ids); err != nil {
- return nil, err
- }
- return ids, nil
- }
- // IDsX is like IDs, but panics if an error occurs.
- func (_q *ProxyQuery) 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 *ProxyQuery) 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[*ProxyQuery](), _q.inters)
- }
- // CountX is like Count, but panics if an error occurs.
- func (_q *ProxyQuery) 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 *ProxyQuery) 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 *ProxyQuery) ExistX(ctx context.Context) bool {
- exist, err := _q.Exist(ctx)
- if err != nil {
- panic(err)
- }
- return exist
- }
- // Clone returns a duplicate of the ProxyQuery 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 *ProxyQuery) Clone() *ProxyQuery {
- if _q == nil {
- return nil
- }
- return &ProxyQuery{
- config: _q.config,
- ctx: _q.ctx.Clone(),
- order: append([]proxy.OrderOption{}, _q.order...),
- inters: append([]Interceptor{}, _q.inters...),
- predicates: append([]predicate.Proxy{}, _q.predicates...),
- withAccounts: _q.withAccounts.Clone(),
- withPrimaryProxies: _q.withPrimaryProxies.Clone(),
- withBackupProxy: _q.withBackupProxy.Clone(),
- // clone intermediate query.
- sql: _q.sql.Clone(),
- path: _q.path,
- }
- }
- // 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 *ProxyQuery) WithAccounts(opts ...func(*AccountQuery)) *ProxyQuery {
- query := (&AccountClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withAccounts = query
- return _q
- }
- // WithPrimaryProxies tells the query-builder to eager-load the nodes that are connected to
- // the "primary_proxies" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *ProxyQuery) WithPrimaryProxies(opts ...func(*ProxyQuery)) *ProxyQuery {
- query := (&ProxyClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withPrimaryProxies = query
- return _q
- }
- // WithBackupProxy tells the query-builder to eager-load the nodes that are connected to
- // the "backup_proxy" edge. The optional arguments are used to configure the query builder of the edge.
- func (_q *ProxyQuery) WithBackupProxy(opts ...func(*ProxyQuery)) *ProxyQuery {
- query := (&ProxyClient{config: _q.config}).Query()
- for _, opt := range opts {
- opt(query)
- }
- _q.withBackupProxy = 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.Proxy.Query().
- // GroupBy(proxy.FieldCreatedAt).
- // Aggregate(ent.Count()).
- // Scan(ctx, &v)
- func (_q *ProxyQuery) GroupBy(field string, fields ...string) *ProxyGroupBy {
- _q.ctx.Fields = append([]string{field}, fields...)
- grbuild := &ProxyGroupBy{build: _q}
- grbuild.flds = &_q.ctx.Fields
- grbuild.label = proxy.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.Proxy.Query().
- // Select(proxy.FieldCreatedAt).
- // Scan(ctx, &v)
- func (_q *ProxyQuery) Select(fields ...string) *ProxySelect {
- _q.ctx.Fields = append(_q.ctx.Fields, fields...)
- sbuild := &ProxySelect{ProxyQuery: _q}
- sbuild.label = proxy.Label
- sbuild.flds, sbuild.scan = &_q.ctx.Fields, sbuild.Scan
- return sbuild
- }
- // Aggregate returns a ProxySelect configured with the given aggregations.
- func (_q *ProxyQuery) Aggregate(fns ...AggregateFunc) *ProxySelect {
- return _q.Select().Aggregate(fns...)
- }
- func (_q *ProxyQuery) 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 !proxy.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 *ProxyQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Proxy, error) {
- var (
- nodes = []*Proxy{}
- _spec = _q.querySpec()
- loadedTypes = [3]bool{
- _q.withAccounts != nil,
- _q.withPrimaryProxies != nil,
- _q.withBackupProxy != nil,
- }
- )
- _spec.ScanValues = func(columns []string) ([]any, error) {
- return (*Proxy).scanValues(nil, columns)
- }
- _spec.Assign = func(columns []string, values []any) error {
- node := &Proxy{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.withAccounts; query != nil {
- if err := _q.loadAccounts(ctx, query, nodes,
- func(n *Proxy) { n.Edges.Accounts = []*Account{} },
- func(n *Proxy, e *Account) { n.Edges.Accounts = append(n.Edges.Accounts, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withPrimaryProxies; query != nil {
- if err := _q.loadPrimaryProxies(ctx, query, nodes,
- func(n *Proxy) { n.Edges.PrimaryProxies = []*Proxy{} },
- func(n *Proxy, e *Proxy) { n.Edges.PrimaryProxies = append(n.Edges.PrimaryProxies, e) }); err != nil {
- return nil, err
- }
- }
- if query := _q.withBackupProxy; query != nil {
- if err := _q.loadBackupProxy(ctx, query, nodes, nil,
- func(n *Proxy, e *Proxy) { n.Edges.BackupProxy = e }); err != nil {
- return nil, err
- }
- }
- return nodes, nil
- }
- func (_q *ProxyQuery) loadAccounts(ctx context.Context, query *AccountQuery, nodes []*Proxy, init func(*Proxy), assign func(*Proxy, *Account)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Proxy)
- 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(account.FieldProxyID)
- }
- query.Where(predicate.Account(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(proxy.AccountsColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.ProxyID
- if fk == nil {
- return fmt.Errorf(`foreign-key "proxy_id" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "proxy_id" returned %v for node %v`, *fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *ProxyQuery) loadPrimaryProxies(ctx context.Context, query *ProxyQuery, nodes []*Proxy, init func(*Proxy), assign func(*Proxy, *Proxy)) error {
- fks := make([]driver.Value, 0, len(nodes))
- nodeids := make(map[int64]*Proxy)
- 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(proxy.FieldBackupProxyID)
- }
- query.Where(predicate.Proxy(func(s *sql.Selector) {
- s.Where(sql.InValues(s.C(proxy.PrimaryProxiesColumn), fks...))
- }))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- fk := n.BackupProxyID
- if fk == nil {
- return fmt.Errorf(`foreign-key "backup_proxy_id" is nil for node %v`, n.ID)
- }
- node, ok := nodeids[*fk]
- if !ok {
- return fmt.Errorf(`unexpected referenced foreign-key "backup_proxy_id" returned %v for node %v`, *fk, n.ID)
- }
- assign(node, n)
- }
- return nil
- }
- func (_q *ProxyQuery) loadBackupProxy(ctx context.Context, query *ProxyQuery, nodes []*Proxy, init func(*Proxy), assign func(*Proxy, *Proxy)) error {
- ids := make([]int64, 0, len(nodes))
- nodeids := make(map[int64][]*Proxy)
- for i := range nodes {
- if nodes[i].BackupProxyID == nil {
- continue
- }
- fk := *nodes[i].BackupProxyID
- if _, ok := nodeids[fk]; !ok {
- ids = append(ids, fk)
- }
- nodeids[fk] = append(nodeids[fk], nodes[i])
- }
- if len(ids) == 0 {
- return nil
- }
- query.Where(proxy.IDIn(ids...))
- neighbors, err := query.All(ctx)
- if err != nil {
- return err
- }
- for _, n := range neighbors {
- nodes, ok := nodeids[n.ID]
- if !ok {
- return fmt.Errorf(`unexpected foreign-key "backup_proxy_id" returned %v`, n.ID)
- }
- for i := range nodes {
- assign(nodes[i], n)
- }
- }
- return nil
- }
- func (_q *ProxyQuery) 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 *ProxyQuery) querySpec() *sqlgraph.QuerySpec {
- _spec := sqlgraph.NewQuerySpec(proxy.Table, proxy.Columns, sqlgraph.NewFieldSpec(proxy.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, proxy.FieldID)
- for i := range fields {
- if fields[i] != proxy.FieldID {
- _spec.Node.Columns = append(_spec.Node.Columns, fields[i])
- }
- }
- if _q.withBackupProxy != nil {
- _spec.Node.AddColumnOnce(proxy.FieldBackupProxyID)
- }
- }
- 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 *ProxyQuery) sqlQuery(ctx context.Context) *sql.Selector {
- builder := sql.Dialect(_q.driver.Dialect())
- t1 := builder.Table(proxy.Table)
- columns := _q.ctx.Fields
- if len(columns) == 0 {
- columns = proxy.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 *ProxyQuery) ForUpdate(opts ...sql.LockOption) *ProxyQuery {
- 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 *ProxyQuery) ForShare(opts ...sql.LockOption) *ProxyQuery {
- if _q.driver.Dialect() == dialect.Postgres {
- _q.Unique(false)
- }
- _q.modifiers = append(_q.modifiers, func(s *sql.Selector) {
- s.ForShare(opts...)
- })
- return _q
- }
- // ProxyGroupBy is the group-by builder for Proxy entities.
- type ProxyGroupBy struct {
- selector
- build *ProxyQuery
- }
- // Aggregate adds the given aggregation functions to the group-by query.
- func (_g *ProxyGroupBy) Aggregate(fns ...AggregateFunc) *ProxyGroupBy {
- _g.fns = append(_g.fns, fns...)
- return _g
- }
- // Scan applies the selector query and scans the result into the given value.
- func (_g *ProxyGroupBy) 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[*ProxyQuery, *ProxyGroupBy](ctx, _g.build, _g, _g.build.inters, v)
- }
- func (_g *ProxyGroupBy) sqlScan(ctx context.Context, root *ProxyQuery, 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)
- }
- // ProxySelect is the builder for selecting fields of Proxy entities.
- type ProxySelect struct {
- *ProxyQuery
- selector
- }
- // Aggregate adds the given aggregation functions to the selector query.
- func (_s *ProxySelect) Aggregate(fns ...AggregateFunc) *ProxySelect {
- _s.fns = append(_s.fns, fns...)
- return _s
- }
- // Scan applies the selector query and scans the result into the given value.
- func (_s *ProxySelect) 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[*ProxyQuery, *ProxySelect](ctx, _s.ProxyQuery, _s, _s.inters, v)
- }
- func (_s *ProxySelect) sqlScan(ctx context.Context, root *ProxyQuery, 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)
- }
|