@@ -9,83 +9,83 @@ package pagehelper
99import "context"
1010
1111type builder struct {
12- page PageInfo
13- order OrderByInfo
12+ page PageInfo
13+ order OrderByInfo
1414
15- ctx context.Context
15+ ctx context.Context
1616}
1717
18- //创建builder
19- //ctx 初始context
18+ // 创建builder
19+ // ctx 初始context
2020func C (ctx context.Context ) * builder {
21- return & builder {ctx : ctx }
21+ return & builder {ctx : ctx }
2222}
2323
24- //分页
25- //page 页码
26- //pageSize 分页大小
27- func (b * builder ) Page (page , pageSize int ) * builder {
28- b .page .Page = page
29- b .page .PageSize = pageSize
30- b .page .total = 0
31- return b
24+ // 分页
25+ // page 页码
26+ // pageSize 分页大小
27+ func (b * builder ) Page (page , pageSize int64 ) * builder {
28+ b .page .Page = page
29+ b .page .PageSize = pageSize
30+ b .page .total = 0
31+ return b
3232}
3333
34- //分页
35- //page 页码
36- //pageSize 分页大小
37- func (b * builder ) PageWithCount (page , pageSize int , countColumn string ) * builder {
38- b .page .Page = page
39- b .page .PageSize = pageSize
40- b .page .countColumn = countColumn
41- b .page .total = - 1
42- return b
34+ // 分页
35+ // page 页码
36+ // pageSize 分页大小
37+ func (b * builder ) PageWithCount (page , pageSize int64 , countColumn string ) * builder {
38+ b .page .Page = page
39+ b .page .PageSize = pageSize
40+ b .page .countColumn = countColumn
41+ b .page .total = - 1
42+ return b
4343}
4444
4545func (b * builder ) Count (countColumn string ) * builder {
46- b .page .countColumn = countColumn
47- b .page .total = - 1
48- return b
46+ b .page .countColumn = countColumn
47+ b .page .total = - 1
48+ return b
4949}
5050
51- //手动指定字段和排序
52- //field 字段
53- //order 排序 [ASC | DESC]
51+ // 手动指定字段和排序
52+ // field 字段
53+ // order 排序 [ASC | DESC]
5454func (b * builder ) OrderBy (field , order string ) * builder {
55- b .order .Field = field
56- b .order .Order = order
57- return b
55+ b .order .Field = field
56+ b .order .Order = order
57+ return b
5858}
5959
60- //升序(默认)
61- //field 字段
60+ // 升序(默认)
61+ // field 字段
6262func (b * builder ) ASC (field string ) * builder {
63- return b .OrderBy (field , ASC )
63+ return b .OrderBy (field , ASC )
6464}
6565
66- //降序
67- //field 字段
66+ // 降序
67+ // field 字段
6868func (b * builder ) DESC (field string ) * builder {
69- return b .OrderBy (field , DESC )
69+ return b .OrderBy (field , DESC )
7070}
7171
72- //获得含分页/排序信息的context
72+ // 获得含分页/排序信息的context
7373func (b * builder ) Build () context.Context {
74- if b .page .PageSize > 0 {
75- if b .page .total != - 1 {
76- b .ctx = StartPage (b .ctx , b .page .Page , b .page .PageSize )
77- } else {
78- b .ctx = StartPageWithCount (b .ctx , b .page .Page , b .page .PageSize , b .page .countColumn )
79- }
80- }
74+ if b .page .PageSize > 0 {
75+ if b .page .total != - 1 {
76+ b .ctx = StartPage (b .ctx , b .page .Page , b .page .PageSize )
77+ } else {
78+ b .ctx = StartPageWithCount (b .ctx , b .page .Page , b .page .PageSize , b .page .countColumn )
79+ }
80+ }
8181
82- if b .order .Field != "" {
83- if b .order .Order == "" {
84- b .order .Order = ASC
85- }
82+ if b .order .Field != "" {
83+ if b .order .Order == "" {
84+ b .order .Order = ASC
85+ }
8686
87- b .ctx = OrderBy (b .ctx , b .order .Field , b .order .Order )
88- }
87+ b .ctx = OrderBy (b .ctx , b .order .Field , b .order .Order )
88+ }
8989
90- return b .ctx
90+ return b .ctx
9191}
0 commit comments