108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
|
package util
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|||
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ReverseMap 映射 payload 到 mapping
|
|||
|
|
func ReverseMap(mapping map[string]any, payload map[string]any) map[string]any {
|
|||
|
|
jsonObj := gjson.New("{}")
|
|||
|
|
for path, defaultValue := range mapping {
|
|||
|
|
val := gjson.New(payload).Get(path)
|
|||
|
|
if !val.IsNil() {
|
|||
|
|
_ = jsonObj.Set(path, val.Val())
|
|||
|
|
} else if defaultValue != nil {
|
|||
|
|
_ = jsonObj.Set(path, defaultValue)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return jsonObj.Map()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MapResponsePayload 映射模型响应为标准格式
|
|||
|
|
func MapResponsePayload(mapping map[string]any, responseBytes []byte) ([]byte, error) {
|
|||
|
|
if len(mapping) == 0 {
|
|||
|
|
return responseBytes, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
responseJson := gjson.New(responseBytes)
|
|||
|
|
resultJson := gjson.New("{}")
|
|||
|
|
|
|||
|
|
for standardField, modelPath := range mapping {
|
|||
|
|
path := gconv.String(modelPath)
|
|||
|
|
if path == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
val := responseJson.Get(path)
|
|||
|
|
if val.IsNil() {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
resultJson.Set(standardField, val.Val())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return []byte(resultJson.String()), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ParseHeadMsgHeaders 支持多个 header 绑定，逗号分隔：
|
|||
|
|
// 示例：
|
|||
|
|
// - X-API-Key:qwen3-tts-key,operation:true,count:123
|
|||
|
|
// - X-API-Key:"qwen3-tts-key",operation:"true"
|
|||
|
|
//
|
|||
|
|
// 说明：
|
|||
|
|
// - HTTP Header 最终都是字符串，这里做的是“值的字符串化表达”。
|
|||
|
|
// - 若 value 用双引号包裹，会去掉外层引号再注入，便于在配置中区分字符串/布尔/数字等表达（以及避免值中包含特殊字符时歧义）。
|
|||
|
|
func ParseHeadMsgHeaders(headMsg string) map[string]string {
|
|||
|
|
headMsg = strings.TrimSpace(headMsg)
|
|||
|
|
if headMsg == "" {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
out := map[string]string{}
|
|||
|
|
parts := strings.Split(headMsg, ",")
|
|||
|
|
for _, p := range parts {
|
|||
|
|
p = strings.TrimSpace(p)
|
|||
|
|
if p == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
// HeaderName:HeaderValue（推荐） / HeaderName=HeaderValue（兼容）
|
|||
|
|
if strings.Contains(p, ":") {
|
|||
|
|
kv := strings.SplitN(p, ":", 2)
|
|||
|
|
k := strings.TrimSpace(kv[0])
|
|||
|
|
v := strings.TrimSpace(kv[1])
|
|||
|
|
v = strings.Trim(v, "\"")
|
|||
|
|
if k != "" && v != "" {
|
|||
|
|
out[k] = v
|
|||
|
|
}
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if strings.Contains(p, "=") {
|
|||
|
|
kv := strings.SplitN(p, "=", 2)
|
|||
|
|
k := strings.TrimSpace(kv[0])
|
|||
|
|
v := strings.TrimSpace(kv[1])
|
|||
|
|
v = strings.Trim(v, "\"")
|
|||
|
|
if k != "" && v != "" {
|
|||
|
|
out[k] = v
|
|||
|
|
}
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(out) == 0 {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
return out
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PayloadToQuery 将 payload 转为 url.Values
|
|||
|
|
func PayloadToQuery(payload map[string]any) (url.Values, error) {
|
|||
|
|
q := url.Values{}
|
|||
|
|
for k, v := range payload {
|
|||
|
|
if v == nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
q.Set(k, gconv.String(v))
|
|||
|
|
}
|
|||
|
|
return q, nil
|
|||
|
|
}
|