91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"model-gateway/model/entity"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
tgjson "github.com/tidwall/gjson"
|
|
)
|
|
|
|
// ======================== 响应解析 ========================
|
|
|
|
// ParseAndValidate 解析模型响应，校验必填字段，返回标准 rounds 格式
|
|
func ParseAndValidate(raw map[string]any, requiredFields []string) (map[string]any, error) {
|
|
contentStr := gconv.String(raw[entity.ResponseBody])
|
|
if strings.TrimSpace(contentStr) == "" {
|
|
return raw, fmt.Errorf("字段 %s 为空", entity.ResponseBody)
|
|
}
|
|
|
|
contentStr = cleanControlChars(contentStr)
|
|
|
|
var arr []any
|
|
if err := json.Unmarshal([]byte(contentStr), &arr); err != nil {
|
|
return raw, fmt.Errorf("JSON解析失败: %w", err)
|
|
}
|
|
if len(arr) == 0 {
|
|
return raw, fmt.Errorf("解析后数组为空")
|
|
}
|
|
|
|
for _, field := range requiredFields {
|
|
for i, r := range arr {
|
|
round, _ := r.(map[string]any)
|
|
if round != nil && gjson.New(round).Get(field).IsNil() {
|
|
return raw, fmt.Errorf("rounds[%d] 缺少必填字段: %s", i, field)
|
|
}
|
|
}
|
|
}
|
|
return map[string]any{"total_rounds": len(arr), "rounds": arr}, nil
|
|
}
|
|
|
|
// MapResponsePayload 映射模型响应为标准格式
|
|
func MapResponsePayload(mapping map[string]any, result map[string]any) (map[string]any, error) {
|
|
if len(mapping) == 0 {
|
|
return result, nil
|
|
}
|
|
|
|
// 把 result 转成 JSON 字符串，tidwall/gjson 需要字符串输入
|
|
resultBytes, _ := json.Marshal(result)
|
|
resultStr := string(resultBytes)
|
|
|
|
mapped := make(map[string]any)
|
|
|
|
for standardField, modelPath := range mapping {
|
|
path := gconv.String(modelPath)
|
|
if path == "" {
|
|
continue
|
|
}
|
|
|
|
value := tgjson.Get(resultStr, path)
|
|
if !value.Exists() {
|
|
continue
|
|
}
|
|
|
|
if strings.Contains(path, "#") {
|
|
var arr []any
|
|
for _, v := range value.Array() {
|
|
arr = append(arr, v.Value())
|
|
}
|
|
mapped[standardField] = arr
|
|
} else {
|
|
mapped[standardField] = value.Value()
|
|
}
|
|
}
|
|
|
|
return mapped, nil
|
|
}
|
|
|
|
// ======================== 内部辅助 ========================
|
|
|
|
func cleanControlChars(s string) string {
|
|
return strings.Map(func(r rune) rune {
|
|
if r < 32 && r != ' ' {
|
|
return -1
|
|
}
|
|
return r
|
|
}, s)
|
|
}
|