48 lines
862 B
Go
48 lines
862 B
Go
package utils
|
||
|
||
import "gorm.io/gorm"
|
||
|
||
// Page
|
||
// @description: 分页组件
|
||
// @param current
|
||
// @param size
|
||
// @return func(db *gorm.DB) *gorm.DB
|
||
func Page(current, size int) func(db *gorm.DB) *gorm.DB {
|
||
// 如果页码是-1，就不分页
|
||
if current == -1 {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
return db
|
||
}
|
||
}
|
||
// 分页
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if current == 0 {
|
||
current = 1
|
||
}
|
||
if size < 1 {
|
||
size = 10
|
||
}
|
||
// 计算偏移量
|
||
offset := (current - 1) * size
|
||
// 返回组装结果
|
||
return db.Offset(offset).Limit(size)
|
||
}
|
||
}
|
||
|
||
// GenTotalPage
|
||
// @description: 计算页码数
|
||
// @param count
|
||
// @param size
|
||
// @return int
|
||
func GenTotalPage(count int64, size int) int {
|
||
totalPage := 0
|
||
if count > 0 {
|
||
upPage := 0
|
||
if int(count)%size > 0 {
|
||
upPage = 1
|
||
}
|
||
totalPage = (int(count) / size) + upPage
|
||
}
|
||
return totalPage
|
||
}
|