106 lines
1.7 KiB
Go
106 lines
1.7 KiB
Go
// Copyright 2024 Patial Tech (Ankit Patial).
|
|
//
|
|
// This file is part of code.patial.tech/go/appcore, which is MIT licensed.
|
|
// See http://opensource.org/licenses/MIT
|
|
|
|
package request
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"code.patial.tech/go/appcore/ptr"
|
|
)
|
|
|
|
// MaxPageSize is the maximum allowed page size to prevent resource exhaustion.
|
|
var MaxPageSize = 1000
|
|
|
|
type Pager struct {
|
|
OrderBy *string `json:"orderBy"`
|
|
OrderAsc *bool `json:"orderAsc"`
|
|
|
|
Search string
|
|
|
|
Page int `json:"page"`
|
|
Size int `json:"size"`
|
|
Total int `json:"total"`
|
|
|
|
TotalNeeded bool `json:"-"`
|
|
}
|
|
|
|
func (i *Pager) HasSearch() bool {
|
|
if i == nil {
|
|
return false
|
|
}
|
|
|
|
return strings.TrimSpace(i.Search) != ""
|
|
}
|
|
|
|
func (i *Pager) Offset() int {
|
|
if i == nil {
|
|
return 0
|
|
}
|
|
|
|
page := max(i.Page, 1)
|
|
|
|
return (page - 1) * i.Limit()
|
|
}
|
|
|
|
func (i *Pager) Limit() int {
|
|
if i == nil || ptr.GetNumber(&i.Size) < 1 {
|
|
return 20
|
|
}
|
|
|
|
return i.Size
|
|
}
|
|
|
|
// GetPager info from request's query parameters
|
|
func GetPager(r *http.Request) Pager {
|
|
q := r.URL.Query()
|
|
|
|
p := Pager{
|
|
Page: 1,
|
|
Size: 20,
|
|
OrderAsc: new(true),
|
|
}
|
|
|
|
if v := q.Get("q"); v != "" {
|
|
p.Search = strings.TrimSpace(v)
|
|
}
|
|
|
|
if v := q.Get("pg"); v != "" {
|
|
if vv, err := strconv.Atoi(v); err == nil && vv > 0 {
|
|
p.Page = vv
|
|
}
|
|
}
|
|
|
|
if v := q.Get("pg_s"); v != "" {
|
|
if vv, err := strconv.Atoi(v); err == nil && vv > 0 {
|
|
if vv > MaxPageSize {
|
|
vv = MaxPageSize
|
|
}
|
|
p.Size = vv
|
|
}
|
|
}
|
|
|
|
if v := q.Get("pg_ob"); v != "" {
|
|
p.OrderBy = &v
|
|
}
|
|
|
|
if v := q.Get("pg_o"); v != "" {
|
|
switch strings.ToLower(v) {
|
|
case "asc":
|
|
p.OrderAsc = new(true)
|
|
case "desc":
|
|
p.OrderAsc = new(false)
|
|
}
|
|
}
|
|
|
|
if v := q.Get("pg_t"); v == "n" {
|
|
p.TotalNeeded = true
|
|
}
|
|
|
|
return p
|
|
}
|