team/sim-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

f

Browse Source
This commit is contained in:
liangzai
2026-02-09 14:23:18 +08:00
parent f961970747
commit b6fd7ab9be
2 changed files with 190 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

145
app/main/api/internal/service/apirequestService.go
View File

@@ -57,6 +57,18 @@ type APIResponseData struct {
Error string `json:"error,omitempty"`
}
// ComboPackageResponseItem 组合包接口返回的单个子项
type ComboPackageResponseItem struct {
ApiCode string `json:"api_code"`
Success bool `json:"success"`
Data json.RawMessage `json:"data"`
}
// ComboPackageResponse 组合包接口返回结构
type ComboPackageResponse struct {
Responses []ComboPackageResponseItem `json:"responses"`
}
// ProcessRequests 处理请求
func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]byte, error) {
var ctx, cancel = context.WithCancel(context.Background())
@@ -85,6 +97,13 @@ func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]
if len(featureList) == 0 {
return nil, errors.New("处理请求错误，产品无对应接口功能")
}
// 若产品包含组合包（api_id 前四位为 COMB），则只调用该组合包，不再并发请求其他 feature
for _, f := range featureList {
if strings.HasPrefix(f.ApiId, "COMB") {
featureList = []*model.Feature{f}
break
}
}
var (
wg sync.WaitGroup
resultsCh = make(chan APIResponseData, len(featureList))
@@ -153,10 +172,20 @@ func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]
close(resultsCh)
close(errorsCh)
}()
// 收集所有结果并合并z
// 收集所有结果并合并；组合包（api_id 前四位为 COMB）返回的是多条 APIResponseData，需展开
var responseData []APIResponseData
for result := range resultsCh {
responseData = append(responseData, result)
if strings.HasPrefix(result.ApiID, "COMB") {
var comboItems []APIResponseData
if err := json.Unmarshal(result.Data, &comboItems); err != nil {
logx.WithContext(ctx).Errorf("组合包响应解析失败: %v", err)
responseData = append(responseData, result)
continue
}
responseData = append(responseData, comboItems...)
} else {
responseData = append(responseData, result)
}
}
if atomic.LoadInt32(&errorCount) >= int32(errorLimit) {
var allErrors []error
@@ -208,6 +237,9 @@ var requestProcessors = map[string]func(*ApiRequestService, []byte) ([]byte, err
"JRZQ7F1A": (*ApiRequestService).ProcessJRZQ7F1ARequest,
"IVYZ3P9M": (*ApiRequestService).ProcessIVYZ3P9MRequest,
"JRZQ6F2A": (*ApiRequestService).ProcessJRZQ6F2ARequest,
"COMBCZGR": (*ApiRequestService).ProcessCOMBCZGRRequest,
"COMBCZDQ": (*ApiRequestService).ProcessCOMBCZDQRequest,
"COMBCZQY": (*ApiRequestService).ProcessCOMBCZQYRequest,
}
// PreprocessRequestApi 调用指定的请求处理函数
@@ -1601,3 +1633,112 @@ func (a *ApiRequestService) ProcessJRZQ6F2ARequest(params []byte) ([]byte, error
return convertTianyuanResponse(resp)
}
// ProcessCOMBCZGRRequest 组合包 COMBCZGR：一次请求返回多个模块数据。必传 authorization_url, id_card, name, mobile_no
func (a *ApiRequestService) ProcessCOMBCZGRRequest(params []byte) ([]byte, error) {
authorizationURL := gjson.GetBytes(params, "authorization_url")
idCard := gjson.GetBytes(params, "id_card")
name := gjson.GetBytes(params, "name")
mobileNo := gjson.GetBytes(params, "mobile_no")
if !mobileNo.Exists() {
mobileNo = gjson.GetBytes(params, "mobile")
}
if !authorizationURL.Exists() || authorizationURL.String() == "" ||
!idCard.Exists() || !name.Exists() || !mobileNo.Exists() {
return nil, errors.New("api请求, COMBCZGR, 缺少必填参数: authorization_url / id_card / name / mobile_no(mobile)")
}
reqParams := map[string]interface{}{
"authorization_url": authorizationURL.String(),
"id_card": idCard.String(),
"name": name.String(),
"mobile_no": mobileNo.String(),
}
resp, err := a.tianyuanapi.CallInterface("COMBCZGR", reqParams)
if err != nil {
return nil, err
}
return a.comboResponseToAPIResponseData(resp)
}
// ProcessCOMBCZDQRequest 组合包 COMBCZDQ：一次请求返回多个模块数据。必传 name, mobile_no, authorization_url, authorized, id_card
func (a *ApiRequestService) ProcessCOMBCZDQRequest(params []byte) ([]byte, error) {
name := gjson.GetBytes(params, "name")
mobileNo := gjson.GetBytes(params, "mobile_no")
if !mobileNo.Exists() {
mobileNo = gjson.GetBytes(params, "mobile")
}
authorizationURL := gjson.GetBytes(params, "authorization_url")
authorized := gjson.GetBytes(params, "authorized")
idCard := gjson.GetBytes(params, "id_card")
if !name.Exists() || !mobileNo.Exists() || !authorizationURL.Exists() || authorizationURL.String() == "" ||
!authorized.Exists() || authorized.String() == "" || !idCard.Exists() {
return nil, errors.New("api请求, COMBCZDQ, 缺少必填参数: name / mobile_no(mobile) / authorization_url / authorized / id_card")
}
reqParams := map[string]interface{}{
"name": name.String(),
"mobile_no": mobileNo.String(),
"authorization_url": authorizationURL.String(),
"authorized": authorized.String(),
"id_card": idCard.String(),
}
resp, err := a.tianyuanapi.CallInterface("COMBCZDQ", reqParams)
if err != nil {
return nil, err
}
return a.comboResponseToAPIResponseData(resp)
}
// ProcessCOMBCZQYRequest 组合包 COMBCZQY：一次请求返回多个模块数据。必传 name, mobile_no, authorization_url, id_card
func (a *ApiRequestService) ProcessCOMBCZQYRequest(params []byte) ([]byte, error) {
name := gjson.GetBytes(params, "name")
mobileNo := gjson.GetBytes(params, "mobile_no")
if !mobileNo.Exists() {
mobileNo = gjson.GetBytes(params, "mobile")
}
authorizationURL := gjson.GetBytes(params, "authorization_url")
idCard := gjson.GetBytes(params, "id_card")
if !name.Exists() || !mobileNo.Exists() || !authorizationURL.Exists() || authorizationURL.String() == "" ||
!idCard.Exists() {
return nil, errors.New("api请求, COMBCZQY, 缺少必填参数: name / mobile_no(mobile) / authorization_url / id_card")
}
reqParams := map[string]interface{}{
"name": name.String(),
"mobile_no": mobileNo.String(),
"authorization_url": authorizationURL.String(),
"id_card": idCard.String(),
}
resp, err := a.tianyuanapi.CallInterface("COMBCZQY", reqParams)
if err != nil {
return nil, err
}
return a.comboResponseToAPIResponseData(resp)
}
// comboResponseToAPIResponseData 将组合包接口解密后的 Data 转为 []APIResponseData 并返回 JSON
func (a *ApiRequestService) comboResponseToAPIResponseData(resp *tianyuanapi.Response) ([]byte, error) {
dataBytes, err := json.Marshal(resp.Data)
if err != nil {
return nil, fmt.Errorf("组合包响应序列化失败: %w", err)
}
var combo ComboPackageResponse
if err := json.Unmarshal(dataBytes, &combo); err != nil {
return nil, fmt.Errorf("组合包响应解析失败: %w", err)
}
timestamp := time.Now().Format("2006-01-02 15:04:05")
out := make([]APIResponseData, 0, len(combo.Responses))
for _, item := range combo.Responses {
out = append(out, APIResponseData{
ApiID: item.ApiCode,
Data: item.Data,
Success: item.Success,
Timestamp: timestamp,
})
}
return json.Marshal(out)
}