pagination.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "context"
  5. "fmt"
  6. "wechat-api/ent/batchmsg"
  7. "wechat-api/ent/contact"
  8. "wechat-api/ent/label"
  9. "wechat-api/ent/labelrelationship"
  10. "wechat-api/ent/message"
  11. "wechat-api/ent/messagerecords"
  12. "wechat-api/ent/msg"
  13. "wechat-api/ent/server"
  14. "wechat-api/ent/sopnode"
  15. "wechat-api/ent/sopstage"
  16. "wechat-api/ent/soptask"
  17. "wechat-api/ent/wx"
  18. )
  19. const errInvalidPage = "INVALID_PAGE"
  20. const (
  21. listField = "list"
  22. pageNumField = "pageNum"
  23. pageSizeField = "pageSize"
  24. )
  25. type PageDetails struct {
  26. Page uint64 `json:"page"`
  27. Size uint64 `json:"size"`
  28. Total uint64 `json:"total"`
  29. }
  30. // OrderDirection defines the directions in which to order a list of items.
  31. type OrderDirection string
  32. const (
  33. // OrderDirectionAsc specifies an ascending order.
  34. OrderDirectionAsc OrderDirection = "ASC"
  35. // OrderDirectionDesc specifies a descending order.
  36. OrderDirectionDesc OrderDirection = "DESC"
  37. )
  38. // Validate the order direction value.
  39. func (o OrderDirection) Validate() error {
  40. if o != OrderDirectionAsc && o != OrderDirectionDesc {
  41. return fmt.Errorf("%s is not a valid OrderDirection", o)
  42. }
  43. return nil
  44. }
  45. // String implements fmt.Stringer interface.
  46. func (o OrderDirection) String() string {
  47. return string(o)
  48. }
  49. func (o OrderDirection) reverse() OrderDirection {
  50. if o == OrderDirectionDesc {
  51. return OrderDirectionAsc
  52. }
  53. return OrderDirectionDesc
  54. }
  55. const errInvalidPagination = "INVALID_PAGINATION"
  56. type BatchMsgPager struct {
  57. Order batchmsg.OrderOption
  58. Filter func(*BatchMsgQuery) (*BatchMsgQuery, error)
  59. }
  60. // BatchMsgPaginateOption enables pagination customization.
  61. type BatchMsgPaginateOption func(*BatchMsgPager)
  62. // DefaultBatchMsgOrder is the default ordering of BatchMsg.
  63. var DefaultBatchMsgOrder = Desc(batchmsg.FieldID)
  64. func newBatchMsgPager(opts []BatchMsgPaginateOption) (*BatchMsgPager, error) {
  65. pager := &BatchMsgPager{}
  66. for _, opt := range opts {
  67. opt(pager)
  68. }
  69. if pager.Order == nil {
  70. pager.Order = DefaultBatchMsgOrder
  71. }
  72. return pager, nil
  73. }
  74. func (p *BatchMsgPager) ApplyFilter(query *BatchMsgQuery) (*BatchMsgQuery, error) {
  75. if p.Filter != nil {
  76. return p.Filter(query)
  77. }
  78. return query, nil
  79. }
  80. // BatchMsgPageList is BatchMsg PageList result.
  81. type BatchMsgPageList struct {
  82. List []*BatchMsg `json:"list"`
  83. PageDetails *PageDetails `json:"pageDetails"`
  84. }
  85. func (bm *BatchMsgQuery) Page(
  86. ctx context.Context, pageNum uint64, pageSize uint64, opts ...BatchMsgPaginateOption,
  87. ) (*BatchMsgPageList, error) {
  88. pager, err := newBatchMsgPager(opts)
  89. if err != nil {
  90. return nil, err
  91. }
  92. if bm, err = pager.ApplyFilter(bm); err != nil {
  93. return nil, err
  94. }
  95. ret := &BatchMsgPageList{}
  96. ret.PageDetails = &PageDetails{
  97. Page: pageNum,
  98. Size: pageSize,
  99. }
  100. query := bm.Clone()
  101. query.ctx.Fields = nil
  102. count, err := query.Count(ctx)
  103. if err != nil {
  104. return nil, err
  105. }
  106. ret.PageDetails.Total = uint64(count)
  107. if pager.Order != nil {
  108. bm = bm.Order(pager.Order)
  109. } else {
  110. bm = bm.Order(DefaultBatchMsgOrder)
  111. }
  112. bm = bm.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  113. list, err := bm.All(ctx)
  114. if err != nil {
  115. return nil, err
  116. }
  117. ret.List = list
  118. return ret, nil
  119. }
  120. type ContactPager struct {
  121. Order contact.OrderOption
  122. Filter func(*ContactQuery) (*ContactQuery, error)
  123. }
  124. // ContactPaginateOption enables pagination customization.
  125. type ContactPaginateOption func(*ContactPager)
  126. // DefaultContactOrder is the default ordering of Contact.
  127. var DefaultContactOrder = Desc(contact.FieldID)
  128. func newContactPager(opts []ContactPaginateOption) (*ContactPager, error) {
  129. pager := &ContactPager{}
  130. for _, opt := range opts {
  131. opt(pager)
  132. }
  133. if pager.Order == nil {
  134. pager.Order = DefaultContactOrder
  135. }
  136. return pager, nil
  137. }
  138. func (p *ContactPager) ApplyFilter(query *ContactQuery) (*ContactQuery, error) {
  139. if p.Filter != nil {
  140. return p.Filter(query)
  141. }
  142. return query, nil
  143. }
  144. // ContactPageList is Contact PageList result.
  145. type ContactPageList struct {
  146. List []*Contact `json:"list"`
  147. PageDetails *PageDetails `json:"pageDetails"`
  148. }
  149. func (c *ContactQuery) Page(
  150. ctx context.Context, pageNum uint64, pageSize uint64, opts ...ContactPaginateOption,
  151. ) (*ContactPageList, error) {
  152. pager, err := newContactPager(opts)
  153. if err != nil {
  154. return nil, err
  155. }
  156. if c, err = pager.ApplyFilter(c); err != nil {
  157. return nil, err
  158. }
  159. ret := &ContactPageList{}
  160. ret.PageDetails = &PageDetails{
  161. Page: pageNum,
  162. Size: pageSize,
  163. }
  164. query := c.Clone()
  165. query.ctx.Fields = nil
  166. count, err := query.Count(ctx)
  167. if err != nil {
  168. return nil, err
  169. }
  170. ret.PageDetails.Total = uint64(count)
  171. if pager.Order != nil {
  172. c = c.Order(pager.Order)
  173. } else {
  174. c = c.Order(DefaultContactOrder)
  175. }
  176. c = c.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  177. list, err := c.All(ctx)
  178. if err != nil {
  179. return nil, err
  180. }
  181. ret.List = list
  182. return ret, nil
  183. }
  184. type LabelPager struct {
  185. Order label.OrderOption
  186. Filter func(*LabelQuery) (*LabelQuery, error)
  187. }
  188. // LabelPaginateOption enables pagination customization.
  189. type LabelPaginateOption func(*LabelPager)
  190. // DefaultLabelOrder is the default ordering of Label.
  191. var DefaultLabelOrder = Desc(label.FieldID)
  192. func newLabelPager(opts []LabelPaginateOption) (*LabelPager, error) {
  193. pager := &LabelPager{}
  194. for _, opt := range opts {
  195. opt(pager)
  196. }
  197. if pager.Order == nil {
  198. pager.Order = DefaultLabelOrder
  199. }
  200. return pager, nil
  201. }
  202. func (p *LabelPager) ApplyFilter(query *LabelQuery) (*LabelQuery, error) {
  203. if p.Filter != nil {
  204. return p.Filter(query)
  205. }
  206. return query, nil
  207. }
  208. // LabelPageList is Label PageList result.
  209. type LabelPageList struct {
  210. List []*Label `json:"list"`
  211. PageDetails *PageDetails `json:"pageDetails"`
  212. }
  213. func (l *LabelQuery) Page(
  214. ctx context.Context, pageNum uint64, pageSize uint64, opts ...LabelPaginateOption,
  215. ) (*LabelPageList, error) {
  216. pager, err := newLabelPager(opts)
  217. if err != nil {
  218. return nil, err
  219. }
  220. if l, err = pager.ApplyFilter(l); err != nil {
  221. return nil, err
  222. }
  223. ret := &LabelPageList{}
  224. ret.PageDetails = &PageDetails{
  225. Page: pageNum,
  226. Size: pageSize,
  227. }
  228. query := l.Clone()
  229. query.ctx.Fields = nil
  230. count, err := query.Count(ctx)
  231. if err != nil {
  232. return nil, err
  233. }
  234. ret.PageDetails.Total = uint64(count)
  235. if pager.Order != nil {
  236. l = l.Order(pager.Order)
  237. } else {
  238. l = l.Order(DefaultLabelOrder)
  239. }
  240. l = l.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  241. list, err := l.All(ctx)
  242. if err != nil {
  243. return nil, err
  244. }
  245. ret.List = list
  246. return ret, nil
  247. }
  248. type LabelRelationshipPager struct {
  249. Order labelrelationship.OrderOption
  250. Filter func(*LabelRelationshipQuery) (*LabelRelationshipQuery, error)
  251. }
  252. // LabelRelationshipPaginateOption enables pagination customization.
  253. type LabelRelationshipPaginateOption func(*LabelRelationshipPager)
  254. // DefaultLabelRelationshipOrder is the default ordering of LabelRelationship.
  255. var DefaultLabelRelationshipOrder = Desc(labelrelationship.FieldID)
  256. func newLabelRelationshipPager(opts []LabelRelationshipPaginateOption) (*LabelRelationshipPager, error) {
  257. pager := &LabelRelationshipPager{}
  258. for _, opt := range opts {
  259. opt(pager)
  260. }
  261. if pager.Order == nil {
  262. pager.Order = DefaultLabelRelationshipOrder
  263. }
  264. return pager, nil
  265. }
  266. func (p *LabelRelationshipPager) ApplyFilter(query *LabelRelationshipQuery) (*LabelRelationshipQuery, error) {
  267. if p.Filter != nil {
  268. return p.Filter(query)
  269. }
  270. return query, nil
  271. }
  272. // LabelRelationshipPageList is LabelRelationship PageList result.
  273. type LabelRelationshipPageList struct {
  274. List []*LabelRelationship `json:"list"`
  275. PageDetails *PageDetails `json:"pageDetails"`
  276. }
  277. func (lr *LabelRelationshipQuery) Page(
  278. ctx context.Context, pageNum uint64, pageSize uint64, opts ...LabelRelationshipPaginateOption,
  279. ) (*LabelRelationshipPageList, error) {
  280. pager, err := newLabelRelationshipPager(opts)
  281. if err != nil {
  282. return nil, err
  283. }
  284. if lr, err = pager.ApplyFilter(lr); err != nil {
  285. return nil, err
  286. }
  287. ret := &LabelRelationshipPageList{}
  288. ret.PageDetails = &PageDetails{
  289. Page: pageNum,
  290. Size: pageSize,
  291. }
  292. query := lr.Clone()
  293. query.ctx.Fields = nil
  294. count, err := query.Count(ctx)
  295. if err != nil {
  296. return nil, err
  297. }
  298. ret.PageDetails.Total = uint64(count)
  299. if pager.Order != nil {
  300. lr = lr.Order(pager.Order)
  301. } else {
  302. lr = lr.Order(DefaultLabelRelationshipOrder)
  303. }
  304. lr = lr.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  305. list, err := lr.All(ctx)
  306. if err != nil {
  307. return nil, err
  308. }
  309. ret.List = list
  310. return ret, nil
  311. }
  312. type MessagePager struct {
  313. Order message.OrderOption
  314. Filter func(*MessageQuery) (*MessageQuery, error)
  315. }
  316. // MessagePaginateOption enables pagination customization.
  317. type MessagePaginateOption func(*MessagePager)
  318. // DefaultMessageOrder is the default ordering of Message.
  319. var DefaultMessageOrder = Desc(message.FieldID)
  320. func newMessagePager(opts []MessagePaginateOption) (*MessagePager, error) {
  321. pager := &MessagePager{}
  322. for _, opt := range opts {
  323. opt(pager)
  324. }
  325. if pager.Order == nil {
  326. pager.Order = DefaultMessageOrder
  327. }
  328. return pager, nil
  329. }
  330. func (p *MessagePager) ApplyFilter(query *MessageQuery) (*MessageQuery, error) {
  331. if p.Filter != nil {
  332. return p.Filter(query)
  333. }
  334. return query, nil
  335. }
  336. // MessagePageList is Message PageList result.
  337. type MessagePageList struct {
  338. List []*Message `json:"list"`
  339. PageDetails *PageDetails `json:"pageDetails"`
  340. }
  341. func (m *MessageQuery) Page(
  342. ctx context.Context, pageNum uint64, pageSize uint64, opts ...MessagePaginateOption,
  343. ) (*MessagePageList, error) {
  344. pager, err := newMessagePager(opts)
  345. if err != nil {
  346. return nil, err
  347. }
  348. if m, err = pager.ApplyFilter(m); err != nil {
  349. return nil, err
  350. }
  351. ret := &MessagePageList{}
  352. ret.PageDetails = &PageDetails{
  353. Page: pageNum,
  354. Size: pageSize,
  355. }
  356. query := m.Clone()
  357. query.ctx.Fields = nil
  358. count, err := query.Count(ctx)
  359. if err != nil {
  360. return nil, err
  361. }
  362. ret.PageDetails.Total = uint64(count)
  363. if pager.Order != nil {
  364. m = m.Order(pager.Order)
  365. } else {
  366. m = m.Order(DefaultMessageOrder)
  367. }
  368. m = m.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  369. list, err := m.All(ctx)
  370. if err != nil {
  371. return nil, err
  372. }
  373. ret.List = list
  374. return ret, nil
  375. }
  376. type MessageRecordsPager struct {
  377. Order messagerecords.OrderOption
  378. Filter func(*MessageRecordsQuery) (*MessageRecordsQuery, error)
  379. }
  380. // MessageRecordsPaginateOption enables pagination customization.
  381. type MessageRecordsPaginateOption func(*MessageRecordsPager)
  382. // DefaultMessageRecordsOrder is the default ordering of MessageRecords.
  383. var DefaultMessageRecordsOrder = Desc(messagerecords.FieldID)
  384. func newMessageRecordsPager(opts []MessageRecordsPaginateOption) (*MessageRecordsPager, error) {
  385. pager := &MessageRecordsPager{}
  386. for _, opt := range opts {
  387. opt(pager)
  388. }
  389. if pager.Order == nil {
  390. pager.Order = DefaultMessageRecordsOrder
  391. }
  392. return pager, nil
  393. }
  394. func (p *MessageRecordsPager) ApplyFilter(query *MessageRecordsQuery) (*MessageRecordsQuery, error) {
  395. if p.Filter != nil {
  396. return p.Filter(query)
  397. }
  398. return query, nil
  399. }
  400. // MessageRecordsPageList is MessageRecords PageList result.
  401. type MessageRecordsPageList struct {
  402. List []*MessageRecords `json:"list"`
  403. PageDetails *PageDetails `json:"pageDetails"`
  404. }
  405. func (mr *MessageRecordsQuery) Page(
  406. ctx context.Context, pageNum uint64, pageSize uint64, opts ...MessageRecordsPaginateOption,
  407. ) (*MessageRecordsPageList, error) {
  408. pager, err := newMessageRecordsPager(opts)
  409. if err != nil {
  410. return nil, err
  411. }
  412. if mr, err = pager.ApplyFilter(mr); err != nil {
  413. return nil, err
  414. }
  415. ret := &MessageRecordsPageList{}
  416. ret.PageDetails = &PageDetails{
  417. Page: pageNum,
  418. Size: pageSize,
  419. }
  420. query := mr.Clone()
  421. query.ctx.Fields = nil
  422. count, err := query.Count(ctx)
  423. if err != nil {
  424. return nil, err
  425. }
  426. ret.PageDetails.Total = uint64(count)
  427. if pager.Order != nil {
  428. mr = mr.Order(pager.Order)
  429. } else {
  430. mr = mr.Order(DefaultMessageRecordsOrder)
  431. }
  432. mr = mr.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  433. list, err := mr.All(ctx)
  434. if err != nil {
  435. return nil, err
  436. }
  437. ret.List = list
  438. return ret, nil
  439. }
  440. type MsgPager struct {
  441. Order msg.OrderOption
  442. Filter func(*MsgQuery) (*MsgQuery, error)
  443. }
  444. // MsgPaginateOption enables pagination customization.
  445. type MsgPaginateOption func(*MsgPager)
  446. // DefaultMsgOrder is the default ordering of Msg.
  447. var DefaultMsgOrder = Desc(msg.FieldID)
  448. func newMsgPager(opts []MsgPaginateOption) (*MsgPager, error) {
  449. pager := &MsgPager{}
  450. for _, opt := range opts {
  451. opt(pager)
  452. }
  453. if pager.Order == nil {
  454. pager.Order = DefaultMsgOrder
  455. }
  456. return pager, nil
  457. }
  458. func (p *MsgPager) ApplyFilter(query *MsgQuery) (*MsgQuery, error) {
  459. if p.Filter != nil {
  460. return p.Filter(query)
  461. }
  462. return query, nil
  463. }
  464. // MsgPageList is Msg PageList result.
  465. type MsgPageList struct {
  466. List []*Msg `json:"list"`
  467. PageDetails *PageDetails `json:"pageDetails"`
  468. }
  469. func (m *MsgQuery) Page(
  470. ctx context.Context, pageNum uint64, pageSize uint64, opts ...MsgPaginateOption,
  471. ) (*MsgPageList, error) {
  472. pager, err := newMsgPager(opts)
  473. if err != nil {
  474. return nil, err
  475. }
  476. if m, err = pager.ApplyFilter(m); err != nil {
  477. return nil, err
  478. }
  479. ret := &MsgPageList{}
  480. ret.PageDetails = &PageDetails{
  481. Page: pageNum,
  482. Size: pageSize,
  483. }
  484. query := m.Clone()
  485. query.ctx.Fields = nil
  486. count, err := query.Count(ctx)
  487. if err != nil {
  488. return nil, err
  489. }
  490. ret.PageDetails.Total = uint64(count)
  491. if pager.Order != nil {
  492. m = m.Order(pager.Order)
  493. } else {
  494. m = m.Order(DefaultMsgOrder)
  495. }
  496. m = m.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  497. list, err := m.All(ctx)
  498. if err != nil {
  499. return nil, err
  500. }
  501. ret.List = list
  502. return ret, nil
  503. }
  504. type ServerPager struct {
  505. Order server.OrderOption
  506. Filter func(*ServerQuery) (*ServerQuery, error)
  507. }
  508. // ServerPaginateOption enables pagination customization.
  509. type ServerPaginateOption func(*ServerPager)
  510. // DefaultServerOrder is the default ordering of Server.
  511. var DefaultServerOrder = Desc(server.FieldID)
  512. func newServerPager(opts []ServerPaginateOption) (*ServerPager, error) {
  513. pager := &ServerPager{}
  514. for _, opt := range opts {
  515. opt(pager)
  516. }
  517. if pager.Order == nil {
  518. pager.Order = DefaultServerOrder
  519. }
  520. return pager, nil
  521. }
  522. func (p *ServerPager) ApplyFilter(query *ServerQuery) (*ServerQuery, error) {
  523. if p.Filter != nil {
  524. return p.Filter(query)
  525. }
  526. return query, nil
  527. }
  528. // ServerPageList is Server PageList result.
  529. type ServerPageList struct {
  530. List []*Server `json:"list"`
  531. PageDetails *PageDetails `json:"pageDetails"`
  532. }
  533. func (s *ServerQuery) Page(
  534. ctx context.Context, pageNum uint64, pageSize uint64, opts ...ServerPaginateOption,
  535. ) (*ServerPageList, error) {
  536. pager, err := newServerPager(opts)
  537. if err != nil {
  538. return nil, err
  539. }
  540. if s, err = pager.ApplyFilter(s); err != nil {
  541. return nil, err
  542. }
  543. ret := &ServerPageList{}
  544. ret.PageDetails = &PageDetails{
  545. Page: pageNum,
  546. Size: pageSize,
  547. }
  548. query := s.Clone()
  549. query.ctx.Fields = nil
  550. count, err := query.Count(ctx)
  551. if err != nil {
  552. return nil, err
  553. }
  554. ret.PageDetails.Total = uint64(count)
  555. if pager.Order != nil {
  556. s = s.Order(pager.Order)
  557. } else {
  558. s = s.Order(DefaultServerOrder)
  559. }
  560. s = s.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  561. list, err := s.All(ctx)
  562. if err != nil {
  563. return nil, err
  564. }
  565. ret.List = list
  566. return ret, nil
  567. }
  568. type SopNodePager struct {
  569. Order sopnode.OrderOption
  570. Filter func(*SopNodeQuery) (*SopNodeQuery, error)
  571. }
  572. // SopNodePaginateOption enables pagination customization.
  573. type SopNodePaginateOption func(*SopNodePager)
  574. // DefaultSopNodeOrder is the default ordering of SopNode.
  575. var DefaultSopNodeOrder = Desc(sopnode.FieldID)
  576. func newSopNodePager(opts []SopNodePaginateOption) (*SopNodePager, error) {
  577. pager := &SopNodePager{}
  578. for _, opt := range opts {
  579. opt(pager)
  580. }
  581. if pager.Order == nil {
  582. pager.Order = DefaultSopNodeOrder
  583. }
  584. return pager, nil
  585. }
  586. func (p *SopNodePager) ApplyFilter(query *SopNodeQuery) (*SopNodeQuery, error) {
  587. if p.Filter != nil {
  588. return p.Filter(query)
  589. }
  590. return query, nil
  591. }
  592. // SopNodePageList is SopNode PageList result.
  593. type SopNodePageList struct {
  594. List []*SopNode `json:"list"`
  595. PageDetails *PageDetails `json:"pageDetails"`
  596. }
  597. func (sn *SopNodeQuery) Page(
  598. ctx context.Context, pageNum uint64, pageSize uint64, opts ...SopNodePaginateOption,
  599. ) (*SopNodePageList, error) {
  600. pager, err := newSopNodePager(opts)
  601. if err != nil {
  602. return nil, err
  603. }
  604. if sn, err = pager.ApplyFilter(sn); err != nil {
  605. return nil, err
  606. }
  607. ret := &SopNodePageList{}
  608. ret.PageDetails = &PageDetails{
  609. Page: pageNum,
  610. Size: pageSize,
  611. }
  612. query := sn.Clone()
  613. query.ctx.Fields = nil
  614. count, err := query.Count(ctx)
  615. if err != nil {
  616. return nil, err
  617. }
  618. ret.PageDetails.Total = uint64(count)
  619. if pager.Order != nil {
  620. sn = sn.Order(pager.Order)
  621. } else {
  622. sn = sn.Order(DefaultSopNodeOrder)
  623. }
  624. sn = sn.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  625. list, err := sn.All(ctx)
  626. if err != nil {
  627. return nil, err
  628. }
  629. ret.List = list
  630. return ret, nil
  631. }
  632. type SopStagePager struct {
  633. Order sopstage.OrderOption
  634. Filter func(*SopStageQuery) (*SopStageQuery, error)
  635. }
  636. // SopStagePaginateOption enables pagination customization.
  637. type SopStagePaginateOption func(*SopStagePager)
  638. // DefaultSopStageOrder is the default ordering of SopStage.
  639. var DefaultSopStageOrder = Desc(sopstage.FieldID)
  640. func newSopStagePager(opts []SopStagePaginateOption) (*SopStagePager, error) {
  641. pager := &SopStagePager{}
  642. for _, opt := range opts {
  643. opt(pager)
  644. }
  645. if pager.Order == nil {
  646. pager.Order = DefaultSopStageOrder
  647. }
  648. return pager, nil
  649. }
  650. func (p *SopStagePager) ApplyFilter(query *SopStageQuery) (*SopStageQuery, error) {
  651. if p.Filter != nil {
  652. return p.Filter(query)
  653. }
  654. return query, nil
  655. }
  656. // SopStagePageList is SopStage PageList result.
  657. type SopStagePageList struct {
  658. List []*SopStage `json:"list"`
  659. PageDetails *PageDetails `json:"pageDetails"`
  660. }
  661. func (ss *SopStageQuery) Page(
  662. ctx context.Context, pageNum uint64, pageSize uint64, opts ...SopStagePaginateOption,
  663. ) (*SopStagePageList, error) {
  664. pager, err := newSopStagePager(opts)
  665. if err != nil {
  666. return nil, err
  667. }
  668. if ss, err = pager.ApplyFilter(ss); err != nil {
  669. return nil, err
  670. }
  671. ret := &SopStagePageList{}
  672. ret.PageDetails = &PageDetails{
  673. Page: pageNum,
  674. Size: pageSize,
  675. }
  676. query := ss.Clone()
  677. query.ctx.Fields = nil
  678. count, err := query.Count(ctx)
  679. if err != nil {
  680. return nil, err
  681. }
  682. ret.PageDetails.Total = uint64(count)
  683. if pager.Order != nil {
  684. ss = ss.Order(pager.Order)
  685. } else {
  686. ss = ss.Order(DefaultSopStageOrder)
  687. }
  688. ss = ss.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  689. list, err := ss.All(ctx)
  690. if err != nil {
  691. return nil, err
  692. }
  693. ret.List = list
  694. return ret, nil
  695. }
  696. type SopTaskPager struct {
  697. Order soptask.OrderOption
  698. Filter func(*SopTaskQuery) (*SopTaskQuery, error)
  699. }
  700. // SopTaskPaginateOption enables pagination customization.
  701. type SopTaskPaginateOption func(*SopTaskPager)
  702. // DefaultSopTaskOrder is the default ordering of SopTask.
  703. var DefaultSopTaskOrder = Desc(soptask.FieldID)
  704. func newSopTaskPager(opts []SopTaskPaginateOption) (*SopTaskPager, error) {
  705. pager := &SopTaskPager{}
  706. for _, opt := range opts {
  707. opt(pager)
  708. }
  709. if pager.Order == nil {
  710. pager.Order = DefaultSopTaskOrder
  711. }
  712. return pager, nil
  713. }
  714. func (p *SopTaskPager) ApplyFilter(query *SopTaskQuery) (*SopTaskQuery, error) {
  715. if p.Filter != nil {
  716. return p.Filter(query)
  717. }
  718. return query, nil
  719. }
  720. // SopTaskPageList is SopTask PageList result.
  721. type SopTaskPageList struct {
  722. List []*SopTask `json:"list"`
  723. PageDetails *PageDetails `json:"pageDetails"`
  724. }
  725. func (st *SopTaskQuery) Page(
  726. ctx context.Context, pageNum uint64, pageSize uint64, opts ...SopTaskPaginateOption,
  727. ) (*SopTaskPageList, error) {
  728. pager, err := newSopTaskPager(opts)
  729. if err != nil {
  730. return nil, err
  731. }
  732. if st, err = pager.ApplyFilter(st); err != nil {
  733. return nil, err
  734. }
  735. ret := &SopTaskPageList{}
  736. ret.PageDetails = &PageDetails{
  737. Page: pageNum,
  738. Size: pageSize,
  739. }
  740. query := st.Clone()
  741. query.ctx.Fields = nil
  742. count, err := query.Count(ctx)
  743. if err != nil {
  744. return nil, err
  745. }
  746. ret.PageDetails.Total = uint64(count)
  747. if pager.Order != nil {
  748. st = st.Order(pager.Order)
  749. } else {
  750. st = st.Order(DefaultSopTaskOrder)
  751. }
  752. st = st.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  753. list, err := st.All(ctx)
  754. if err != nil {
  755. return nil, err
  756. }
  757. ret.List = list
  758. return ret, nil
  759. }
  760. type WxPager struct {
  761. Order wx.OrderOption
  762. Filter func(*WxQuery) (*WxQuery, error)
  763. }
  764. // WxPaginateOption enables pagination customization.
  765. type WxPaginateOption func(*WxPager)
  766. // DefaultWxOrder is the default ordering of Wx.
  767. var DefaultWxOrder = Desc(wx.FieldID)
  768. func newWxPager(opts []WxPaginateOption) (*WxPager, error) {
  769. pager := &WxPager{}
  770. for _, opt := range opts {
  771. opt(pager)
  772. }
  773. if pager.Order == nil {
  774. pager.Order = DefaultWxOrder
  775. }
  776. return pager, nil
  777. }
  778. func (p *WxPager) ApplyFilter(query *WxQuery) (*WxQuery, error) {
  779. if p.Filter != nil {
  780. return p.Filter(query)
  781. }
  782. return query, nil
  783. }
  784. // WxPageList is Wx PageList result.
  785. type WxPageList struct {
  786. List []*Wx `json:"list"`
  787. PageDetails *PageDetails `json:"pageDetails"`
  788. }
  789. func (w *WxQuery) Page(
  790. ctx context.Context, pageNum uint64, pageSize uint64, opts ...WxPaginateOption,
  791. ) (*WxPageList, error) {
  792. pager, err := newWxPager(opts)
  793. if err != nil {
  794. return nil, err
  795. }
  796. if w, err = pager.ApplyFilter(w); err != nil {
  797. return nil, err
  798. }
  799. ret := &WxPageList{}
  800. ret.PageDetails = &PageDetails{
  801. Page: pageNum,
  802. Size: pageSize,
  803. }
  804. query := w.Clone()
  805. query.ctx.Fields = nil
  806. count, err := query.Count(ctx)
  807. if err != nil {
  808. return nil, err
  809. }
  810. ret.PageDetails.Total = uint64(count)
  811. if pager.Order != nil {
  812. w = w.Order(pager.Order)
  813. } else {
  814. w = w.Order(DefaultWxOrder)
  815. }
  816. w = w.Offset(int((pageNum - 1) * pageSize)).Limit(int(pageSize))
  817. list, err := w.All(ctx)
  818. if err != nil {
  819. return nil, err
  820. }
  821. ret.List = list
  822. return ret, nil
  823. }