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 }