Browse Source

高危的入参修改

kongwz 6 years ago
parent
commit
caefe2b4b7

+ 8 - 3
graphdb/src/main/java/org/diagbot/service/impl/KnowledgeServiceImpl.java

@@ -180,9 +180,14 @@ public class KnowledgeServiceImpl implements KnowledgeService {
 
     @Override
     public Map<String, Object> getHighRiskDisease(SearchData searchData) {
-        String diag = searchData.getDiag();
-        String[] diseaseArray = diag.split(",|,|、");
-        List<String> diseaseList = Arrays.asList(diseaseArray);
+        List<FeatureRate> diags = searchData.getPushDiags();
+        List<String> diseaseList = new ArrayList<>();
+        if(diags != null && diags.size()>0){
+            for (FeatureRate fe:diags) {
+              String featureName = fe.getFeatureName();
+              diseaseList.add(featureName);
+            }
+        }
         Map<String, Object> map = new HashMap<>();
         List<Map<String, Object>> list = baseNodeRepository.getHighRisk(diseaseList);
         for (Map hrmap:list) {