自己封装了一个GORM的分页插件

package paginate
/*
使用方法
	var demo entity.Demo
	pageReq := paginate.PageReq{Page: 1, PageSize: 10}
	pageResp, err := paginate.Page(db, &pageReq, &demo)
	for i, i2 := range pageResp.Data {
		fmt.Println(i)
		fmt.Println(i2)
	}
*/

import (
	"fmt"
	"gorm.io/gorm"
	"math"
)

// PageResp 分页返回值
// data: 使用范型，返回值是数组
type PageResp[T any] struct {
	TotalCount int64 `json:"total_count"` // 总条数
	TotalPage  int   `json:"total_page"`  // 总页数
	Data       []T   `json:"data"`        // 消息实体
	PageSize   int   `json:"page_size"`   // 每页条数
	Page       int   `json:"page"`        // 当前页数
}

// PageReq 分页输入参数
type PageReq struct {
	Page     int `json:"page" form:"page"`           // 第几页
	PageSize int `json:"page_size" form:"page_size"` // 每页条数
	Sort     int `json:"sort" form:"sort"`           // 排序
}

// Page 分页插件
func Page[T any](db *gorm.DB, pageReq *PageReq, model T) (*PageResp[T], error) {
	var (
		err    error
		count  int64
		result PageResp[T]
		data   []T
	)
	err = db.Model(model).Count(&count).Error
	fmt.Println(count)
	if err := db.Model(model).Scopes(paginate(pageReq)).Scan(&data).Error; err != nil {
		return nil, err
	}
	result.TotalCount = count
	result.Data = data
	result.Page = pageReq.Page
	result.PageSize = pageReq.PageSize
	result.TotalPage = int(math.Ceil(float64(count) / float64(pageReq.PageSize)))
	return &result, err
}

// 使用方法：db.Scopes(Paginate(pageReq)).Find(&users)
func paginate(pageReq *PageReq) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		var (
			page     = pageReq.Page
			pageSize = pageReq.PageSize
		)
		if page == 0 {
			page = 1
		}
		switch {
		case pageSize > 1000:
			pageSize = 1000
		case pageSize <= 0:
			pageSize = 10
		}
		offset := (page - 1) * pageSize
		return db.Offset(offset).Limit(pageSize)
	}
}