forked from netlify/gotrue
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pagination.go
69 lines (61 loc) · 1.56 KB
/
pagination.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package api
import (
"fmt"
"net/http"
"net/url"
"strconv"
"github.com/netlify/gotrue/models"
)
const defaultPerPage = 50
func calculateTotalPages(perPage, total uint64) uint64 {
pages := total / perPage
if total%perPage > 0 {
return pages + 1
}
return pages
}
func addPaginationHeaders(w http.ResponseWriter, r *http.Request, p *models.Pagination) {
totalPages := calculateTotalPages(p.PerPage, p.Count)
url, _ := url.ParseRequestURI(r.URL.String())
query := url.Query()
header := ""
if totalPages > p.Page {
query.Set("page", fmt.Sprintf("%v", p.Page+1))
url.RawQuery = query.Encode()
header += "<" + url.String() + ">; rel=\"next\", "
}
if p.Page > 1 {
query.Set("page", fmt.Sprintf("%v", p.Page-1))
url.RawQuery = query.Encode()
header += "<" + url.String() + ">; rel=\"prev\", "
}
query.Set("page", fmt.Sprintf("%v", totalPages))
url.RawQuery = query.Encode()
header += "<" + url.String() + ">; rel=\"last\""
w.Header().Add("Link", header)
w.Header().Add("X-Total-Count", fmt.Sprintf("%v", p.Count))
}
func paginate(r *http.Request) (*models.Pagination, error) {
params := r.URL.Query()
queryPage := params.Get("page")
queryPerPage := params.Get("per_page")
var page uint64 = 1
var perPage uint64 = defaultPerPage
var err error
if queryPage != "" {
page, err = strconv.ParseUint(queryPage, 10, 64)
if err != nil {
return nil, err
}
}
if queryPerPage != "" {
perPage, err = strconv.ParseUint(queryPerPage, 10, 64)
if err != nil {
return nil, err
}
}
return &models.Pagination{
Page: page,
PerPage: perPage,
}, nil
}