修改失败补偿

This commit is contained in:
zren25
2025-03-14 18:33:49 +08:00
parent 696ee93f61
commit 890017d305
2 changed files with 15 additions and 5 deletions

View File

@@ -129,14 +129,14 @@ public class CorpusFailJob {
} }
oldAiAnalysisRequestLogs.setDifyResponse(JSON.toJSONString(execDifyFlow)); oldAiAnalysisRequestLogs.setDifyResponse(JSON.toJSONString(execDifyFlow));
aiAnalysisRequestLogsService.saveAiAnalysisRequestLogs(oldAiAnalysisRequestLogs); updateDiFyRequest(oldAiAnalysisRequestLogs,oldAiAnalysisRequestLogs.getAiAnalysisRequestId());
aiAnalysisErrors.setRetryCount(aiAnalysisErrors.getRetryCount() + 1); aiAnalysisErrors.setRetryCount(aiAnalysisErrors.getRetryCount() + 1);
aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("1"); aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("1");
aiAnalysisErrorsService.saveAiAnalysisErrors(aiAnalysisErrors); updateAiAnalysisErrors(aiAnalysisErrors,oldAiAnalysisRequestLogs.getAiAnalysisRequestId());
} }
} catch (Exception e) { } catch (Exception e) {
aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("0"); aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("0");
aiAnalysisErrorsService.saveAiAnalysisErrors(aiAnalysisErrors); updateAiAnalysisErrors(aiAnalysisErrors,aiAnalysisErrors.getAiAnalysisRequestId());
} }
}); });
@@ -144,4 +144,14 @@ public class CorpusFailJob {
} }
protected void updateDiFyRequest(AiAnalysisRequestLogs oldAiAnalysisRequestLogs, String aiAnalysisRequestId) {
aiAnalysisRequestLogsMapper.update(new AiAnalysisRequestLogs(),
new LambdaQueryWrapper<AiAnalysisRequestLogs>().setEntity(oldAiAnalysisRequestLogs)
.eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, aiAnalysisRequestId));
}
protected void updateAiAnalysisErrors(AiAnalysisErrors aiAnalysisErrors, String aiAnalysisRequestId) {
aiAnalysisErrorsService.update( new LambdaQueryWrapper<AiAnalysisErrors>().setEntity(aiAnalysisErrors)
.eq(AiAnalysisErrors::getAiAnalysisRequestId, aiAnalysisRequestId));
}
} }

View File

@@ -98,8 +98,8 @@ public class DiFyServiceImpl implements DiFyService{
map.put("inputs",diFyReq.getInputs()); map.put("inputs",diFyReq.getInputs());
map.put("response_mode","blocking"); map.put("response_mode","blocking");
map.put("user",diFyReq.getUser()); map.put("user",diFyReq.getUser());
JSONObject difyResult = diFyFeign.runWorkflows("Bearer "+diFyReq.getFlowId(),map); JSONObject difyResult = diFyFeign.runWorkflows("Bearer "+diFyReq.getFlowId(),map);
return difyResult; JSONObject data = difyResult.getJSONObject("data");
return data;
} }
} }