redeemcode_query.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "context"
  5. "fmt"
  6. "math"
  7. "entgo.io/ent"
  8. "entgo.io/ent/dialect"
  9. "entgo.io/ent/dialect/sql"
  10. "entgo.io/ent/dialect/sql/sqlgraph"
  11. "entgo.io/ent/schema/field"
  12. "github.com/Wei-Shaw/sub2api/ent/group"
  13. "github.com/Wei-Shaw/sub2api/ent/predicate"
  14. "github.com/Wei-Shaw/sub2api/ent/redeemcode"
  15. "github.com/Wei-Shaw/sub2api/ent/user"
  16. )
  17. // RedeemCodeQuery is the builder for querying RedeemCode entities.
  18. type RedeemCodeQuery struct {
  19. config
  20. ctx *QueryContext
  21. order []redeemcode.OrderOption
  22. inters []Interceptor
  23. predicates []predicate.RedeemCode
  24. withUser *UserQuery
  25. withGroup *GroupQuery
  26. modifiers []func(*sql.Selector)
  27. // intermediate query (i.e. traversal path).
  28. sql *sql.Selector
  29. path func(context.Context) (*sql.Selector, error)
  30. }
  31. // Where adds a new predicate for the RedeemCodeQuery builder.
  32. func (_q *RedeemCodeQuery) Where(ps ...predicate.RedeemCode) *RedeemCodeQuery {
  33. _q.predicates = append(_q.predicates, ps...)
  34. return _q
  35. }
  36. // Limit the number of records to be returned by this query.
  37. func (_q *RedeemCodeQuery) Limit(limit int) *RedeemCodeQuery {
  38. _q.ctx.Limit = &limit
  39. return _q
  40. }
  41. // Offset to start from.
  42. func (_q *RedeemCodeQuery) Offset(offset int) *RedeemCodeQuery {
  43. _q.ctx.Offset = &offset
  44. return _q
  45. }
  46. // Unique configures the query builder to filter duplicate records on query.
  47. // By default, unique is set to true, and can be disabled using this method.
  48. func (_q *RedeemCodeQuery) Unique(unique bool) *RedeemCodeQuery {
  49. _q.ctx.Unique = &unique
  50. return _q
  51. }
  52. // Order specifies how the records should be ordered.
  53. func (_q *RedeemCodeQuery) Order(o ...redeemcode.OrderOption) *RedeemCodeQuery {
  54. _q.order = append(_q.order, o...)
  55. return _q
  56. }
  57. // QueryUser chains the current query on the "user" edge.
  58. func (_q *RedeemCodeQuery) QueryUser() *UserQuery {
  59. query := (&UserClient{config: _q.config}).Query()
  60. query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
  61. if err := _q.prepareQuery(ctx); err != nil {
  62. return nil, err
  63. }
  64. selector := _q.sqlQuery(ctx)
  65. if err := selector.Err(); err != nil {
  66. return nil, err
  67. }
  68. step := sqlgraph.NewStep(
  69. sqlgraph.From(redeemcode.Table, redeemcode.FieldID, selector),
  70. sqlgraph.To(user.Table, user.FieldID),
  71. sqlgraph.Edge(sqlgraph.M2O, true, redeemcode.UserTable, redeemcode.UserColumn),
  72. )
  73. fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
  74. return fromU, nil
  75. }
  76. return query
  77. }
  78. // QueryGroup chains the current query on the "group" edge.
  79. func (_q *RedeemCodeQuery) QueryGroup() *GroupQuery {
  80. query := (&GroupClient{config: _q.config}).Query()
  81. query.path = func(ctx context.Context) (fromU *sql.Selector, err error) {
  82. if err := _q.prepareQuery(ctx); err != nil {
  83. return nil, err
  84. }
  85. selector := _q.sqlQuery(ctx)
  86. if err := selector.Err(); err != nil {
  87. return nil, err
  88. }
  89. step := sqlgraph.NewStep(
  90. sqlgraph.From(redeemcode.Table, redeemcode.FieldID, selector),
  91. sqlgraph.To(group.Table, group.FieldID),
  92. sqlgraph.Edge(sqlgraph.M2O, true, redeemcode.GroupTable, redeemcode.GroupColumn),
  93. )
  94. fromU = sqlgraph.SetNeighbors(_q.driver.Dialect(), step)
  95. return fromU, nil
  96. }
  97. return query
  98. }
  99. // First returns the first RedeemCode entity from the query.
  100. // Returns a *NotFoundError when no RedeemCode was found.
  101. func (_q *RedeemCodeQuery) First(ctx context.Context) (*RedeemCode, error) {
  102. nodes, err := _q.Limit(1).All(setContextOp(ctx, _q.ctx, ent.OpQueryFirst))
  103. if err != nil {
  104. return nil, err
  105. }
  106. if len(nodes) == 0 {
  107. return nil, &NotFoundError{redeemcode.Label}
  108. }
  109. return nodes[0], nil
  110. }
  111. // FirstX is like First, but panics if an error occurs.
  112. func (_q *RedeemCodeQuery) FirstX(ctx context.Context) *RedeemCode {
  113. node, err := _q.First(ctx)
  114. if err != nil && !IsNotFound(err) {
  115. panic(err)
  116. }
  117. return node
  118. }
  119. // FirstID returns the first RedeemCode ID from the query.
  120. // Returns a *NotFoundError when no RedeemCode ID was found.
  121. func (_q *RedeemCodeQuery) FirstID(ctx context.Context) (id int64, err error) {
  122. var ids []int64
  123. if ids, err = _q.Limit(1).IDs(setContextOp(ctx, _q.ctx, ent.OpQueryFirstID)); err != nil {
  124. return
  125. }
  126. if len(ids) == 0 {
  127. err = &NotFoundError{redeemcode.Label}
  128. return
  129. }
  130. return ids[0], nil
  131. }
  132. // FirstIDX is like FirstID, but panics if an error occurs.
  133. func (_q *RedeemCodeQuery) FirstIDX(ctx context.Context) int64 {
  134. id, err := _q.FirstID(ctx)
  135. if err != nil && !IsNotFound(err) {
  136. panic(err)
  137. }
  138. return id
  139. }
  140. // Only returns a single RedeemCode entity found by the query, ensuring it only returns one.
  141. // Returns a *NotSingularError when more than one RedeemCode entity is found.
  142. // Returns a *NotFoundError when no RedeemCode entities are found.
  143. func (_q *RedeemCodeQuery) Only(ctx context.Context) (*RedeemCode, error) {
  144. nodes, err := _q.Limit(2).All(setContextOp(ctx, _q.ctx, ent.OpQueryOnly))
  145. if err != nil {
  146. return nil, err
  147. }
  148. switch len(nodes) {
  149. case 1:
  150. return nodes[0], nil
  151. case 0:
  152. return nil, &NotFoundError{redeemcode.Label}
  153. default:
  154. return nil, &NotSingularError{redeemcode.Label}
  155. }
  156. }
  157. // OnlyX is like Only, but panics if an error occurs.
  158. func (_q *RedeemCodeQuery) OnlyX(ctx context.Context) *RedeemCode {
  159. node, err := _q.Only(ctx)
  160. if err != nil {
  161. panic(err)
  162. }
  163. return node
  164. }
  165. // OnlyID is like Only, but returns the only RedeemCode ID in the query.
  166. // Returns a *NotSingularError when more than one RedeemCode ID is found.
  167. // Returns a *NotFoundError when no entities are found.
  168. func (_q *RedeemCodeQuery) OnlyID(ctx context.Context) (id int64, err error) {
  169. var ids []int64
  170. if ids, err = _q.Limit(2).IDs(setContextOp(ctx, _q.ctx, ent.OpQueryOnlyID)); err != nil {
  171. return
  172. }
  173. switch len(ids) {
  174. case 1:
  175. id = ids[0]
  176. case 0:
  177. err = &NotFoundError{redeemcode.Label}
  178. default:
  179. err = &NotSingularError{redeemcode.Label}
  180. }
  181. return
  182. }
  183. // OnlyIDX is like OnlyID, but panics if an error occurs.
  184. func (_q *RedeemCodeQuery) OnlyIDX(ctx context.Context) int64 {
  185. id, err := _q.OnlyID(ctx)
  186. if err != nil {
  187. panic(err)
  188. }
  189. return id
  190. }
  191. // All executes the query and returns a list of RedeemCodes.
  192. func (_q *RedeemCodeQuery) All(ctx context.Context) ([]*RedeemCode, error) {
  193. ctx = setContextOp(ctx, _q.ctx, ent.OpQueryAll)
  194. if err := _q.prepareQuery(ctx); err != nil {
  195. return nil, err
  196. }
  197. qr := querierAll[[]*RedeemCode, *RedeemCodeQuery]()
  198. return withInterceptors[[]*RedeemCode](ctx, _q, qr, _q.inters)
  199. }
  200. // AllX is like All, but panics if an error occurs.
  201. func (_q *RedeemCodeQuery) AllX(ctx context.Context) []*RedeemCode {
  202. nodes, err := _q.All(ctx)
  203. if err != nil {
  204. panic(err)
  205. }
  206. return nodes
  207. }
  208. // IDs executes the query and returns a list of RedeemCode IDs.
  209. func (_q *RedeemCodeQuery) IDs(ctx context.Context) (ids []int64, err error) {
  210. if _q.ctx.Unique == nil && _q.path != nil {
  211. _q.Unique(true)
  212. }
  213. ctx = setContextOp(ctx, _q.ctx, ent.OpQueryIDs)
  214. if err = _q.Select(redeemcode.FieldID).Scan(ctx, &ids); err != nil {
  215. return nil, err
  216. }
  217. return ids, nil
  218. }
  219. // IDsX is like IDs, but panics if an error occurs.
  220. func (_q *RedeemCodeQuery) IDsX(ctx context.Context) []int64 {
  221. ids, err := _q.IDs(ctx)
  222. if err != nil {
  223. panic(err)
  224. }
  225. return ids
  226. }
  227. // Count returns the count of the given query.
  228. func (_q *RedeemCodeQuery) Count(ctx context.Context) (int, error) {
  229. ctx = setContextOp(ctx, _q.ctx, ent.OpQueryCount)
  230. if err := _q.prepareQuery(ctx); err != nil {
  231. return 0, err
  232. }
  233. return withInterceptors[int](ctx, _q, querierCount[*RedeemCodeQuery](), _q.inters)
  234. }
  235. // CountX is like Count, but panics if an error occurs.
  236. func (_q *RedeemCodeQuery) CountX(ctx context.Context) int {
  237. count, err := _q.Count(ctx)
  238. if err != nil {
  239. panic(err)
  240. }
  241. return count
  242. }
  243. // Exist returns true if the query has elements in the graph.
  244. func (_q *RedeemCodeQuery) Exist(ctx context.Context) (bool, error) {
  245. ctx = setContextOp(ctx, _q.ctx, ent.OpQueryExist)
  246. switch _, err := _q.FirstID(ctx); {
  247. case IsNotFound(err):
  248. return false, nil
  249. case err != nil:
  250. return false, fmt.Errorf("ent: check existence: %w", err)
  251. default:
  252. return true, nil
  253. }
  254. }
  255. // ExistX is like Exist, but panics if an error occurs.
  256. func (_q *RedeemCodeQuery) ExistX(ctx context.Context) bool {
  257. exist, err := _q.Exist(ctx)
  258. if err != nil {
  259. panic(err)
  260. }
  261. return exist
  262. }
  263. // Clone returns a duplicate of the RedeemCodeQuery builder, including all associated steps. It can be
  264. // used to prepare common query builders and use them differently after the clone is made.
  265. func (_q *RedeemCodeQuery) Clone() *RedeemCodeQuery {
  266. if _q == nil {
  267. return nil
  268. }
  269. return &RedeemCodeQuery{
  270. config: _q.config,
  271. ctx: _q.ctx.Clone(),
  272. order: append([]redeemcode.OrderOption{}, _q.order...),
  273. inters: append([]Interceptor{}, _q.inters...),
  274. predicates: append([]predicate.RedeemCode{}, _q.predicates...),
  275. withUser: _q.withUser.Clone(),
  276. withGroup: _q.withGroup.Clone(),
  277. // clone intermediate query.
  278. sql: _q.sql.Clone(),
  279. path: _q.path,
  280. }
  281. }
  282. // WithUser tells the query-builder to eager-load the nodes that are connected to
  283. // the "user" edge. The optional arguments are used to configure the query builder of the edge.
  284. func (_q *RedeemCodeQuery) WithUser(opts ...func(*UserQuery)) *RedeemCodeQuery {
  285. query := (&UserClient{config: _q.config}).Query()
  286. for _, opt := range opts {
  287. opt(query)
  288. }
  289. _q.withUser = query
  290. return _q
  291. }
  292. // WithGroup tells the query-builder to eager-load the nodes that are connected to
  293. // the "group" edge. The optional arguments are used to configure the query builder of the edge.
  294. func (_q *RedeemCodeQuery) WithGroup(opts ...func(*GroupQuery)) *RedeemCodeQuery {
  295. query := (&GroupClient{config: _q.config}).Query()
  296. for _, opt := range opts {
  297. opt(query)
  298. }
  299. _q.withGroup = query
  300. return _q
  301. }
  302. // GroupBy is used to group vertices by one or more fields/columns.
  303. // It is often used with aggregate functions, like: count, max, mean, min, sum.
  304. //
  305. // Example:
  306. //
  307. // var v []struct {
  308. // Code string `json:"code,omitempty"`
  309. // Count int `json:"count,omitempty"`
  310. // }
  311. //
  312. // client.RedeemCode.Query().
  313. // GroupBy(redeemcode.FieldCode).
  314. // Aggregate(ent.Count()).
  315. // Scan(ctx, &v)
  316. func (_q *RedeemCodeQuery) GroupBy(field string, fields ...string) *RedeemCodeGroupBy {
  317. _q.ctx.Fields = append([]string{field}, fields...)
  318. grbuild := &RedeemCodeGroupBy{build: _q}
  319. grbuild.flds = &_q.ctx.Fields
  320. grbuild.label = redeemcode.Label
  321. grbuild.scan = grbuild.Scan
  322. return grbuild
  323. }
  324. // Select allows the selection one or more fields/columns for the given query,
  325. // instead of selecting all fields in the entity.
  326. //
  327. // Example:
  328. //
  329. // var v []struct {
  330. // Code string `json:"code,omitempty"`
  331. // }
  332. //
  333. // client.RedeemCode.Query().
  334. // Select(redeemcode.FieldCode).
  335. // Scan(ctx, &v)
  336. func (_q *RedeemCodeQuery) Select(fields ...string) *RedeemCodeSelect {
  337. _q.ctx.Fields = append(_q.ctx.Fields, fields...)
  338. sbuild := &RedeemCodeSelect{RedeemCodeQuery: _q}
  339. sbuild.label = redeemcode.Label
  340. sbuild.flds, sbuild.scan = &_q.ctx.Fields, sbuild.Scan
  341. return sbuild
  342. }
  343. // Aggregate returns a RedeemCodeSelect configured with the given aggregations.
  344. func (_q *RedeemCodeQuery) Aggregate(fns ...AggregateFunc) *RedeemCodeSelect {
  345. return _q.Select().Aggregate(fns...)
  346. }
  347. func (_q *RedeemCodeQuery) prepareQuery(ctx context.Context) error {
  348. for _, inter := range _q.inters {
  349. if inter == nil {
  350. return fmt.Errorf("ent: uninitialized interceptor (forgotten import ent/runtime?)")
  351. }
  352. if trv, ok := inter.(Traverser); ok {
  353. if err := trv.Traverse(ctx, _q); err != nil {
  354. return err
  355. }
  356. }
  357. }
  358. for _, f := range _q.ctx.Fields {
  359. if !redeemcode.ValidColumn(f) {
  360. return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
  361. }
  362. }
  363. if _q.path != nil {
  364. prev, err := _q.path(ctx)
  365. if err != nil {
  366. return err
  367. }
  368. _q.sql = prev
  369. }
  370. return nil
  371. }
  372. func (_q *RedeemCodeQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*RedeemCode, error) {
  373. var (
  374. nodes = []*RedeemCode{}
  375. _spec = _q.querySpec()
  376. loadedTypes = [2]bool{
  377. _q.withUser != nil,
  378. _q.withGroup != nil,
  379. }
  380. )
  381. _spec.ScanValues = func(columns []string) ([]any, error) {
  382. return (*RedeemCode).scanValues(nil, columns)
  383. }
  384. _spec.Assign = func(columns []string, values []any) error {
  385. node := &RedeemCode{config: _q.config}
  386. nodes = append(nodes, node)
  387. node.Edges.loadedTypes = loadedTypes
  388. return node.assignValues(columns, values)
  389. }
  390. if len(_q.modifiers) > 0 {
  391. _spec.Modifiers = _q.modifiers
  392. }
  393. for i := range hooks {
  394. hooks[i](ctx, _spec)
  395. }
  396. if err := sqlgraph.QueryNodes(ctx, _q.driver, _spec); err != nil {
  397. return nil, err
  398. }
  399. if len(nodes) == 0 {
  400. return nodes, nil
  401. }
  402. if query := _q.withUser; query != nil {
  403. if err := _q.loadUser(ctx, query, nodes, nil,
  404. func(n *RedeemCode, e *User) { n.Edges.User = e }); err != nil {
  405. return nil, err
  406. }
  407. }
  408. if query := _q.withGroup; query != nil {
  409. if err := _q.loadGroup(ctx, query, nodes, nil,
  410. func(n *RedeemCode, e *Group) { n.Edges.Group = e }); err != nil {
  411. return nil, err
  412. }
  413. }
  414. return nodes, nil
  415. }
  416. func (_q *RedeemCodeQuery) loadUser(ctx context.Context, query *UserQuery, nodes []*RedeemCode, init func(*RedeemCode), assign func(*RedeemCode, *User)) error {
  417. ids := make([]int64, 0, len(nodes))
  418. nodeids := make(map[int64][]*RedeemCode)
  419. for i := range nodes {
  420. if nodes[i].UsedBy == nil {
  421. continue
  422. }
  423. fk := *nodes[i].UsedBy
  424. if _, ok := nodeids[fk]; !ok {
  425. ids = append(ids, fk)
  426. }
  427. nodeids[fk] = append(nodeids[fk], nodes[i])
  428. }
  429. if len(ids) == 0 {
  430. return nil
  431. }
  432. query.Where(user.IDIn(ids...))
  433. neighbors, err := query.All(ctx)
  434. if err != nil {
  435. return err
  436. }
  437. for _, n := range neighbors {
  438. nodes, ok := nodeids[n.ID]
  439. if !ok {
  440. return fmt.Errorf(`unexpected foreign-key "used_by" returned %v`, n.ID)
  441. }
  442. for i := range nodes {
  443. assign(nodes[i], n)
  444. }
  445. }
  446. return nil
  447. }
  448. func (_q *RedeemCodeQuery) loadGroup(ctx context.Context, query *GroupQuery, nodes []*RedeemCode, init func(*RedeemCode), assign func(*RedeemCode, *Group)) error {
  449. ids := make([]int64, 0, len(nodes))
  450. nodeids := make(map[int64][]*RedeemCode)
  451. for i := range nodes {
  452. if nodes[i].GroupID == nil {
  453. continue
  454. }
  455. fk := *nodes[i].GroupID
  456. if _, ok := nodeids[fk]; !ok {
  457. ids = append(ids, fk)
  458. }
  459. nodeids[fk] = append(nodeids[fk], nodes[i])
  460. }
  461. if len(ids) == 0 {
  462. return nil
  463. }
  464. query.Where(group.IDIn(ids...))
  465. neighbors, err := query.All(ctx)
  466. if err != nil {
  467. return err
  468. }
  469. for _, n := range neighbors {
  470. nodes, ok := nodeids[n.ID]
  471. if !ok {
  472. return fmt.Errorf(`unexpected foreign-key "group_id" returned %v`, n.ID)
  473. }
  474. for i := range nodes {
  475. assign(nodes[i], n)
  476. }
  477. }
  478. return nil
  479. }
  480. func (_q *RedeemCodeQuery) sqlCount(ctx context.Context) (int, error) {
  481. _spec := _q.querySpec()
  482. if len(_q.modifiers) > 0 {
  483. _spec.Modifiers = _q.modifiers
  484. }
  485. _spec.Node.Columns = _q.ctx.Fields
  486. if len(_q.ctx.Fields) > 0 {
  487. _spec.Unique = _q.ctx.Unique != nil && *_q.ctx.Unique
  488. }
  489. return sqlgraph.CountNodes(ctx, _q.driver, _spec)
  490. }
  491. func (_q *RedeemCodeQuery) querySpec() *sqlgraph.QuerySpec {
  492. _spec := sqlgraph.NewQuerySpec(redeemcode.Table, redeemcode.Columns, sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64))
  493. _spec.From = _q.sql
  494. if unique := _q.ctx.Unique; unique != nil {
  495. _spec.Unique = *unique
  496. } else if _q.path != nil {
  497. _spec.Unique = true
  498. }
  499. if fields := _q.ctx.Fields; len(fields) > 0 {
  500. _spec.Node.Columns = make([]string, 0, len(fields))
  501. _spec.Node.Columns = append(_spec.Node.Columns, redeemcode.FieldID)
  502. for i := range fields {
  503. if fields[i] != redeemcode.FieldID {
  504. _spec.Node.Columns = append(_spec.Node.Columns, fields[i])
  505. }
  506. }
  507. if _q.withUser != nil {
  508. _spec.Node.AddColumnOnce(redeemcode.FieldUsedBy)
  509. }
  510. if _q.withGroup != nil {
  511. _spec.Node.AddColumnOnce(redeemcode.FieldGroupID)
  512. }
  513. }
  514. if ps := _q.predicates; len(ps) > 0 {
  515. _spec.Predicate = func(selector *sql.Selector) {
  516. for i := range ps {
  517. ps[i](selector)
  518. }
  519. }
  520. }
  521. if limit := _q.ctx.Limit; limit != nil {
  522. _spec.Limit = *limit
  523. }
  524. if offset := _q.ctx.Offset; offset != nil {
  525. _spec.Offset = *offset
  526. }
  527. if ps := _q.order; len(ps) > 0 {
  528. _spec.Order = func(selector *sql.Selector) {
  529. for i := range ps {
  530. ps[i](selector)
  531. }
  532. }
  533. }
  534. return _spec
  535. }
  536. func (_q *RedeemCodeQuery) sqlQuery(ctx context.Context) *sql.Selector {
  537. builder := sql.Dialect(_q.driver.Dialect())
  538. t1 := builder.Table(redeemcode.Table)
  539. columns := _q.ctx.Fields
  540. if len(columns) == 0 {
  541. columns = redeemcode.Columns
  542. }
  543. selector := builder.Select(t1.Columns(columns...)...).From(t1)
  544. if _q.sql != nil {
  545. selector = _q.sql
  546. selector.Select(selector.Columns(columns...)...)
  547. }
  548. if _q.ctx.Unique != nil && *_q.ctx.Unique {
  549. selector.Distinct()
  550. }
  551. for _, m := range _q.modifiers {
  552. m(selector)
  553. }
  554. for _, p := range _q.predicates {
  555. p(selector)
  556. }
  557. for _, p := range _q.order {
  558. p(selector)
  559. }
  560. if offset := _q.ctx.Offset; offset != nil {
  561. // limit is mandatory for offset clause. We start
  562. // with default value, and override it below if needed.
  563. selector.Offset(*offset).Limit(math.MaxInt32)
  564. }
  565. if limit := _q.ctx.Limit; limit != nil {
  566. selector.Limit(*limit)
  567. }
  568. return selector
  569. }
  570. // ForUpdate locks the selected rows against concurrent updates, and prevent them from being
  571. // updated, deleted or "selected ... for update" by other sessions, until the transaction is
  572. // either committed or rolled-back.
  573. func (_q *RedeemCodeQuery) ForUpdate(opts ...sql.LockOption) *RedeemCodeQuery {
  574. if _q.driver.Dialect() == dialect.Postgres {
  575. _q.Unique(false)
  576. }
  577. _q.modifiers = append(_q.modifiers, func(s *sql.Selector) {
  578. s.ForUpdate(opts...)
  579. })
  580. return _q
  581. }
  582. // ForShare behaves similarly to ForUpdate, except that it acquires a shared mode lock
  583. // on any rows that are read. Other sessions can read the rows, but cannot modify them
  584. // until your transaction commits.
  585. func (_q *RedeemCodeQuery) ForShare(opts ...sql.LockOption) *RedeemCodeQuery {
  586. if _q.driver.Dialect() == dialect.Postgres {
  587. _q.Unique(false)
  588. }
  589. _q.modifiers = append(_q.modifiers, func(s *sql.Selector) {
  590. s.ForShare(opts...)
  591. })
  592. return _q
  593. }
  594. // RedeemCodeGroupBy is the group-by builder for RedeemCode entities.
  595. type RedeemCodeGroupBy struct {
  596. selector
  597. build *RedeemCodeQuery
  598. }
  599. // Aggregate adds the given aggregation functions to the group-by query.
  600. func (_g *RedeemCodeGroupBy) Aggregate(fns ...AggregateFunc) *RedeemCodeGroupBy {
  601. _g.fns = append(_g.fns, fns...)
  602. return _g
  603. }
  604. // Scan applies the selector query and scans the result into the given value.
  605. func (_g *RedeemCodeGroupBy) Scan(ctx context.Context, v any) error {
  606. ctx = setContextOp(ctx, _g.build.ctx, ent.OpQueryGroupBy)
  607. if err := _g.build.prepareQuery(ctx); err != nil {
  608. return err
  609. }
  610. return scanWithInterceptors[*RedeemCodeQuery, *RedeemCodeGroupBy](ctx, _g.build, _g, _g.build.inters, v)
  611. }
  612. func (_g *RedeemCodeGroupBy) sqlScan(ctx context.Context, root *RedeemCodeQuery, v any) error {
  613. selector := root.sqlQuery(ctx).Select()
  614. aggregation := make([]string, 0, len(_g.fns))
  615. for _, fn := range _g.fns {
  616. aggregation = append(aggregation, fn(selector))
  617. }
  618. if len(selector.SelectedColumns()) == 0 {
  619. columns := make([]string, 0, len(*_g.flds)+len(_g.fns))
  620. for _, f := range *_g.flds {
  621. columns = append(columns, selector.C(f))
  622. }
  623. columns = append(columns, aggregation...)
  624. selector.Select(columns...)
  625. }
  626. selector.GroupBy(selector.Columns(*_g.flds...)...)
  627. if err := selector.Err(); err != nil {
  628. return err
  629. }
  630. rows := &sql.Rows{}
  631. query, args := selector.Query()
  632. if err := _g.build.driver.Query(ctx, query, args, rows); err != nil {
  633. return err
  634. }
  635. defer rows.Close()
  636. return sql.ScanSlice(rows, v)
  637. }
  638. // RedeemCodeSelect is the builder for selecting fields of RedeemCode entities.
  639. type RedeemCodeSelect struct {
  640. *RedeemCodeQuery
  641. selector
  642. }
  643. // Aggregate adds the given aggregation functions to the selector query.
  644. func (_s *RedeemCodeSelect) Aggregate(fns ...AggregateFunc) *RedeemCodeSelect {
  645. _s.fns = append(_s.fns, fns...)
  646. return _s
  647. }
  648. // Scan applies the selector query and scans the result into the given value.
  649. func (_s *RedeemCodeSelect) Scan(ctx context.Context, v any) error {
  650. ctx = setContextOp(ctx, _s.ctx, ent.OpQuerySelect)
  651. if err := _s.prepareQuery(ctx); err != nil {
  652. return err
  653. }
  654. return scanWithInterceptors[*RedeemCodeQuery, *RedeemCodeSelect](ctx, _s.RedeemCodeQuery, _s, _s.inters, v)
  655. }
  656. func (_s *RedeemCodeSelect) sqlScan(ctx context.Context, root *RedeemCodeQuery, v any) error {
  657. selector := root.sqlQuery(ctx)
  658. aggregation := make([]string, 0, len(_s.fns))
  659. for _, fn := range _s.fns {
  660. aggregation = append(aggregation, fn(selector))
  661. }
  662. switch n := len(*_s.selector.flds); {
  663. case n == 0 && len(aggregation) > 0:
  664. selector.Select(aggregation...)
  665. case n != 0 && len(aggregation) > 0:
  666. selector.AppendSelect(aggregation...)
  667. }
  668. rows := &sql.Rows{}
  669. query, args := selector.Query()
  670. if err := _s.driver.Query(ctx, query, args, rows); err != nil {
  671. return err
  672. }
  673. defer rows.Close()
  674. return sql.ScanSlice(rows, v)
  675. }