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
import (
	"gorm.io/gorm"
)

// PageResults 分页结果
type PageResults struct {
	Total    int64       `json:"total"` // 总数
	PageNum  int         `json:"page"`  // 当前分页数
	PageSize int         `json:"size"`  // 当前分页大小
	Pages    int         `json:"pages"` // 分页总数
	Data     interface{} `json:"data"`  // 返回数据
}

// PageInfo 分页参数
type PageInfo struct {
	Page int `json:"page" form:"page"`
	Size int `json:"Size" form:"size"`
}

// check 检查分页数
func (p *PageInfo) check() {
	if p.Page <= 0 {
		p.Page = 1
	}
	if p.Size <= 0 {
		p.Size = 10
	}
}

// Paginate 进行分页
func (p *PageInfo) Paginate() func(db *gorm.DB) *gorm.DB {
	p.check()
	return func(db *gorm.DB) *gorm.DB {
		offset := (p.Page - 1) * p.Size
		return db.Offset(offset).Limit(p.Size)
	}
}

// BuildPageResults 生成分页结果
func (p *PageInfo) BuildPageResults(data interface{}, total int64) *PageResults {
	t := int(total)
	pages := t / p.Size
	if t%p.Size != 0 {
		pages++
	}
	return &PageResults{
		Total:    total,
		PageNum:  p.Page,
		PageSize: p.Size,
		Pages:    pages,
		Data:     data,
	}
}

req类

1
2
3
4
type XxxReq struct {
	Xxx       string   `json:"Xxx" form:"Xxx"`
	PageInfo  PageInfo `json:"pageInfo"`
}

controller层

1
2
3
4
5
6
7
8
9
10
11
12
13
func (handler *XxxHandler) GetPage(context *gin.Context) {
	queryCondition := model.XxxReq{}
	if err := context.ShouldBind(&queryCondition); err != nil {
		util.ResponseError(context, util.CodeBadRequest, err)
		return
	}
	list, total, err := handler.drillService.GetPage(&queryCondition)
	if err != nil {
		util.ResponseError(context, util.CodeFoundFailed, err)
		return
	}
	util.ResponseOk(context, queryCondition.PageInfo.BuildPageResults(list, total))
}

dao层

1
2
3
4
5
6
7
8
func (dao *XxxDao) GetPage(req *model.XxxReq) (Xxx *[]model.Xxx, total int64, err error) {
	db := dao.Mysql.Model(model.Xxx{})
	if err = db.Count(&total).Error; err != nil {
		return
	}
	err = db.Scopes(req.PageInfo.Paginate()).Find(&Xxx).Error
	return
}