修改治疗featureType为6

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

@@ -8,6 +8,8 @@ import org.diagbot.nlp.relation.module.Lis;
 import org.diagbot.pub.api.Response;
 import org.diagbot.pub.utils.PropertiesUtil;
 import org.diagbot.pub.utils.http.HttpApi;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import javax.servlet.http.HttpServletRequest;
 import java.util.*;
@@ -16,8 +18,12 @@ import java.util.*;
  * 推送诊断和治疗方案
  */
 public class DiseaseCalculate {
+    Logger logger = LoggerFactory.getLogger(DiseaseCalculate.class);
     public ResponseData getDisease(SearchData searchData) throws Exception {
         ResponseData responseData = new ResponseData();
+        String[] featureSplits = searchData.getFeatureType().split(",");
+        List<String> featureList = Arrays.asList(featureSplits);
+        String webDiag = searchData.getDiag();
         PropertiesUtil propertiesUtil = new PropertiesUtil("url.properties");
         //调用图谱API
         HttpApi<ResponseData> api = new HttpApi<>();
@@ -26,8 +32,11 @@ public class DiseaseCalculate {
         //走诊断推送
         List<FeatureRate> dis = (List<FeatureRate>)diseaseResponse.getDis();
         List<FeatureRate> diseaseName = getDisease(dis);
-        Map<String, Filnlly> treat = getTreat(diseaseResponse.getTreat());
-        responseData.setTreat(treat);
+        //走治疗推送
+        if(featureList.contains("8") && webDiag.trim()!=null && webDiag.trim()!="" ){
+            Map<String, Filnlly> treat = getTreat(diseaseResponse.getTreat());
+            responseData.setTreat(treat);
+        }
         responseData.setDis(diseaseName);
         return responseData;
     }

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

@@ -60,7 +60,7 @@ public class GraphCalculate {
         logger.info("diseaseSet :"+diseaseSet);
         Integer diseaseType = searchData.getDisType();
         //走治疗
-        if(webDiag.trim()!=null && webDiag.trim()!="" && featureTypeList.contains("8")){
+        if(webDiag.trim()!=null && webDiag.trim()!="" && featureTypeList.contains("6")){
             //走平常诊断治疗
             Map<String, Filnlly> mulDiseaseTreat = neo4jAPI.getMulDiseaseTreat_2(webDiag,diseaseType,diseaseSet,String.join(",", inputList));
             responseData.setTreat(mulDiseaseTreat);