156 lines
3.0 KiB
Go
156 lines
3.0 KiB
Go
|
/*
|
||
|
* @Date: 2021-04-27 13:50:37
|
||
|
* @LastEditors: viletyy
|
||
|
* @LastEditTime: 2021-04-27 14:31:42
|
||
|
* @FilePath: /yolk/paginator.go
|
||
|
*/
|
||
|
package yolk
|
||
|
|
||
|
import (
|
||
|
"math"
|
||
|
"net/http"
|
||
|
"net/url"
|
||
|
"strconv"
|
||
|
)
|
||
|
|
||
|
type Paginator struct {
|
||
|
Request *http.Request
|
||
|
PerPageNums int // 每页个数
|
||
|
MaxPages int // 最大页
|
||
|
nums int64 // 总个数
|
||
|
pageRange []int // 页码可选择范围
|
||
|
pageNums int // 最大页
|
||
|
page int // 当前页
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageNums() int {
|
||
|
if p.pageNums != 0 {
|
||
|
return p.pageNums
|
||
|
}
|
||
|
pageNums := math.Ceil(float64(p.nums) / float64(p.PerPageNums))
|
||
|
if p.MaxPages > 0 {
|
||
|
pageNums = math.Min(pageNums, float64(p.MaxPages))
|
||
|
}
|
||
|
p.pageNums = int(pageNums)
|
||
|
return p.pageNums
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) Nums() int64 {
|
||
|
return p.nums
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) SetNums(nums interface{}) {
|
||
|
p.nums, _ = ToInt64(nums)
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) Page() int {
|
||
|
if p.page != 0 {
|
||
|
return p.page
|
||
|
}
|
||
|
if p.Request.Form == nil {
|
||
|
p.Request.ParseForm()
|
||
|
}
|
||
|
p.page, _ = strconv.Atoi(p.Request.Form.Get("page"))
|
||
|
if p.page > p.PageNums() {
|
||
|
p.page = p.PageNums()
|
||
|
}
|
||
|
if p.page <= 0 {
|
||
|
p.page = 1
|
||
|
}
|
||
|
return p.page
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) Pages() []int {
|
||
|
if p.pageRange == nil && p.nums > 0 {
|
||
|
var pages []int
|
||
|
pageNums := p.PageNums()
|
||
|
page := p.Page()
|
||
|
switch {
|
||
|
case page >= pageNums-4 && pageNums > 9:
|
||
|
start := pageNums - 9 + 1
|
||
|
pages = make([]int, 9)
|
||
|
for i, _ := range pages {
|
||
|
pages[i] = start + i
|
||
|
}
|
||
|
case page >= 5 && pageNums > 9:
|
||
|
start := page - 5 + 1
|
||
|
pages = make([]int, int(math.Min(9, float64(page+4+1))))
|
||
|
for i, _ := range pages {
|
||
|
pages[i] = start + 1
|
||
|
}
|
||
|
default:
|
||
|
pages = make([]int, int(math.Min(9, float64(pageNums))))
|
||
|
for i, _ := range pages {
|
||
|
pages[i] = i + 1
|
||
|
}
|
||
|
}
|
||
|
p.pageRange = pages
|
||
|
}
|
||
|
return p.pageRange
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageLink(page int) string {
|
||
|
link, _ := url.ParseRequestURI(p.Request.RequestURI)
|
||
|
values := link.Query()
|
||
|
if page == 1 {
|
||
|
values.Del("page")
|
||
|
} else {
|
||
|
values.Set("page", strconv.Itoa(page))
|
||
|
}
|
||
|
link.RawQuery = values.Encode()
|
||
|
return link.String()
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageLinkPrev() (link string) {
|
||
|
if p.HasPrev() {
|
||
|
link = p.PageLink(p.Page() - 1)
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageLinkNext() (link string) {
|
||
|
if p.HasNext() {
|
||
|
link = p.PageLink(p.Page() + 1)
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageLinkFirst() (link string) {
|
||
|
return p.PageLink(1)
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) PageLinkLast() (link string) {
|
||
|
return p.PageLink(p.PageNums())
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) HasPrev() bool {
|
||
|
return p.Page() > 1
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) HasNext() bool {
|
||
|
return p.Page() < p.PageNums()
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) IsActive(page int) bool {
|
||
|
return p.Page() == page
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) Offset() int {
|
||
|
return (p.Page() - 1) * p.PerPageNums
|
||
|
}
|
||
|
|
||
|
func (p *Paginator) HasPages() bool {
|
||
|
return p.PageNums() > 1
|
||
|
}
|
||
|
|
||
|
func NewPaginator(req *http.Request, per int, nums interface{}) *Paginator {
|
||
|
p := Paginator{}
|
||
|
p.Request = req
|
||
|
if per <= 0 {
|
||
|
per = 10
|
||
|
}
|
||
|
p.PerPageNums = per
|
||
|
p.SetNums(nums)
|
||
|
return &p
|
||
|
}
|