package http import ( "net/http" "strconv" ) // getQueryInt 获取整数查询参数(内部方法,供ParsePaginationRequest使用) func getQueryInt(r *http.Request, key string, defaultValue int) int { value := r.URL.Query().Get(key) if value == "" { return defaultValue } intValue, err := strconv.Atoi(value) if err != nil { return defaultValue } return intValue } // getFormInt 获取表单整数(内部方法,供ParsePaginationRequest使用) func getFormInt(r *http.Request, key string, defaultValue int) int { value := r.FormValue(key) if value == "" { return defaultValue } intValue, err := strconv.Atoi(value) if err != nil { return defaultValue } return intValue } // PaginationRequest 分页请求结构 // 支持从JSON和form中解析分页参数 type PaginationRequest struct { Page int `json:"page" form:"page"` // 页码,默认1 Size int `json:"size" form:"size"` // 每页数量(兼容旧版本) PageSize int `json:"page_size" form:"page_size"` // 每页数量(推荐使用) } // GetPage 获取页码,如果未设置则返回默认值1 func (p *PaginationRequest) GetPage() int { if p.Page <= 0 { return 1 } return p.Page } // GetSize 获取每页数量,如果未设置则返回默认值20,最大限制100 // 优先使用 PageSize 字段,如果未设置则使用 Size 字段(兼容旧版本) func (p *PaginationRequest) GetSize() int { size := p.PageSize if size <= 0 { size = p.Size // 兼容旧版本的 Size 字段 } 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.GetSize() } // getPaginationFromQuery 从查询参数获取分页参数(内部辅助方法) func getPaginationFromQuery(r *http.Request) (page, size, pageSize int) { page = getQueryInt(r, "page", 0) size = getQueryInt(r, "size", 0) pageSize = getQueryInt(r, "page_size", 0) return } // getPaginationFromForm 从form表单获取分页参数(内部辅助方法) func getPaginationFromForm(r *http.Request) (page, size, pageSize int) { page = getFormInt(r, "page", 0) size = getFormInt(r, "size", 0) pageSize = getFormInt(r, "page_size", 0) return } // ParsePaginationRequest 从请求中解析分页参数 // 支持从查询参数和form表单中解析 // 优先级:查询参数 > form表单 // 注意:如果请求体是JSON格式且包含分页字段,建议先使用ParseJSON解析完整请求体到包含PaginationRequest的结构体中 func ParsePaginationRequest(r *http.Request) *PaginationRequest { req := &PaginationRequest{} // 1. 从查询参数解析(优先级最高) req.Page, req.Size, req.PageSize = getPaginationFromQuery(r) // 2. 如果查询参数中没有,尝试从form表单解析 if req.Page == 0 || (req.Size == 0 && req.PageSize == 0) { page, size, pageSize := getPaginationFromForm(r) if req.Page == 0 && page != 0 { req.Page = page } if req.Size == 0 && size != 0 { req.Size = size } if req.PageSize == 0 && pageSize != 0 { req.PageSize = pageSize } } return req }