高危修改

graphdb/src/main/java/org/diagbot/service/impl/KnowledgeServiceImpl.java

@@ -181,16 +181,18 @@ public class KnowledgeServiceImpl implements KnowledgeService {
 
     @Override
     public Map<String, Object> getHighRiskDisease(SearchData searchData) {
-        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);
-            }
-        }
+//        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);
+//            }
+//        }
+        String[] splitsDiag = searchData.getDiag().split(",");
+        List<String> diseaseNameList = Arrays.asList(splitsDiag);
         Map<String, Object> map = new HashMap<>();
-        List<Map<String, Object>> list = baseNodeRepository.getHighRisk(diseaseList);
+        List<Map<String, Object>> list = baseNodeRepository.getHighRisk(diseaseNameList);
         for (Map hrmap:list) {
             String name = hrmap.get("name").toString();
             String risk = hrmap.get("risk").toString();