github.com/NpoolPlatform/chain-middleware@v0.0.0-20240228100535-eb1bcf896eb9/pkg/db/ent/currency_query.go (about)

     1  // Code generated by ent, DO NOT EDIT.
     2  
     3  package ent
     4  
     5  import (
     6  	"context"
     7  	"errors"
     8  	"fmt"
     9  	"math"
    10  
    11  	"entgo.io/ent/dialect"
    12  	"entgo.io/ent/dialect/sql"
    13  	"entgo.io/ent/dialect/sql/sqlgraph"
    14  	"entgo.io/ent/schema/field"
    15  	"github.com/NpoolPlatform/chain-middleware/pkg/db/ent/currency"
    16  	"github.com/NpoolPlatform/chain-middleware/pkg/db/ent/predicate"
    17  )
    18  
    19  // CurrencyQuery is the builder for querying Currency entities.
    20  type CurrencyQuery struct {
    21  	config
    22  	limit      *int
    23  	offset     *int
    24  	unique     *bool
    25  	order      []OrderFunc
    26  	fields     []string
    27  	predicates []predicate.Currency
    28  	modifiers  []func(*sql.Selector)
    29  	// intermediate query (i.e. traversal path).
    30  	sql  *sql.Selector
    31  	path func(context.Context) (*sql.Selector, error)
    32  }
    33  
    34  // Where adds a new predicate for the CurrencyQuery builder.
    35  func (cq *CurrencyQuery) Where(ps ...predicate.Currency) *CurrencyQuery {
    36  	cq.predicates = append(cq.predicates, ps...)
    37  	return cq
    38  }
    39  
    40  // Limit adds a limit step to the query.
    41  func (cq *CurrencyQuery) Limit(limit int) *CurrencyQuery {
    42  	cq.limit = &limit
    43  	return cq
    44  }
    45  
    46  // Offset adds an offset step to the query.
    47  func (cq *CurrencyQuery) Offset(offset int) *CurrencyQuery {
    48  	cq.offset = &offset
    49  	return cq
    50  }
    51  
    52  // Unique configures the query builder to filter duplicate records on query.
    53  // By default, unique is set to true, and can be disabled using this method.
    54  func (cq *CurrencyQuery) Unique(unique bool) *CurrencyQuery {
    55  	cq.unique = &unique
    56  	return cq
    57  }
    58  
    59  // Order adds an order step to the query.
    60  func (cq *CurrencyQuery) Order(o ...OrderFunc) *CurrencyQuery {
    61  	cq.order = append(cq.order, o...)
    62  	return cq
    63  }
    64  
    65  // First returns the first Currency entity from the query.
    66  // Returns a *NotFoundError when no Currency was found.
    67  func (cq *CurrencyQuery) First(ctx context.Context) (*Currency, error) {
    68  	nodes, err := cq.Limit(1).All(ctx)
    69  	if err != nil {
    70  		return nil, err
    71  	}
    72  	if len(nodes) == 0 {
    73  		return nil, &NotFoundError{currency.Label}
    74  	}
    75  	return nodes[0], nil
    76  }
    77  
    78  // FirstX is like First, but panics if an error occurs.
    79  func (cq *CurrencyQuery) FirstX(ctx context.Context) *Currency {
    80  	node, err := cq.First(ctx)
    81  	if err != nil && !IsNotFound(err) {
    82  		panic(err)
    83  	}
    84  	return node
    85  }
    86  
    87  // FirstID returns the first Currency ID from the query.
    88  // Returns a *NotFoundError when no Currency ID was found.
    89  func (cq *CurrencyQuery) FirstID(ctx context.Context) (id uint32, err error) {
    90  	var ids []uint32
    91  	if ids, err = cq.Limit(1).IDs(ctx); err != nil {
    92  		return
    93  	}
    94  	if len(ids) == 0 {
    95  		err = &NotFoundError{currency.Label}
    96  		return
    97  	}
    98  	return ids[0], nil
    99  }
   100  
   101  // FirstIDX is like FirstID, but panics if an error occurs.
   102  func (cq *CurrencyQuery) FirstIDX(ctx context.Context) uint32 {
   103  	id, err := cq.FirstID(ctx)
   104  	if err != nil && !IsNotFound(err) {
   105  		panic(err)
   106  	}
   107  	return id
   108  }
   109  
   110  // Only returns a single Currency entity found by the query, ensuring it only returns one.
   111  // Returns a *NotSingularError when more than one Currency entity is found.
   112  // Returns a *NotFoundError when no Currency entities are found.
   113  func (cq *CurrencyQuery) Only(ctx context.Context) (*Currency, error) {
   114  	nodes, err := cq.Limit(2).All(ctx)
   115  	if err != nil {
   116  		return nil, err
   117  	}
   118  	switch len(nodes) {
   119  	case 1:
   120  		return nodes[0], nil
   121  	case 0:
   122  		return nil, &NotFoundError{currency.Label}
   123  	default:
   124  		return nil, &NotSingularError{currency.Label}
   125  	}
   126  }
   127  
   128  // OnlyX is like Only, but panics if an error occurs.
   129  func (cq *CurrencyQuery) OnlyX(ctx context.Context) *Currency {
   130  	node, err := cq.Only(ctx)
   131  	if err != nil {
   132  		panic(err)
   133  	}
   134  	return node
   135  }
   136  
   137  // OnlyID is like Only, but returns the only Currency ID in the query.
   138  // Returns a *NotSingularError when more than one Currency ID is found.
   139  // Returns a *NotFoundError when no entities are found.
   140  func (cq *CurrencyQuery) OnlyID(ctx context.Context) (id uint32, err error) {
   141  	var ids []uint32
   142  	if ids, err = cq.Limit(2).IDs(ctx); err != nil {
   143  		return
   144  	}
   145  	switch len(ids) {
   146  	case 1:
   147  		id = ids[0]
   148  	case 0:
   149  		err = &NotFoundError{currency.Label}
   150  	default:
   151  		err = &NotSingularError{currency.Label}
   152  	}
   153  	return
   154  }
   155  
   156  // OnlyIDX is like OnlyID, but panics if an error occurs.
   157  func (cq *CurrencyQuery) OnlyIDX(ctx context.Context) uint32 {
   158  	id, err := cq.OnlyID(ctx)
   159  	if err != nil {
   160  		panic(err)
   161  	}
   162  	return id
   163  }
   164  
   165  // All executes the query and returns a list of Currencies.
   166  func (cq *CurrencyQuery) All(ctx context.Context) ([]*Currency, error) {
   167  	if err := cq.prepareQuery(ctx); err != nil {
   168  		return nil, err
   169  	}
   170  	return cq.sqlAll(ctx)
   171  }
   172  
   173  // AllX is like All, but panics if an error occurs.
   174  func (cq *CurrencyQuery) AllX(ctx context.Context) []*Currency {
   175  	nodes, err := cq.All(ctx)
   176  	if err != nil {
   177  		panic(err)
   178  	}
   179  	return nodes
   180  }
   181  
   182  // IDs executes the query and returns a list of Currency IDs.
   183  func (cq *CurrencyQuery) IDs(ctx context.Context) ([]uint32, error) {
   184  	var ids []uint32
   185  	if err := cq.Select(currency.FieldID).Scan(ctx, &ids); err != nil {
   186  		return nil, err
   187  	}
   188  	return ids, nil
   189  }
   190  
   191  // IDsX is like IDs, but panics if an error occurs.
   192  func (cq *CurrencyQuery) IDsX(ctx context.Context) []uint32 {
   193  	ids, err := cq.IDs(ctx)
   194  	if err != nil {
   195  		panic(err)
   196  	}
   197  	return ids
   198  }
   199  
   200  // Count returns the count of the given query.
   201  func (cq *CurrencyQuery) Count(ctx context.Context) (int, error) {
   202  	if err := cq.prepareQuery(ctx); err != nil {
   203  		return 0, err
   204  	}
   205  	return cq.sqlCount(ctx)
   206  }
   207  
   208  // CountX is like Count, but panics if an error occurs.
   209  func (cq *CurrencyQuery) CountX(ctx context.Context) int {
   210  	count, err := cq.Count(ctx)
   211  	if err != nil {
   212  		panic(err)
   213  	}
   214  	return count
   215  }
   216  
   217  // Exist returns true if the query has elements in the graph.
   218  func (cq *CurrencyQuery) Exist(ctx context.Context) (bool, error) {
   219  	if err := cq.prepareQuery(ctx); err != nil {
   220  		return false, err
   221  	}
   222  	return cq.sqlExist(ctx)
   223  }
   224  
   225  // ExistX is like Exist, but panics if an error occurs.
   226  func (cq *CurrencyQuery) ExistX(ctx context.Context) bool {
   227  	exist, err := cq.Exist(ctx)
   228  	if err != nil {
   229  		panic(err)
   230  	}
   231  	return exist
   232  }
   233  
   234  // Clone returns a duplicate of the CurrencyQuery builder, including all associated steps. It can be
   235  // used to prepare common query builders and use them differently after the clone is made.
   236  func (cq *CurrencyQuery) Clone() *CurrencyQuery {
   237  	if cq == nil {
   238  		return nil
   239  	}
   240  	return &CurrencyQuery{
   241  		config:     cq.config,
   242  		limit:      cq.limit,
   243  		offset:     cq.offset,
   244  		order:      append([]OrderFunc{}, cq.order...),
   245  		predicates: append([]predicate.Currency{}, cq.predicates...),
   246  		// clone intermediate query.
   247  		sql:    cq.sql.Clone(),
   248  		path:   cq.path,
   249  		unique: cq.unique,
   250  	}
   251  }
   252  
   253  // GroupBy is used to group vertices by one or more fields/columns.
   254  // It is often used with aggregate functions, like: count, max, mean, min, sum.
   255  //
   256  // Example:
   257  //
   258  //	var v []struct {
   259  //		CreatedAt uint32 `json:"created_at,omitempty"`
   260  //		Count int `json:"count,omitempty"`
   261  //	}
   262  //
   263  //	client.Currency.Query().
   264  //		GroupBy(currency.FieldCreatedAt).
   265  //		Aggregate(ent.Count()).
   266  //		Scan(ctx, &v)
   267  //
   268  func (cq *CurrencyQuery) GroupBy(field string, fields ...string) *CurrencyGroupBy {
   269  	grbuild := &CurrencyGroupBy{config: cq.config}
   270  	grbuild.fields = append([]string{field}, fields...)
   271  	grbuild.path = func(ctx context.Context) (prev *sql.Selector, err error) {
   272  		if err := cq.prepareQuery(ctx); err != nil {
   273  			return nil, err
   274  		}
   275  		return cq.sqlQuery(ctx), nil
   276  	}
   277  	grbuild.label = currency.Label
   278  	grbuild.flds, grbuild.scan = &grbuild.fields, grbuild.Scan
   279  	return grbuild
   280  }
   281  
   282  // Select allows the selection one or more fields/columns for the given query,
   283  // instead of selecting all fields in the entity.
   284  //
   285  // Example:
   286  //
   287  //	var v []struct {
   288  //		CreatedAt uint32 `json:"created_at,omitempty"`
   289  //	}
   290  //
   291  //	client.Currency.Query().
   292  //		Select(currency.FieldCreatedAt).
   293  //		Scan(ctx, &v)
   294  //
   295  func (cq *CurrencyQuery) Select(fields ...string) *CurrencySelect {
   296  	cq.fields = append(cq.fields, fields...)
   297  	selbuild := &CurrencySelect{CurrencyQuery: cq}
   298  	selbuild.label = currency.Label
   299  	selbuild.flds, selbuild.scan = &cq.fields, selbuild.Scan
   300  	return selbuild
   301  }
   302  
   303  func (cq *CurrencyQuery) prepareQuery(ctx context.Context) error {
   304  	for _, f := range cq.fields {
   305  		if !currency.ValidColumn(f) {
   306  			return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
   307  		}
   308  	}
   309  	if cq.path != nil {
   310  		prev, err := cq.path(ctx)
   311  		if err != nil {
   312  			return err
   313  		}
   314  		cq.sql = prev
   315  	}
   316  	if currency.Policy == nil {
   317  		return errors.New("ent: uninitialized currency.Policy (forgotten import ent/runtime?)")
   318  	}
   319  	if err := currency.Policy.EvalQuery(ctx, cq); err != nil {
   320  		return err
   321  	}
   322  	return nil
   323  }
   324  
   325  func (cq *CurrencyQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Currency, error) {
   326  	var (
   327  		nodes = []*Currency{}
   328  		_spec = cq.querySpec()
   329  	)
   330  	_spec.ScanValues = func(columns []string) ([]interface{}, error) {
   331  		return (*Currency).scanValues(nil, columns)
   332  	}
   333  	_spec.Assign = func(columns []string, values []interface{}) error {
   334  		node := &Currency{config: cq.config}
   335  		nodes = append(nodes, node)
   336  		return node.assignValues(columns, values)
   337  	}
   338  	if len(cq.modifiers) > 0 {
   339  		_spec.Modifiers = cq.modifiers
   340  	}
   341  	for i := range hooks {
   342  		hooks[i](ctx, _spec)
   343  	}
   344  	if err := sqlgraph.QueryNodes(ctx, cq.driver, _spec); err != nil {
   345  		return nil, err
   346  	}
   347  	if len(nodes) == 0 {
   348  		return nodes, nil
   349  	}
   350  	return nodes, nil
   351  }
   352  
   353  func (cq *CurrencyQuery) sqlCount(ctx context.Context) (int, error) {
   354  	_spec := cq.querySpec()
   355  	if len(cq.modifiers) > 0 {
   356  		_spec.Modifiers = cq.modifiers
   357  	}
   358  	_spec.Node.Columns = cq.fields
   359  	if len(cq.fields) > 0 {
   360  		_spec.Unique = cq.unique != nil && *cq.unique
   361  	}
   362  	return sqlgraph.CountNodes(ctx, cq.driver, _spec)
   363  }
   364  
   365  func (cq *CurrencyQuery) sqlExist(ctx context.Context) (bool, error) {
   366  	n, err := cq.sqlCount(ctx)
   367  	if err != nil {
   368  		return false, fmt.Errorf("ent: check existence: %w", err)
   369  	}
   370  	return n > 0, nil
   371  }
   372  
   373  func (cq *CurrencyQuery) querySpec() *sqlgraph.QuerySpec {
   374  	_spec := &sqlgraph.QuerySpec{
   375  		Node: &sqlgraph.NodeSpec{
   376  			Table:   currency.Table,
   377  			Columns: currency.Columns,
   378  			ID: &sqlgraph.FieldSpec{
   379  				Type:   field.TypeUint32,
   380  				Column: currency.FieldID,
   381  			},
   382  		},
   383  		From:   cq.sql,
   384  		Unique: true,
   385  	}
   386  	if unique := cq.unique; unique != nil {
   387  		_spec.Unique = *unique
   388  	}
   389  	if fields := cq.fields; len(fields) > 0 {
   390  		_spec.Node.Columns = make([]string, 0, len(fields))
   391  		_spec.Node.Columns = append(_spec.Node.Columns, currency.FieldID)
   392  		for i := range fields {
   393  			if fields[i] != currency.FieldID {
   394  				_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
   395  			}
   396  		}
   397  	}
   398  	if ps := cq.predicates; len(ps) > 0 {
   399  		_spec.Predicate = func(selector *sql.Selector) {
   400  			for i := range ps {
   401  				ps[i](selector)
   402  			}
   403  		}
   404  	}
   405  	if limit := cq.limit; limit != nil {
   406  		_spec.Limit = *limit
   407  	}
   408  	if offset := cq.offset; offset != nil {
   409  		_spec.Offset = *offset
   410  	}
   411  	if ps := cq.order; len(ps) > 0 {
   412  		_spec.Order = func(selector *sql.Selector) {
   413  			for i := range ps {
   414  				ps[i](selector)
   415  			}
   416  		}
   417  	}
   418  	return _spec
   419  }
   420  
   421  func (cq *CurrencyQuery) sqlQuery(ctx context.Context) *sql.Selector {
   422  	builder := sql.Dialect(cq.driver.Dialect())
   423  	t1 := builder.Table(currency.Table)
   424  	columns := cq.fields
   425  	if len(columns) == 0 {
   426  		columns = currency.Columns
   427  	}
   428  	selector := builder.Select(t1.Columns(columns...)...).From(t1)
   429  	if cq.sql != nil {
   430  		selector = cq.sql
   431  		selector.Select(selector.Columns(columns...)...)
   432  	}
   433  	if cq.unique != nil && *cq.unique {
   434  		selector.Distinct()
   435  	}
   436  	for _, m := range cq.modifiers {
   437  		m(selector)
   438  	}
   439  	for _, p := range cq.predicates {
   440  		p(selector)
   441  	}
   442  	for _, p := range cq.order {
   443  		p(selector)
   444  	}
   445  	if offset := cq.offset; offset != nil {
   446  		// limit is mandatory for offset clause. We start
   447  		// with default value, and override it below if needed.
   448  		selector.Offset(*offset).Limit(math.MaxInt32)
   449  	}
   450  	if limit := cq.limit; limit != nil {
   451  		selector.Limit(*limit)
   452  	}
   453  	return selector
   454  }
   455  
   456  // ForUpdate locks the selected rows against concurrent updates, and prevent them from being
   457  // updated, deleted or "selected ... for update" by other sessions, until the transaction is
   458  // either committed or rolled-back.
   459  func (cq *CurrencyQuery) ForUpdate(opts ...sql.LockOption) *CurrencyQuery {
   460  	if cq.driver.Dialect() == dialect.Postgres {
   461  		cq.Unique(false)
   462  	}
   463  	cq.modifiers = append(cq.modifiers, func(s *sql.Selector) {
   464  		s.ForUpdate(opts...)
   465  	})
   466  	return cq
   467  }
   468  
   469  // ForShare behaves similarly to ForUpdate, except that it acquires a shared mode lock
   470  // on any rows that are read. Other sessions can read the rows, but cannot modify them
   471  // until your transaction commits.
   472  func (cq *CurrencyQuery) ForShare(opts ...sql.LockOption) *CurrencyQuery {
   473  	if cq.driver.Dialect() == dialect.Postgres {
   474  		cq.Unique(false)
   475  	}
   476  	cq.modifiers = append(cq.modifiers, func(s *sql.Selector) {
   477  		s.ForShare(opts...)
   478  	})
   479  	return cq
   480  }
   481  
   482  // Modify adds a query modifier for attaching custom logic to queries.
   483  func (cq *CurrencyQuery) Modify(modifiers ...func(s *sql.Selector)) *CurrencySelect {
   484  	cq.modifiers = append(cq.modifiers, modifiers...)
   485  	return cq.Select()
   486  }
   487  
   488  // CurrencyGroupBy is the group-by builder for Currency entities.
   489  type CurrencyGroupBy struct {
   490  	config
   491  	selector
   492  	fields []string
   493  	fns    []AggregateFunc
   494  	// intermediate query (i.e. traversal path).
   495  	sql  *sql.Selector
   496  	path func(context.Context) (*sql.Selector, error)
   497  }
   498  
   499  // Aggregate adds the given aggregation functions to the group-by query.
   500  func (cgb *CurrencyGroupBy) Aggregate(fns ...AggregateFunc) *CurrencyGroupBy {
   501  	cgb.fns = append(cgb.fns, fns...)
   502  	return cgb
   503  }
   504  
   505  // Scan applies the group-by query and scans the result into the given value.
   506  func (cgb *CurrencyGroupBy) Scan(ctx context.Context, v interface{}) error {
   507  	query, err := cgb.path(ctx)
   508  	if err != nil {
   509  		return err
   510  	}
   511  	cgb.sql = query
   512  	return cgb.sqlScan(ctx, v)
   513  }
   514  
   515  func (cgb *CurrencyGroupBy) sqlScan(ctx context.Context, v interface{}) error {
   516  	for _, f := range cgb.fields {
   517  		if !currency.ValidColumn(f) {
   518  			return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)}
   519  		}
   520  	}
   521  	selector := cgb.sqlQuery()
   522  	if err := selector.Err(); err != nil {
   523  		return err
   524  	}
   525  	rows := &sql.Rows{}
   526  	query, args := selector.Query()
   527  	if err := cgb.driver.Query(ctx, query, args, rows); err != nil {
   528  		return err
   529  	}
   530  	defer rows.Close()
   531  	return sql.ScanSlice(rows, v)
   532  }
   533  
   534  func (cgb *CurrencyGroupBy) sqlQuery() *sql.Selector {
   535  	selector := cgb.sql.Select()
   536  	aggregation := make([]string, 0, len(cgb.fns))
   537  	for _, fn := range cgb.fns {
   538  		aggregation = append(aggregation, fn(selector))
   539  	}
   540  	// If no columns were selected in a custom aggregation function, the default
   541  	// selection is the fields used for "group-by", and the aggregation functions.
   542  	if len(selector.SelectedColumns()) == 0 {
   543  		columns := make([]string, 0, len(cgb.fields)+len(cgb.fns))
   544  		for _, f := range cgb.fields {
   545  			columns = append(columns, selector.C(f))
   546  		}
   547  		columns = append(columns, aggregation...)
   548  		selector.Select(columns...)
   549  	}
   550  	return selector.GroupBy(selector.Columns(cgb.fields...)...)
   551  }
   552  
   553  // CurrencySelect is the builder for selecting fields of Currency entities.
   554  type CurrencySelect struct {
   555  	*CurrencyQuery
   556  	selector
   557  	// intermediate query (i.e. traversal path).
   558  	sql *sql.Selector
   559  }
   560  
   561  // Scan applies the selector query and scans the result into the given value.
   562  func (cs *CurrencySelect) Scan(ctx context.Context, v interface{}) error {
   563  	if err := cs.prepareQuery(ctx); err != nil {
   564  		return err
   565  	}
   566  	cs.sql = cs.CurrencyQuery.sqlQuery(ctx)
   567  	return cs.sqlScan(ctx, v)
   568  }
   569  
   570  func (cs *CurrencySelect) sqlScan(ctx context.Context, v interface{}) error {
   571  	rows := &sql.Rows{}
   572  	query, args := cs.sql.Query()
   573  	if err := cs.driver.Query(ctx, query, args, rows); err != nil {
   574  		return err
   575  	}
   576  	defer rows.Close()
   577  	return sql.ScanSlice(rows, v)
   578  }
   579  
   580  // Modify adds a query modifier for attaching custom logic to queries.
   581  func (cs *CurrencySelect) Modify(modifiers ...func(s *sql.Selector)) *CurrencySelect {
   582  	cs.modifiers = append(cs.modifiers, modifiers...)
   583  	return cs
   584  }