kpt-tmr-group/http/middleware/pagination.go

package middleware

import (
	"fmt"
	"strconv"

	"github.com/gin-gonic/gin"
)

const (
	Page       = "page"
	PageSize   = "page_size"
	PageOffset = "page_offset"
)

// Pagination sets page, pageSize and pageOffset to *gin.Context
func Pagination() gin.HandlerFunc {
	return func(c *gin.Context) {
		page := getSetItem(c, Page, 1)
		size := getSetItem(c, PageSize, 20)
		c.Set(PageOffset, (page-1)*size)
		c.Next()
	}
}

func getSetItem(c *gin.Context, k string, d int) int {
	var n int
	if v := c.Query(k); v != "" {
		if i, err := strconv.Atoi(v); err == nil {
			if i > 0 {
				n = i
			}
		}
	}

	if n == 0 {
		n = d
	}

	c.Set(k, n)
	c.Request.Header.Set(k, fmt.Sprintf("%d", n))
	return n
}