54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package helper
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
|
|
"github.com/qianfree/team-api/relay/constant"
|
|
)
|
|
|
|
// RemapStatusCode 根据渠道配置重映射错误响应的 HTTP 状态码
|
|
// mappingJSON 格式如 {"429": 500, "403": 500}
|
|
func RemapStatusCode(err error, mappingJSON string) error {
|
|
if err == nil || mappingJSON == "" || mappingJSON == "{}" {
|
|
return err
|
|
}
|
|
|
|
relayErr, ok := err.(*constant.RelayError)
|
|
if !ok || relayErr.StatusCode == 0 || relayErr.StatusCode == 200 {
|
|
return err
|
|
}
|
|
|
|
mapping := make(map[string]any)
|
|
if err := json.Unmarshal([]byte(mappingJSON), &mapping); err != nil {
|
|
return relayErr // mapping 解析失败，返回原始错误
|
|
}
|
|
|
|
codeStr := strconv.Itoa(relayErr.StatusCode)
|
|
if mapped, exists := mapping[codeStr]; exists {
|
|
newCode := parseMappingValue(mapped)
|
|
if newCode > 0 {
|
|
relayErr.StatusCode = newCode
|
|
}
|
|
}
|
|
|
|
return relayErr
|
|
}
|
|
|
|
func parseMappingValue(v any) int {
|
|
switch val := v.(type) {
|
|
case string:
|
|
n, _ := strconv.Atoi(val)
|
|
return n
|
|
case float64:
|
|
return int(val)
|
|
case int:
|
|
return val
|
|
case json.Number:
|
|
n, _ := val.Int64()
|
|
return int(n)
|
|
default:
|
|
return 0
|
|
}
|
|
}
|