kongwz 5 роки тому
батько
коміт
23778a5617

+ 13 - 5
graph/src/main/java/org/diagbot/graph/jdbc/Neo4jAPI.java

@@ -878,11 +878,12 @@ public class Neo4jAPI {
                     }
                 }
             }
-            //判断急诊
-            Set<String> disSet = diseaseCondition.keySet();
             List<String> newDis = new ArrayList<>();
-            if(disSet.size()>0){
-                for (String dis:disSet) {
+            //判断急诊和警惕
+            for (Map.Entry<String, Map<String, String>> fs:diseaseCondition.entrySet()) {
+                String dis = fs.getKey();
+                Set<String> leiSet = fs.getValue().keySet();
+                if(leiSet.contains("确诊")){
                     newDis.add("\""+dis+"\"");
                 }
             }
@@ -892,8 +893,15 @@ public class Neo4jAPI {
             while (result.hasNext()) {
                 Record record = result.next();
                 String emDis = record.get("emDis").toString();
+                int em = record.get("em").asInt();//急诊
+                int hi = record.get("hi").asInt();//警惕
                 Map<String, String> stringStringMap = diseaseCondition.get(emDis.replace("\"", ""));
-                stringStringMap.put("急诊", "");
+                if(em == 1){
+                    stringStringMap.put("急诊", "");
+                }
+                if(hi == 1){
+                    stringStringMap.put("警惕", "");
+                }
                 diseaseCondition.put(emDis.replace("\"", ""), stringStringMap);
             }
         } catch (Exception e) {

+ 1 - 1
graph/src/main/resources/bolt.properties

@@ -88,7 +88,7 @@ searchBigSubDrugs=match (n:Disease{name:diseaseName})-[r0:\u63A8\u8350]->(m:Drug
 searchDifferentialDiagnose=match(d:Disease)-[r:\u9274\u522B\u8BCA\u65AD]->(h) where d.name='mainDis' return collect(h.name) as coll
 
 #\u5224\u65AD\u662F\u5426\u4E3A\u6025\u8BCA\u7684\u8BED\u53E5
-searchEmergency=match(d:Disease) where d.name in disList and d.emergency=1 return d.name as emDis
+searchEmergency=match(d:Disease) where d.name in disList return d.name as emDis,d.emergency as em,d.high_risk as hi
 
 #\u67E5\u627E\u6307\u6807\u7684\u8BED\u53E5
 searchIndication=match (n:Condition)-[r:\u786E\u8BCA|:\u62DF\u8BCA]->(m:Indicators)\n \