|
|
@@ -5,6 +5,7 @@ import com.yc.ship.framework.common.pojo.CommonResult;
|
|
|
import com.yc.ship.framework.common.pojo.PageParam;
|
|
|
import com.yc.ship.framework.common.pojo.PageResult;
|
|
|
import com.yc.ship.framework.common.util.collection.CollectionUtils;
|
|
|
+import com.yc.ship.framework.common.util.collection.MapUtils;
|
|
|
import com.yc.ship.framework.common.util.object.BeanUtils;
|
|
|
import com.yc.ship.framework.excel.core.util.ExcelUtils;
|
|
|
import com.yc.ship.module.product.controller.admin.voyage.vo.*;
|
|
|
@@ -107,8 +108,8 @@ public class VoyageController {
|
|
|
List<ResourceRouteDO> routeList = routeService.getList(routeIds);
|
|
|
Map<Long, ResourceRouteDO> routeDOMap = CollectionUtils.convertMap(routeList, ResourceRouteDO::getId);
|
|
|
list.stream().forEach(item -> {
|
|
|
- item.setShipName(shipDOMap.get(item.getShipId()).getName());
|
|
|
- item.setRouteName(routeDOMap.get(item.getRouteId()).getName());
|
|
|
+ MapUtils.findAndThen(shipDOMap, item.getShipId(), shipDO -> item.setShipName(shipDO.getName()));
|
|
|
+ MapUtils.findAndThen(routeDOMap, item.getRouteId(), routeDO -> item.setRouteName(routeDO.getName()));
|
|
|
});
|
|
|
return success(page);
|
|
|
}
|
|
|
@@ -132,9 +133,11 @@ public class VoyageController {
|
|
|
List<Long> routeIds = CollectionUtils.convertList(voyageRespVOS, VoyageRespVO::getRouteId);
|
|
|
Map<Long, RouteRespDTO> longRouteRespDTOMap = routeApi.queryRoute(routeIds);
|
|
|
voyageRespVOS.stream().forEach(item -> {
|
|
|
- item.setRouteName(longRouteRespDTOMap.get(item.getRouteId()).getName());
|
|
|
- item.setRouteResp(longRouteRespDTOMap.get(item.getRouteId()));
|
|
|
- item.setShipName(shipDOMap.get(item.getShipId()).getName());
|
|
|
+ MapUtils.findAndThen(shipDOMap, item.getShipId(), shipDO -> item.setShipName(shipDO.getName()));
|
|
|
+ MapUtils.findAndThen(longRouteRespDTOMap, item.getRouteId(), routeRespDTO -> {
|
|
|
+ item.setRouteResp(routeRespDTO);
|
|
|
+ item.setRouteName(routeRespDTO.getName());
|
|
|
+ });
|
|
|
});
|
|
|
return success(voyageRespVOS);
|
|
|
}
|