71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
// Package parse_body ...
|
||
//
|
||
// Description : parse_body ...
|
||
//
|
||
// Author : go_developer@163.com<白茶清欢>
|
||
//
|
||
// Date : 2024-10-22 16:43
|
||
package parse_body
|
||
|
||
import (
|
||
"errors"
|
||
"git.zhangdeman.cn/zhangdeman/gin/request/abstract"
|
||
"github.com/gin-gonic/gin"
|
||
"strings"
|
||
)
|
||
|
||
var (
|
||
requestBodyParseAdaptorTable = map[string]abstract.RequestBodyParseAdaptor{}
|
||
)
|
||
|
||
func init() {
|
||
adaptorList := []abstract.RequestBodyParseAdaptor{
|
||
JsonAdaptor{},
|
||
FormUrlEncode{},
|
||
XmlAdaptor{},
|
||
}
|
||
for _, itemAdaptor := range adaptorList {
|
||
Register(itemAdaptor)
|
||
}
|
||
}
|
||
|
||
// Register 注册适配器实例
|
||
//
|
||
// Author : go_developer@163.com<白茶清欢>
|
||
//
|
||
// Date : 16:45 2024/10/22
|
||
func Register(adaptor abstract.RequestBodyParseAdaptor) {
|
||
if nil == adaptor {
|
||
return
|
||
}
|
||
requestBodyParseAdaptorTable[adaptor.ContentType()] = adaptor
|
||
}
|
||
|
||
// Execute 解析请求BODY数据
|
||
func Execute(ctx *gin.Context, receiver any) ([]byte, error) {
|
||
contentType := strings.ToLower(strings.ReplaceAll(ctx.ContentType(), " ", ""))
|
||
// 裁剪出真实的类型，之所以截取，是因为 content_type 中可能还包含编码信息, 如 : application/json;charset=utf8
|
||
contentTypeArr := strings.Split(contentType, ";")
|
||
contentType = contentTypeArr[0]
|
||
if _, exist := requestBodyParseAdaptorTable[contentType]; !exist {
|
||
return nil, errors.New(contentType + " : adaptor not found")
|
||
}
|
||
if parseResult, err := requestBodyParseAdaptorTable[contentType].Parse(ctx, receiver); nil != err {
|
||
return nil, err
|
||
} else {
|
||
return parseResult, err
|
||
}
|
||
}
|
||
|
||
// ExecuteForMap 高层级包装，表单解析为map
|
||
func ExecuteForMap(ctx *gin.Context) (map[string]any, error) {
|
||
var (
|
||
err error
|
||
result map[string]any
|
||
)
|
||
if _, err = Execute(ctx, &result); nil != err {
|
||
return nil, err
|
||
}
|
||
return result, nil
|
||
}
|