合同列表产品信息添加图片地址

hx-sale/src/main/java/com/fjhx/sale/entity/contract/vo/ContractProductVo.java

@@ -79,6 +79,11 @@ public class ContractProductVo extends ContractProduct {
     private String productType;
 
     /**
+     * 产品图片地址
+     */
+    private String productImgUrl;
+
+    /**
      * 所属分类id
      */
     private Long productCategoryId;

hx-sale/src/main/java/com/fjhx/sale/service/contract/impl/ContractServiceImpl.java

@@ -431,6 +431,18 @@ public class ContractServiceImpl extends ServiceImpl<ContractMapper, Contract>
             item.setProductWidth(productInfo.getWidth());
             item.setProductHeight(productInfo.getHeight());
         });
+        //赋值产品图片信息
+        List<Long> productIds = contractProductVos.stream().map(ContractProduct::getProductId).distinct().collect(Collectors.toList());
+        Map<Long, List<FileInfoVo>> productFileMap = ObsFileUtil.getFileMap(productIds, 0);
+        for (ContractProductVo contractProductVo : contractProductVos) {
+            List<FileInfoVo> fileInfoVoList = productFileMap.get(contractProductVo.getProductId());
+            if (ObjectUtil.isNotEmpty(fileInfoVoList)) {
+                FileInfoVo fileInfoVo = fileInfoVoList.get(0);
+                contractProductVo.setProductImgUrl(fileInfoVo.getFileUrl());
+            }
+        }
+
+
         Map<Long, List<ContractProductVo>> cpMap = contractProductVos.stream().collect(Collectors.groupingBy(ContractProduct::getContractId));