diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/AiAnalysisDifyController.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/AiAnalysisDifyController.java index ffb5f4e..8cf1dd9 100644 --- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/AiAnalysisDifyController.java +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/AiAnalysisDifyController.java @@ -1,7 +1,6 @@ package com.volvo.ai.analytic.center.controller; -import com.alibaba.fastjson.JSON; import com.fasterxml.jackson.core.JsonParser; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.DeserializationFeature; @@ -17,12 +16,11 @@ import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; import org.springframework.cloud.context.config.annotation.RefreshScope; +import org.springframework.messaging.handler.annotation.Header; import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.PostMapping; -import org.springframework.web.bind.annotation.RequestBody; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; @RestController() @@ -38,6 +36,9 @@ public class AiAnalysisDifyController { @Autowired private UserQuestionSummaryService userQuestionSummaryService; + @Value("${rocketmq.producer.corpus.dcctopic}") + private String corpusQuestionAppKey; + @PostMapping("/updateByAiId") @ApiOperation(value = "更新dify结果") public ResultMsg updateByAiId(@RequestBody String message) { @@ -80,9 +81,18 @@ public class AiAnalysisDifyController { @PostMapping("/corpusSummaryQuestion") @ApiOperation(value = "单条语料分析结论入库") - public AnalysisResp corpusSummaryQuestion(@RequestBody SummaryQuestionRequestDTO requestDTO) { - log.info("aiAnalyze corpusSummaryQuestion requestDTO: {}", JSON.toJSONString(requestDTO)); + public AnalysisResp corpusSummaryQuestion( + @Header String appKey, + @RequestParam("summaryReportId") String summaryReportId, + @RequestParam("preliminarySummaryResults") String preliminarySummaryResults) { + log.info("aiAnalyze corpusSummaryQuestion appKey={},summaryReportId={},preliminarySummaryResults={}", appKey, summaryReportId, preliminarySummaryResults); + if (!corpusQuestionAppKey.equals(appKey)) { + return AnalysisResp.failed("appKey 不正确 "); + } // 打印接收到的参数 + SummaryQuestionRequestDTO requestDTO = new SummaryQuestionRequestDTO(); + requestDTO.setSummaryReportId(summaryReportId); + requestDTO.setPreliminarySummaryResults(preliminarySummaryResults); userQuestionSummaryService.updateSummery(requestDTO); return AnalysisResp.success("ok"); }