package gin_frame

import "gorm.io/gorm"

// PageInfo 分页请求
type PageInfo struct {
	Page int `form:"page" json:"page" comment:"页码"`
	Size int `form:"size" json:"size" comment:"每页条数"`
}

// PageListResp 分页列表 响应
type PageListResp struct {
	Other    interface{} `json:"other"`
	List     interface{} `json:"list"`
	Total    int64       `json:"total"`
	PageNum  int         `json:"page_num"`
	PageSize int         `json:"page_size"`
}

// PageList 分页获取数据  默认每页10条
func PageList(query *gorm.DB, param PageInfo) (*gorm.DB, PageListResp) {
	if param.Page <= 0 {
		param.Page = 1
	}

	if param.Size <= 0 {
		param.Size = 10
	}

	//计算总页数
	query1 := query.Session(&gorm.Session{})
	total := int64(0) //总条数
	query1.Count(&total)
	pageNum := int(total) / param.Size //总页数
	if int(total)%param.Size != 0 {
		pageNum++
	}

	return query.Limit(param.Size).Offset((param.Page - 1) * param.Size), PageListResp{
		Total:    total,
		PageNum:  pageNum,
		PageSize: param.Size,
	}
}