优化代码

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

@@ -9,9 +9,9 @@ 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;
+import java.util.stream.Collectors;
 
 /**
  * @Description:
@@ -23,6 +23,7 @@ public class RetrievalFacade extends RetrievalServiceImpl {
 
     @Autowired
     QuestionFacade questionFacade;
+
     /**
      * 获取标签信息
      *
@@ -30,18 +31,17 @@ public class RetrievalFacade extends RetrievalServiceImpl {
      * @return
      */
     public List<RetrievalDTO> getTagInfos(@Valid @RequestBody RetrievalVO retrievalVO) {
-        if(retrievalVO.getInputStr() == null || retrievalVO.getInputStr() == ""){
+        if (retrievalVO.getInputStr() == null || retrievalVO.getInputStr() == "") {
             retrievalVO.setInputStr(" ");
         }
         //获取同义词标签信息
         List<RetrievalDTO> data = this.getSymptopInfo(retrievalVO);
-        List<Long> questionIdList = new ArrayList<>();
-        data.stream().forEach(
-                retrievalDTO -> questionIdList.add(retrievalDTO.getQuestionId())
-        );
+        List<Long> questionIdList = data.stream()
+                .map(r -> r.getQuestionId())
+                .collect(Collectors.toList());
         //将标签名放入实体
         Map<Long, QuestionInfo> map = questionFacade.getQuestionInfos(questionIdList);
-        for (RetrievalDTO retrievalDTO:data) {
+        for (RetrievalDTO retrievalDTO : data) {
             retrievalDTO.setName(map.get(retrievalDTO.getQuestionId()).getTagName());
         }
         return data;