wangyu 6 gadi atpakaļ
vecāks
revīzija
5989adc209

+ 3 - 2
icss-service/src/main/java/com/diagbot/facade/QuestionFacade.java

@@ -131,9 +131,10 @@ public class QuestionFacade extends QuestionInfoServiceImpl {
      *
      * @return
      */
-    public Map<Long,QuestionInfo> getQuestionInfos(){
+    public Map<Long,QuestionInfo> getQuestionInfos(List<Long> questionIdList){
         QueryWrapper<QuestionInfo> questionInfoQueryWrapper = new QueryWrapper<>();
-        questionInfoQueryWrapper.eq("is_deleted", IsDeleteEnum.N.getKey());
+        questionInfoQueryWrapper.eq("is_deleted", IsDeleteEnum.N.getKey())
+            .in("id",questionIdList);
         List<QuestionInfo> questionInfoList = this.list(questionInfoQueryWrapper);
         Map<Long,QuestionInfo> map = questionInfoList.stream().collect(Collectors.toMap(QuestionInfo::getId,questionInfo -> questionInfo));
         return map;

+ 6 - 1
icss-service/src/main/java/com/diagbot/facade/RetrievalFacade.java

@@ -9,6 +9,7 @@ import org.springframework.stereotype.Component;
 import org.springframework.web.bind.annotation.RequestBody;
 
 import javax.validation.Valid;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 
@@ -34,8 +35,12 @@ public class RetrievalFacade extends RetrievalServiceImpl {
         }
         //获取同义词标签信息
         List<RetrievalDTO> data = this.getSymptopInfo(retrievalVO);
+        List<Long> questionIdList = new ArrayList<>();
+        data.stream().forEach(
+                retrievalDTO -> questionIdList.add(retrievalDTO.getQuestionId())
+        );
         //将标签名放入实体
-        Map<Long, QuestionInfo> map = questionFacade.getQuestionInfos();
+        Map<Long, QuestionInfo> map = questionFacade.getQuestionInfos(questionIdList);
         for (RetrievalDTO retrievalDTO:data) {
             retrievalDTO.setName(map.get(retrievalDTO.getQuestionId()).getTagName());
         }