Selaa lähdekoodia

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

kongwz 4 vuotta sitten
vanhempi
commit
6219e28e4f
1 muutettua tiedostoa jossa 9 lisäystä ja 6 poistoa
  1. 9 6
      src/main/java/com/diagbot/facade/PushFacade.java

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

@@ -1,10 +1,7 @@
 package com.diagbot.facade;
 
 import com.diagbot.biz.push.entity.Item;
-import com.diagbot.dto.NeoPushDTO;
-import com.diagbot.dto.PushBaseDTO;
-import com.diagbot.dto.PushDTO;
-import com.diagbot.dto.WordCrfDTO;
+import com.diagbot.dto.*;
 import com.diagbot.enums.DiseaseTypeEnum;
 import com.diagbot.enums.MedicalAdviceEnum;
 import com.diagbot.enums.StandConvertEnum;
@@ -186,8 +183,14 @@ public class PushFacade {
                     dis.put(DiseaseTypeEnum.identify.getName(),pushDiffBaseDTOS);
                     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 -> {
+                    treatDTO.setName(x.getName());
+                    return treatDTO;
+                }).collect(Collectors.toList());
+                pushDTO.setGeneraTreat(collect);
             }
         }
 //            return pushDTO;