Browse Source

修改按照指定长度返回疾病、症状信息

louhr 5 years ago
parent
commit
710e86efa4

+ 5 - 5
bigdata-web/src/main/java/org/diagbot/bigdata/work/ResultDataProxy.java

@@ -68,11 +68,11 @@ public class ResultDataProxy {
                 }
             }
             featureList.add(featureRate);
-                if (cursor < searchData.getLength()) {
-                    cursor++;
-                } else {
-                    break;
-                }
+//                if (cursor < searchData.getLength()) {
+//                    cursor++;
+//                } else {
+//                    break;
+//                }
         }
 
         return featureList;

+ 6 - 0
push-web/src/main/java/org/diagbot/push/controller/AlgorithmController.java

@@ -228,6 +228,12 @@ public class AlgorithmController extends BaseController {
         List<FeatureRate> upfes = classifyDiag.filterDiag(graphResponseData, pushDiags);
         List<FeatureRate> classify = classifyDiag.diagClassify(upfes);
         List<FeatureRate> featureRates = classifyDiag.sortDiag(classify);
+        if (featureRates.size() > searchData.getLength()) {
+            featureRates = featureRates.subList(0, searchData.getLength());
+        }
+        if (bigDataResponseData.getSymptom().size() > searchData.getLength()) {
+            bigDataResponseData.setSymptom(bigDataResponseData.getSymptom().subList(0, searchData.getLength()));
+        }
         bigDataResponseData.setDis(featureRates);
         //返回结果化验、辅检、体征、进行性别年龄过滤
         ResultDataProxy resultDataProxy = new ResultDataProxy();