gin/request/form.go

77 lines
1.8 KiB
Go
Raw Normal View History

2022-07-03 00:56:37 +08:00
// Package request ...
//
// Description : request ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2022-07-03 00:33
package request
import (
2022-07-14 11:08:03 +08:00
"bytes"
2022-07-03 00:56:37 +08:00
"errors"
2024-09-20 17:31:23 +08:00
"github.com/mcuadros/go-defaults"
2024-07-23 21:40:53 +08:00
"io"
2022-07-14 11:08:03 +08:00
"io/ioutil"
2022-07-03 00:56:37 +08:00
"net/http"
"strings"
"github.com/go-playground/validator/v10"
2022-07-14 11:08:03 +08:00
"git.zhangdeman.cn/zhangdeman/gin/define"
2022-07-03 00:56:37 +08:00
"github.com/gin-gonic/gin"
)
// form 表单相关操作
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:51 2022/7/3
type form struct {
checkInstance *validator.Validate
2022-07-03 00:56:37 +08:00
}
2022-07-03 00:57:41 +08:00
// Parse 解析请求表单
2022-07-03 00:56:37 +08:00
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:34 2022/7/3
2022-07-03 00:57:41 +08:00
func (f *form) Parse(ctx *gin.Context, receiver interface{}) error {
2024-07-23 21:40:53 +08:00
handleConfig := define.GetHttpHandleConfig()
requestBody, _ := io.ReadAll(ctx.Request.Body)
2022-07-14 11:08:03 +08:00
// 请求信息写入上下文
2024-07-23 21:40:53 +08:00
ctx.Set(handleConfig.RecordRequestDataField, string(requestBody))
2022-07-14 11:08:03 +08:00
// 因为请求体被读一遍之后就没了,重新赋值 requestBody
ctx.Request.Body = ioutil.NopCloser(bytes.NewReader(requestBody))
2022-07-03 00:56:37 +08:00
method := strings.ToUpper(ctx.Request.Method)
if method == http.MethodGet ||
method == http.MethodPatch ||
method == http.MethodTrace ||
method == http.MethodConnect ||
method == http.MethodOptions {
2024-09-20 17:34:53 +08:00
if err := ctx.ShouldBindQuery(receiver); nil != err {
return err
}
} else if method == http.MethodPost ||
2022-07-03 00:56:37 +08:00
method == http.MethodPut ||
method == http.MethodDelete {
if ContentType.IsJson(ctx) {
2024-09-20 17:31:23 +08:00
if err := ctx.ShouldBindJSON(receiver); nil != err {
return err
}
} else if ContentType.IsFormURLEncoded(ctx) {
if err := ctx.ShouldBind(receiver); nil != err {
return err
}
} else {
return errors.New(ctx.ContentType() + " is not support")
2022-07-03 00:56:37 +08:00
}
2024-09-20 17:34:53 +08:00
} else {
2024-09-20 17:37:03 +08:00
return errors.New(method + " : request method is not support")
2022-07-03 00:56:37 +08:00
}
2024-09-20 17:31:23 +08:00
// 设置默认值
defaults.SetDefaults(receiver)
2024-09-20 17:37:03 +08:00
return nil
2022-07-03 00:56:37 +08:00
}