sku图稿库

sd-business/src/main/java/com/sd/business/service/sku/impl/SkuServiceImpl.java

@@ -18,6 +18,7 @@ import com.sd.business.entity.sku.po.SkuSpec;
 import com.sd.business.entity.sku.po.SkuSpecLink;
 import com.sd.business.entity.sku.vo.*;
 import com.sd.business.mapper.sku.SkuMapper;
+import com.sd.business.service.artwork.ArtworkLibraryService;
 import com.sd.business.service.bom.BomSpecService;
 import com.sd.business.service.department.DepartmentService;
 import com.sd.business.service.sku.SkuService;
@@ -57,6 +58,9 @@ public class SkuServiceImpl extends ServiceImpl<SkuMapper, Sku> implements SkuSe
     @Autowired
     private DepartmentService departmentService;
 
+    @Autowired
+    private ArtworkLibraryService artworkLibraryService;
+
     @Override
     public Page<SkuVo> getPage(SkuSelectDto dto) {
         IWrapper<Sku> wrapper = getWrapper();
@@ -115,6 +119,10 @@ public class SkuServiceImpl extends ServiceImpl<SkuMapper, Sku> implements SkuSe
         // sku规格
         List<SkuSpec> list = skuSpecService.list(q -> q.in(SkuSpec::getSkuId, id));
         List<SkuSpecVo> skuSpecList = BeanUtil.copyToList(list, SkuSpecVo.class);
+        artworkLibraryService.attributeAssign(skuSpecList, SkuSpec::getArtworkLibraryId, (item, artworkLibrary) -> {
+            item.setDesignImgUrl(artworkLibrary.getImgUrl());
+            item.setSharedFolder(artworkLibrary.getFileUrl());
+        });
         result.setSkuSpecList(skuSpecList);
 
         // sku规格关联
@@ -130,6 +138,7 @@ public class SkuServiceImpl extends ServiceImpl<SkuMapper, Sku> implements SkuSe
         bomSpecIdList.addAll(list.stream().map(SkuSpec::getBomSpecId).filter(ObjectUtil::isNotNull).collect(Collectors.toList()));
         bomSpecIdList.addAll(skuSpecLinkList.stream().map(SkuSpecLink::getBomSpecId).collect(Collectors.toList()));
         Map<Long, BomSpec> bomSpecMap = new HashMap<>();
+
         if (!bomSpecIdList.isEmpty()) {
             bomSpecMap = bomSpecService.mapKEntity(BaseIdPo::getId, q -> q.in(BaseIdPo::getId, bomSpecIdList));
         }