diff --git a/core/domain/src/main/java/com/wansenai/vo/product/ProductStockKeepUnitVO.java b/core/domain/src/main/java/com/wansenai/vo/product/ProductStockKeepUnitVO.java index a372665..579c938 100644 --- a/core/domain/src/main/java/com/wansenai/vo/product/ProductStockKeepUnitVO.java +++ b/core/domain/src/main/java/com/wansenai/vo/product/ProductStockKeepUnitVO.java @@ -56,6 +56,8 @@ public class ProductStockKeepUnitVO { private String extendInfo; + private String imageUrl; + @JsonSerialize(using = BigDecimalSerializerBO.class) private BigDecimal retailPrice; diff --git a/core/service/src/main/java/com/wansenai/service/product/impl/ProductStockServiceImpl.java b/core/service/src/main/java/com/wansenai/service/product/impl/ProductStockServiceImpl.java index 98ae616..0457ab3 100644 --- a/core/service/src/main/java/com/wansenai/service/product/impl/ProductStockServiceImpl.java +++ b/core/service/src/main/java/com/wansenai/service/product/impl/ProductStockServiceImpl.java @@ -16,8 +16,10 @@ import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.wansenai.dto.product.QueryProductStockKeepUnitDTO; +import com.wansenai.entities.product.ProductImage; import com.wansenai.entities.product.ProductStock; import com.wansenai.mappers.product.ProductStockMapper; +import com.wansenai.service.product.ProductImageService; import com.wansenai.service.product.ProductStockService; import com.wansenai.service.user.ISysUserService; import com.wansenai.service.warehouse.WarehouseService; @@ -28,6 +30,7 @@ import com.wansenai.vo.product.ProductStockVO; import com.wansenai.vo.report.ProductStockSkuVO; import org.springframework.beans.BeanUtils; import org.springframework.stereotype.Service; +import org.springframework.util.CollectionUtils; import java.util.ArrayList; import java.util.List; @@ -43,17 +46,32 @@ public class ProductStockServiceImpl extends ServiceImpl getProductExtendPriceInfo(QueryProductStockKeepUnitDTO priceDTO) { var page = new Page(priceDTO.getPage(), priceDTO.getPageSize()); - return productStockMapper.getProductSkuList(page, priceDTO); + IPage iPage = productStockMapper.getProductSkuList(page, priceDTO); + iPage.getRecords().forEach(model->{ + if(model.getProductId() != null){ + List imgList = productImageService.lambdaQuery().select(ProductImage::getImageUrl) + .eq(ProductImage::getProductId, model.getProductId()) + .list(); + if(!CollectionUtils.isEmpty(imgList)){ + model.setImageUrl(imgList.get(0).getImageUrl()); + } + } + }); + return iPage; } @Override