Browse Source

通过HTTP POST调用知识图谱的API接口

MarkHuang 6 years ago
parent
commit
dfeb6be74c

+ 2 - 2
graph-web/src/main/resources/url.properties

@@ -1,2 +1,2 @@
-#highrisk.http.url=http://192.168.2.234:8080/knowledge/getHighRisk
-highrisk.http.url=http://192.168.3.9:8080/knowledge/getHighRisk
+highrisk.http.url=http://192.168.2.234:8080/knowledge/getHighRisk
+#highrisk.http.url=http://192.168.3.9:8080/knowledge/getHighRisk

+ 1 - 1
graphdb/src/main/java/org/diagbot/service/KnowledgeService.java

@@ -39,7 +39,7 @@ public interface KnowledgeService {
     List<Map<String, Object>> findHighRiskDisease(QueryVo queryVo);
 
     // 获取高危疾病列表
-    Map<String, String> getHighRiskDisease(SearchData searchData);
+    Map<String, Object> getHighRiskDisease(SearchData searchData);
 
     // 获取推荐项目
     List<Map<String, Object>> getRecommendItem(QueryVo queryVo);

+ 11 - 2
graphdb/src/main/java/org/diagbot/service/impl/KnowledgeServiceImpl.java

@@ -180,11 +180,20 @@ public class KnowledgeServiceImpl implements KnowledgeService {
     }
 
     @Override
-    public List<Map<String, Object>> getHighRiskDisease(SearchData searchData) {
+    public Map<String, Object> getHighRiskDisease(SearchData searchData) {
         String diag = searchData.getDiag();
         String[] diseaseArray = diag.split(",|,|、");
         List<String> diseaseList = Arrays.asList(diseaseArray);
-        List<Map<String, Object>> map = baseNodeRepository.getHighRisk(diseaseList);
+
+        Map<String, Object> map = new HashMap<>();
+        List<Map<String, Object>> list = baseNodeRepository.getHighRisk(diseaseList);
+
+        for (Map hrmap:list) {
+            String name = hrmap.get("name").toString();
+            String risk = hrmap.get("risk").toString();
+            map.put(name, risk);
+        }
+
         return map;
     }
 

+ 1 - 1
graphdb/src/main/java/org/diagbot/vo/GdbResponse.java

@@ -6,6 +6,6 @@ import java.util.Map;
 @Getter
 @Setter
 public class GdbResponse {
-    private Map<String, String> Result = new HashMap<>();
+    private Map<String, Object> Result = new HashMap<>();
     private String status = "unknown";
 }