115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package http
|
||
|
||
import (
|
||
"encoding/json"
|
||
"io"
|
||
"net/http"
|
||
|
||
"git.toowon.com/jimmy/go-common/middleware"
|
||
"git.toowon.com/jimmy/go-common/tools"
|
||
)
|
||
|
||
// PaginationRequest 分页请求结构
|
||
// 支持从JSON和form中解析分页参数
|
||
type PaginationRequest struct {
|
||
Page int `json:"page" form:"page"` // 页码，默认1
|
||
PageSize int `json:"page_size" form:"page_size"` // 每页数量
|
||
Keyword string `json:"keyword" form:"keyword"` // 关键字
|
||
}
|
||
|
||
// GetPage 获取页码，如果未设置则返回默认值1
|
||
func (p *PaginationRequest) GetPage() int {
|
||
if p.Page <= 0 {
|
||
return 1
|
||
}
|
||
return p.Page
|
||
}
|
||
|
||
// GetPageSize 获取每页数量，如果未设置则返回默认值20，最大限制100
|
||
func (p *PaginationRequest) GetPageSize() int {
|
||
size := p.PageSize
|
||
if size <= 0 {
|
||
return 20 // 默认20条
|
||
}
|
||
if size > 100 {
|
||
return 100 // 最大100条
|
||
}
|
||
return size
|
||
}
|
||
|
||
// GetOffset 计算数据库查询的偏移量
|
||
func (p *PaginationRequest) GetOffset() int {
|
||
return (p.GetPage() - 1) * p.GetPageSize()
|
||
}
|
||
|
||
// getPaginationFromQuery 从查询参数获取分页参数（内部辅助方法）
|
||
func getPaginationFromQuery(r *http.Request) (page, pageSize int) {
|
||
page = tools.ConvertInt(r.URL.Query().Get("page"), 0)
|
||
pageSize = tools.ConvertInt(r.URL.Query().Get("page_size"), 0)
|
||
return
|
||
}
|
||
|
||
// getPaginationFromForm 从form表单获取分页参数（内部辅助方法）
|
||
func getPaginationFromForm(r *http.Request) (page, pageSize int) {
|
||
page = tools.ConvertInt(r.FormValue("page"), 0)
|
||
pageSize = tools.ConvertInt(r.FormValue("page_size"), 0)
|
||
return
|
||
}
|
||
|
||
// ParsePaginationRequest 从请求中解析分页参数
|
||
// 支持从查询参数和form表单中解析
|
||
// 优先级：查询参数 > form表单
|
||
// 注意：如果请求体是JSON格式且包含分页字段，建议先使用ParseJSON解析完整请求体到包含PaginationRequest的结构体中
|
||
func ParsePaginationRequest(r *http.Request) *PaginationRequest {
|
||
req := &PaginationRequest{}
|
||
|
||
// 1. 从查询参数解析（优先级最高）
|
||
req.Page, req.PageSize = getPaginationFromQuery(r)
|
||
|
||
// 2. 如果查询参数中没有，尝试从form表单解析
|
||
if req.Page == 0 || req.PageSize == 0 {
|
||
page, pageSize := getPaginationFromForm(r)
|
||
if req.Page == 0 && page != 0 {
|
||
req.Page = page
|
||
}
|
||
if req.PageSize == 0 && pageSize != 0 {
|
||
req.PageSize = pageSize
|
||
}
|
||
}
|
||
|
||
return req
|
||
}
|
||
|
||
// ParseJSON 解析JSON请求体（公共方法）
|
||
// r: HTTP请求
|
||
// v: 目标结构体指针
|
||
func ParseJSON(r *http.Request, v interface{}) error {
|
||
body, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer r.Body.Close()
|
||
|
||
if len(body) == 0 {
|
||
return nil
|
||
}
|
||
|
||
return json.Unmarshal(body, v)
|
||
}
|
||
|
||
// GetTimezone 从请求的context中获取时区（公共方法）
|
||
// r: HTTP请求
|
||
// 如果使用了middleware.Timezone中间件，可以从context中获取时区信息
|
||
// 如果未设置，返回默认时区 AsiaShanghai
|
||
func GetTimezone(r *http.Request) string {
|
||
return middleware.GetTimezoneFromContext(r.Context())
|
||
}
|
||
|
||
// GetLanguage 从请求的context中获取语言（公共方法）
|
||
// r: HTTP请求
|
||
// 如果使用了middleware.Language中间件，可以从context中获取语言信息
|
||
// 如果未设置，返回默认语言 zh-CN
|
||
func GetLanguage(r *http.Request) string {
|
||
return middleware.GetLanguageFromContext(r.Context())
|
||
}
|