疾病的高危处理

graph-web/src/main/java/org/diagbot/graphWeb/controller/GraphController.java

@@ -3,10 +3,12 @@ package org.diagbot.graphWeb.controller;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
+import org.diagbot.graph.javabean.GdbResponse;
 import org.diagbot.graph.jdbc.DriverManager;
 import org.diagbot.graph.jdbc.Neo4jAPI;
 import org.diagbot.common.work.FeatureRate;
 import org.diagbot.graphWeb.work.GraphCalculate;
+import org.diagbot.graphWeb.work.HighRiskCalculate;
 import org.diagbot.graphWeb.work.ParamsDataProxy;
 import org.diagbot.common.work.ResponseData;
 import org.diagbot.common.work.SearchData;
@@ -34,5 +36,12 @@ public class GraphController {
         response.setData(responseData);
         return response;
     }
+    @RequestMapping(value = "/highRisk", method = RequestMethod.POST)
+    @ResponseBody
+    public  Response<GdbResponse> getHighRisk(HttpServletRequest request, @RequestBody SearchData searchData) throws Exception {
+        HighRiskCalculate highRiskCalculate = new HighRiskCalculate();
+        Response<GdbResponse> gdbResponseResponse = highRiskCalculate.calculateHighRisk(request, searchData);
+        return gdbResponseResponse;
+    }
 }
 

graph-web/src/main/java/org/diagbot/graphWeb/work/HighRiskCalculate.java

@@ -0,0 +1,37 @@
+package org.diagbot.graphWeb.work;
+
+import com.alibaba.fastjson.JSONObject;
+import org.diagbot.common.work.Filnlly;
+import org.diagbot.common.work.ResponseData;
+import org.diagbot.common.work.SearchData;
+import org.diagbot.graph.javabean.GdbResponse;
+import org.diagbot.graph.jdbc.DriverManager;
+import org.diagbot.graph.jdbc.Neo4jAPI;
+import org.diagbot.pub.api.Response;
+import org.neo4j.driver.v1.Driver;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Map;
+
+public class HighRiskCalculate {
+   private Driver driver=null;
+   private Neo4jAPI n4api=null;
+    public Response<GdbResponse> calculateHighRisk(HttpServletRequest request, SearchData searchData) throws Exception {
+        GdbResponse responseData = new GdbResponse();
+        Response<GdbResponse> objectResponse = new Response<>();
+        Map<String,String> RSet = null;
+        if (driver == null) {
+            driver = DriverManager.newDrive();
+        }
+        if (driver != null && driver.session() != null) {
+            n4api = new Neo4jAPI(driver);
+            if(searchData.getDiag().trim()!=null){
+                String diag = searchData.getDiag();
+                RSet = n4api.getHighRisk(diag);
+            }
+           responseData.setResult(RSet);
+        }
+        objectResponse.setData(responseData);
+        return objectResponse;
+    }
+}

graph/src/main/java/org/diagbot/graph/javabean/GdbResponse.java

@@ -0,0 +1,28 @@
+package org.diagbot.graph.javabean;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 图形数据库数据返回结构
+ * Created by Mark Huang on 2018/10/26.
+ */
+public class GdbResponse {
+    private Map<String, String> Result = new HashMap<>();
+    private String status = "unknown";
+
+    public Map<String, String> getResult() {
+        return Result;
+    }
+
+    public void setResult(Map<String, String> Result ) {
+        this.Result = Result;
+    }
+
+    public String getStatus() {
+        return status;
+    }
+
+    public void setStatus(String status) { this.status = status; }
+
+}