更新症状查询的功能

src/main/java/com/diagbot/facade/RetrievalFacade.java

@@ -102,7 +102,7 @@ public class RetrievalFacade {
                 retrievalDTO.setTransfusionNames(getTransfusion(retrievalVO.getInputStr()));
                 break;
             case 9:
-                retrievalDTO.setSymptomNames(getSymptoms(retrievalVO.getInputStr()));
+                retrievalDTO.setSymptomNames(getSymptoms(retrievalVO.getInputStr(), retrievalVO.getSize()));
                 break;
         }
 
@@ -207,10 +207,10 @@ public class RetrievalFacade {
     /**
      * 获取症状列表
      */
-    public List<String> getSymptoms(String name) {
+    public List<String> getSymptoms(String name, Integer size) {
 
         SymptomNameNode symptomNameNode = new SymptomNameNode();
-        List<String> symplist = symptomNameNode.getSymptomList(name, symptomNameRepository, baseNodeRepository);
+        List<String> symplist = symptomNameNode.getSymptomList(name, size, symptomNameRepository, baseNodeRepository);
 
         return symplist;
     }

src/main/java/com/diagbot/repository/SymptomNameNode.java

@@ -11,12 +11,16 @@ public class SymptomNameNode {
     /**
      * 获取症状名称列表
      */
-    public List<String> getSymptomList(String name, SymptomNameRepository symptomNameRepository, BaseNodeRepository baseNodeRepository) {
+    public List<String> getSymptomList(String name, Integer size, SymptomNameRepository symptomNameRepository, BaseNodeRepository baseNodeRepository) {
         List<String> symplt = new ArrayList<>();
-        List<Symptom> symplist = symptomNameRepository.index(name, name, 100);
+        if (null == size) {
+            size = 100;
+        }
+
+        List<Symptom> symplist = symptomNameRepository.index(name, name, size);
 
-        if (ListUtil.isNotEmpty(symplist) && symplist.size() > 100) {
-            symplist = symplist.subList(0, 100);
+        if (ListUtil.isNotEmpty(symplist) && symplist.size() > size) {
+            symplist = symplist.subList(0, size);
         }
 
         for (Symptom symptom : symplist) {

src/main/java/com/diagbot/vo/RetrievalVO.java

@@ -27,4 +27,8 @@ public class RetrievalVO {
 //     * 年龄
 //     */
 //    private Integer age;
+    /**
+     * 术语返回数量
+     */
+    private Integer size;
 }