소스 검색

融合场景列表修改

lyhzzz 3 년 전
부모
커밋
0c65a85890

+ 14 - 0
src/main/java/com/fdkankan/fusion/response/FusionNumVo.java

@@ -0,0 +1,14 @@
+package com.fdkankan.fusion.response;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.fdkankan.fusion.entity.FusionNum;
+import lombok.Data;
+
+@Data
+public class FusionNumVo extends FusionNum {
+
+    private SceneVo sceneData;
+
+}

+ 2 - 2
src/main/java/com/fdkankan/fusion/service/IFusionNumService.java

@@ -3,7 +3,7 @@ package com.fdkankan.fusion.service;
 import com.fdkankan.fusion.entity.FusionNum;
 import com.baomidou.mybatisplus.extension.service.IService;
 import com.fdkankan.fusion.request.FusionParam;
-import com.fdkankan.fusion.response.SceneVo;
+import com.fdkankan.fusion.response.FusionNumVo;
 
 import java.util.List;
 
@@ -21,7 +21,7 @@ public interface IFusionNumService extends IService<FusionNum> {
 
     List<FusionNum> getListByFusionIdList(List<Integer> fusionIdList);
 
-    List<SceneVo> getListByCaseId(Integer caseId, String token);
+    List<FusionNumVo> getListByCaseId(Integer caseId, String token);
 
     void add(FusionParam param);
 

+ 12 - 3
src/main/java/com/fdkankan/fusion/service/impl/FusionNumServiceImpl.java

@@ -9,6 +9,7 @@ import com.fdkankan.fusion.exception.BusinessException;
 import com.fdkankan.fusion.mapper.IFusionNumMapper;
 import com.fdkankan.fusion.request.CaseParam;
 import com.fdkankan.fusion.request.FusionParam;
+import com.fdkankan.fusion.response.FusionNumVo;
 import com.fdkankan.fusion.response.SceneVo;
 import com.fdkankan.fusion.service.ICaseFusionService;
 import com.fdkankan.fusion.service.ICaseService;
@@ -56,7 +57,7 @@ public class FusionNumServiceImpl extends ServiceImpl<IFusionNumMapper, FusionNu
     }
 
     @Override
-    public List<SceneVo> getListByCaseId(Integer caseId, String token) {
+    public List<FusionNumVo> getListByCaseId(Integer caseId, String token) {
         List<CaseFusion> fusionList = caseFusionService.getListByCaseId(caseId);
         if(fusionList.size()<=0){
             return new ArrayList<>();
@@ -68,9 +69,17 @@ public class FusionNumServiceImpl extends ServiceImpl<IFusionNumMapper, FusionNu
         param.setTypeMap(typeMap);
         param.setCaseId(caseId);
         List<SceneVo> sceneVos = caseService.sceneList(param, token);
-        for (SceneVo sceneVo : sceneVos) {
+        HashMap<Integer,SceneVo> sceneMap = new HashMap<>();
+        sceneVos.forEach(entity -> sceneMap.put(entity.getModelId(),entity));
+
+        List<FusionNumVo> numVoList = new ArrayList<>();
+        for (FusionNum fusionNum : listByFusionIdList) {
+            FusionNumVo fusionNumVo = new FusionNumVo();
+            BeanUtils.copyProperties(fusionNum,fusionNumVo);
+            fusionNumVo.setSceneData(sceneMap.get(fusionNum.getModelId()));
+            numVoList.add(fusionNumVo);
         }
-        return caseService.sceneList(param,token);
+        return numVoList;
     }
 
     private HashMap<Integer, List<String>> getTypeMap( List<FusionNum> list) {