-
Notifications
You must be signed in to change notification settings - Fork 0
/
paginator.go
93 lines (78 loc) · 1.76 KB
/
paginator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package paginator
import (
"errors"
"math"
"gorm.io/gorm"
)
// Param parameter
type Param struct {
DB *gorm.DB
Page int
Limit int
OrderBy []string
ShowSQL bool
}
// Paginator return
type Paginator struct {
TotalRecord int64 `json:"total_record"`
TotalPage int `json:"total_page"`
Records interface{} `json:"records"`
Offset int `json:"offset"`
Limit int `json:"limit"`
Page int `json:"page"`
PrevPage int `json:"prev_page"`
NextPage int `json:"next_page"`
}
// Paging Pagination
func Paging(p *Param, result interface{}) (*Paginator, error) {
db := p.DB
if p.ShowSQL {
db = db.Debug()
}
if p.Page < 1 {
p.Page = 1
}
if p.Limit == 0 {
p.Limit = 10
}
if len(p.OrderBy) > 0 {
for _, o := range p.OrderBy {
db = db.Order(o)
}
}
done := make(chan bool, 1)
var paginator Paginator
var count int64
var offset int
go countRecords(db, result, done, &count)
if p.Page == 1 {
offset = 0
} else {
offset = (p.Page - 1) * p.Limit
}
if err := db.Limit(p.Limit).Offset(offset).Find(result).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, err
}
<-done
paginator.TotalRecord = count
paginator.Records = result
paginator.Page = p.Page
paginator.Offset = offset
paginator.Limit = p.Limit
paginator.TotalPage = int(math.Ceil(float64(count) / float64(p.Limit)))
if p.Page > 1 {
paginator.PrevPage = p.Page - 1
} else {
paginator.PrevPage = p.Page
}
if p.Page == paginator.TotalPage {
paginator.NextPage = p.Page
} else {
paginator.NextPage = p.Page + 1
}
return &paginator, nil
}
func countRecords(db *gorm.DB, anyType interface{}, done chan bool, count *int64) {
db.Model(anyType).Count(count)
done <- true
}