73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
|
package paginator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/gofiber/fiber/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
defaultLimit = 10
|
||
|
|
defaultSort = "desc"
|
||
|
|
defaultSortBy = "id"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Pagination struct {
|
||
|
|
Limit int `json:"limit,omitempty"`
|
||
|
|
Offset int `json:"-"`
|
||
|
|
Page int `json:"page,omitempty"`
|
||
|
|
NextPage int `json:"nextPage,omitempty"`
|
||
|
|
PreviousPage int `json:"previousPage,omitempty"`
|
||
|
|
Count int64 `json:"count,omitempty"`
|
||
|
|
TotalPage int `json:"totalPage,omitempty"`
|
||
|
|
Sort string `json:"sort,omitempty"`
|
||
|
|
SortBy string `json:"sortBy,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func Paging(p *Pagination) *Pagination {
|
||
|
|
p.TotalPage = int(math.Ceil(float64(p.Count) / float64(p.Limit)))
|
||
|
|
if p.Page > 1 {
|
||
|
|
p.PreviousPage = p.Page - 1
|
||
|
|
} else {
|
||
|
|
p.PreviousPage = p.Page
|
||
|
|
}
|
||
|
|
if p.Page == p.TotalPage {
|
||
|
|
p.NextPage = p.Page
|
||
|
|
} else {
|
||
|
|
p.NextPage = p.Page + 1
|
||
|
|
}
|
||
|
|
return p
|
||
|
|
}
|
||
|
|
|
||
|
|
func Paginate(c *fiber.Ctx) (*Pagination, error) {
|
||
|
|
limit, err := strconv.Atoi(c.Query("limit"))
|
||
|
|
if err != nil {
|
||
|
|
limit = defaultLimit
|
||
|
|
}
|
||
|
|
page, err := strconv.Atoi(c.Query("page"))
|
||
|
|
if err != nil {
|
||
|
|
page = 1
|
||
|
|
}
|
||
|
|
sort := c.Query("sort")
|
||
|
|
if sort == "" {
|
||
|
|
sort = defaultSort
|
||
|
|
}
|
||
|
|
sortBy := c.Query("sortBy")
|
||
|
|
if sortBy == "" {
|
||
|
|
sortBy = defaultSortBy
|
||
|
|
}
|
||
|
|
p := &Pagination{
|
||
|
|
Limit: limit,
|
||
|
|
Page: page,
|
||
|
|
Sort: sort,
|
||
|
|
SortBy: sortBy,
|
||
|
|
}
|
||
|
|
if p.Page == 0 {
|
||
|
|
p.Page = 1
|
||
|
|
}
|
||
|
|
|
||
|
|
p.Offset = (p.Page - 1) * p.Limit
|
||
|
|
return p, nil
|
||
|
|
}
|