diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/TestController.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/TestController.java index 34c6021..2173b17 100644 --- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/TestController.java +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/TestController.java @@ -2,6 +2,8 @@ package com.volvo.ai.analytic.center.controller; import com.alibaba.fastjson.JSONObject; +import com.volvo.ai.analytic.center.entity.AiAnalysisRequestLogs; +import com.volvo.ai.analytic.center.service.AiAnalysisRequestLogsService; import com.volvo.common.core.util.ResultMsg; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; @@ -25,6 +27,9 @@ public class TestController { @Autowired private RocketMQTemplate rocketMQTemplate; + @Autowired + private AiAnalysisRequestLogsService aiAnalysisRequestLogsService; + @PostMapping("/mockMq") @@ -35,6 +40,16 @@ public class TestController { return ResultMsg.ok("ok"); } + @PostMapping("/updateByAiId") + @ApiOperation(value = "更新dify结果") + public ResultMsg updateByAiId(@RequestBody String message) { + JSONObject messageJson = JSONObject.parseObject(message); + AiAnalysisRequestLogs aiAnalysisRequestLogs = new AiAnalysisRequestLogs(); + aiAnalysisRequestLogs.setAiAnalysisRequestId(messageJson.getString("aiAnalysisRequestId")); + aiAnalysisRequestLogs.setDifyResponse(messageJson.getString("difyResponse")); + aiAnalysisRequestLogsService.saveAiAnalysisRequestLogs(aiAnalysisRequestLogs); + return ResultMsg.ok("ok"); + } }