diff --git a/coolstore-partner-model/src/main/java/com/cool/store/vo/shop/RentInfoToDoVO.java b/coolstore-partner-model/src/main/java/com/cool/store/vo/shop/RentInfoToDoVO.java index 42a90de15..9ecaeb584 100644 --- a/coolstore-partner-model/src/main/java/com/cool/store/vo/shop/RentInfoToDoVO.java +++ b/coolstore-partner-model/src/main/java/com/cool/store/vo/shop/RentInfoToDoVO.java @@ -1,6 +1,7 @@ package com.cool.store.vo.shop; import com.cool.store.entity.PointInfoDO; +import com.cool.store.entity.ShopInfoDO; import com.google.common.collect.Lists; import io.swagger.annotations.ApiModelProperty; import lombok.Data; @@ -44,20 +45,23 @@ public class RentInfoToDoVO { @ApiModelProperty("提交时间") private Date submitTime; - public static List convert(List list, Map shopPointMap, Map pointMap, Map regionNameMap, Map rentContractSubmitTimeMap){ + public static List convert(List list, Map shopPointMap, Map pointMap, Map regionNameMap, Map rentContractSubmitTimeMap){ if(CollectionUtils.isEmpty(list)){ return Lists.newArrayList(); } List resultList = new ArrayList<>(); for (RentInfoToDoVO rent : list) { - Long pointId = shopPointMap.get(rent.getShopId()); - rent.setPointId(pointId); - PointInfoDO pointInfo = pointMap.get(pointId); + ShopInfoDO shopInfo = shopPointMap.get(rent.getShopId()); + if(Objects.isNull(shopInfo)){ + continue; + } + PointInfoDO pointInfo = pointMap.get(shopInfo.getPointId()); if(Objects.nonNull(pointInfo)){ rent.setPointName(pointInfo.getPointName()); rent.setAddress(pointInfo.getAddress()); - rent.setRegionNodeName(regionNameMap.get(pointInfo.getRegionId())); } + rent.setPointId(shopInfo.getPointId()); + rent.setRegionNodeName(regionNameMap.get(shopInfo.getRegionId())); rent.setSubmitTime(rentContractSubmitTimeMap.get(rent.getShopId())); resultList.add(rent); } diff --git a/coolstore-partner-service/src/main/java/com/cool/store/service/impl/PointServiceImpl.java b/coolstore-partner-service/src/main/java/com/cool/store/service/impl/PointServiceImpl.java index 0e39b4f7f..577f6d227 100644 --- a/coolstore-partner-service/src/main/java/com/cool/store/service/impl/PointServiceImpl.java +++ b/coolstore-partner-service/src/main/java/com/cool/store/service/impl/PointServiceImpl.java @@ -1035,7 +1035,7 @@ public class PointServiceImpl implements PointService { if(Objects.nonNull(todoPage) && CollectionUtils.isNotEmpty(todoPage.getResult())){ List shopIds = todoPage.getResult().stream().map(RentInfoToDoVO::getShopId).collect(Collectors.toList()); List shopList = shopInfoDAO.getShopListByIds(shopIds); - Map shopPointMap = shopList.stream().collect(Collectors.toMap(k->k.getId(), v->v.getPointId())); + Map shopPointMap = shopList.stream().collect(Collectors.toMap(k->k.getId(), Function.identity())); List regionIds = shopList.stream().map(ShopInfoDO::getRegionId).distinct().collect(Collectors.toList()); List pointIds = shopList.stream().map(ShopInfoDO::getPointId).collect(Collectors.toList()); List pointList = pointInfoDAO.getPointListByIds(pointIds);