Просмотр исходного кода

如果featureType包含10,返回的时候把入参的diag赋值treatDTO

kongwz 4 лет назад
Родитель
Сommit
ec99e072e2
1 измененных файлов с 9 добавлено и 8 удалено
  1. 9 8
      src/main/java/com/diagbot/facade/PushFacade.java

+ 9 - 8
src/main/java/com/diagbot/facade/PushFacade.java

@@ -53,7 +53,7 @@ public class PushFacade {
         Map<String, List<PushBaseDTO>> dis = new HashMap<>();
         //如果下的诊断有数据就反推
         if(pushVO.getDiagVo() != null && pushVO.getDiagVo().getDiags().size() > 0){
-            reversePushPackage(length, pushDTO, ruleTypeList, typeWords, pushVO,dis);
+            reversePushPackage(length, pushDTO, ruleTypeList, typeWords, pushVO,dis,wordCrfDTO);
         }else {
             //正推
             List<NeoPushDTO> push = neoFacade.getPush(pushVO);
@@ -68,7 +68,7 @@ public class PushFacade {
                 //把第一个推送出来的诊断set到diagOrder中,再反推
                 setPushVo(pushVO, push);
                 //调用反推
-                reversePushPackage(length, pushDTO, ruleTypeList, typeWords, pushVO,dis);
+                reversePushPackage(length, pushDTO, ruleTypeList, typeWords, pushVO,dis,wordCrfDTO);
             }
         }
 
@@ -86,7 +86,7 @@ public class PushFacade {
         pushVO.setDiagVo(diagVo);
     }
 
-    private void reversePushPackage(int length, PushDTO pushDTO, List<String> ruleTypeList, Map<String, List<String>> typeWords, NeoPushVO pushVO,Map<String, List<PushBaseDTO>> dis) {
+    private void reversePushPackage(int length, PushDTO pushDTO, List<String> ruleTypeList, Map<String, List<String>> typeWords, NeoPushVO pushVO,Map<String, List<PushBaseDTO>> dis,WordCrfDTO wordCrfDTO) {
         List<NeoPushDTO> reversePush = neoFacade.getReversePush(pushVO);
         if(ListUtil.isNotEmpty(reversePush)){
             // 症状
@@ -184,11 +184,11 @@ public class PushFacade {
                     pushDTO.setDis(dis);
                 }
             }
-            if (ruleTypeList.contains("10") && ListUtil.isNotEmpty(pushVO.getDiagVo().getDiags())) {
-                TreatDTO treatDTO = new TreatDTO();
-                List<TreatDTO> collect = pushVO.getDiagVo().getDiags().stream().map(x -> {
+            if (ruleTypeList.contains("10") && ListUtil.isNotEmpty(wordCrfDTO.getDiag())) {
+                List<TreatDTO> collect = wordCrfDTO.getDiag().stream().map(x -> {
+                    TreatDTO treatDTO = new TreatDTO();
                     treatDTO.setName(x.getName());
-                    treatDTO.setUniqueName(x.getName());
+                    treatDTO.setUniqueName(x.getUniqueName());
                     return treatDTO;
                 }).collect(Collectors.toList());
                 pushDTO.setGeneraTreat(collect);
@@ -198,8 +198,9 @@ public class PushFacade {
     }
 
     private List<PushBaseDTO> getPackagePushBaseDTO(Map<String, List<String>> typeWords, List<String> symptoms, StandConvertEnum symptom) {
-        PushBaseDTO pushBaseDTO = new PushBaseDTO();
+
         List<PushBaseDTO> symptomPush = symptoms.stream().map(x -> {
+            PushBaseDTO pushBaseDTO = new PushBaseDTO();
             pushBaseDTO.setName(x);
             return pushBaseDTO;
         }).collect(Collectors.toList());