refactor(product): 重构产品列表查询逻辑

- 提取公共方法 suppleProducts 补充产品信息- 优化产品价格显示逻辑，移除重复代码 - 统一处理产品图片列表解析 - 删除冗余变量，简化代码结构

refactor(product): 重构产品列表查询逻辑
- 提取公共方法 suppleProducts 补充产品信息- 优化产品价格显示逻辑，移除重复代码 - 统一处理产品图片列表解析 - 删除冗余变量，简化代码结构
曾国涛
1 parent 926db6dd
Showing 1 changed file with 30 additions and 21 deletions
shop/src/main/java/com/canrd/shop/service/impl/ProductServiceImpl.java
@@ -160,10 +160,39 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
         }
         Page page = new Page<>(productQueryVO.getPageNo(), productQueryVO.getPageSize());
         IPage<ProductDO> iPage = productMapper.selectAll(page,queryWapper);
+        suppleProducts(iPage.getRecords());
         productQueryVO.setTotal(Long.valueOf(iPage.getTotal()).intValue());
         return ServerResult.success().setData(PageUtils.getPageReturn(iPage.getRecords(), productQueryVO));
     }
+    private void suppleProducts(List<ProductDO> records) {
+        Boolean productPriceShow = switchControlService.getEnabledByName(SwitchControlConstants.PRODUCT_PRICE_SHOW);
+        List<String> productIds = records.stream().map(ProductDO::getId).collect(Collectors.toList());
+        List<TicketTypeDO> tickeyTypeDOList = ticketTypeService.lambdaQuery().in(TicketTypeDO::getProductId, productIds).list();
+        Map<String,List<TicketTypeDO>> pId2ttDOsMap = tickeyTypeDOList.stream().collect(Collectors.groupingBy(TicketTypeDO::getProductId));
+        Map<String,BigDecimal> pId2ttMinPriceMap = new HashMap<>();
+        for (TicketTypeDO ticketTypeDO : tickeyTypeDOList) {
+            if (Objects.isNull(pId2ttMinPriceMap.get(ticketTypeDO.getProductId()))){
+                pId2ttMinPriceMap.put(ticketTypeDO.getProductId(), ticketTypeDO.getPrice());
+            }else {
+                if (pId2ttMinPriceMap.get(ticketTypeDO.getProductId()).compareTo(ticketTypeDO.getPrice())>0){
+                    pId2ttMinPriceMap.put(ticketTypeDO.getProductId(), ticketTypeDO.getPrice());
+                }
+            }
+        }
+        records.forEach(product -> {
+            if (productPriceShow){
+                if (Objects.nonNull(pId2ttMinPriceMap.get(product.getId()))){
+                    product.setPrice(pId2ttMinPriceMap.get(product.getId()));
+                }
+            }
+            //将productDO的productimageliststore解析为map
+            List<Map> maps = JSON.parseArray(product.getProductimageliststore(), Map.class);
+            Map map = maps.get(0);
+            product.setImageFileKey(map.get("fileKey").toString());
+        });
+    }
+
     @Override
     public ServerResult listBySimilar(ProductQueryVO productQueryVO){
         if(StringUtils.isBlank(productQueryVO.getKeyword())){
@@ -181,24 +210,15 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
         }
         Set<String> productIds = null;
         Map<String,List<TicketTypeDO>> pId2ttDOsMap = null;
-        Map<String,BigDecimal> pId2ttMinPriceMap = null;
         boolean needLikeMatch = false;
         if(StringUtils.isNotBlank(productQueryVO.getKeyword())){
             List<TicketTypeDO> tickeyTypeDOList = ticketTypeService.lambdaQuery().like(TicketTypeDO::getRank, productQueryVO.getKeyword()).list();
             productIds = tickeyTypeDOList.stream().map(TicketTypeDO::getProductId).collect(Collectors.toSet());
             needLikeMatch = true;
             pId2ttDOsMap = new HashMap<>();
-            pId2ttMinPriceMap = new HashMap<>();
             for (TicketTypeDO ticketTypeDO : tickeyTypeDOList) {
                 pId2ttDOsMap.computeIfAbsent(ticketTypeDO.getProductId(), k -> new ArrayList<>());
                 pId2ttDOsMap.get(ticketTypeDO.getProductId()).add(ticketTypeDO);
-                if (Objects.isNull(pId2ttMinPriceMap.get(ticketTypeDO.getProductId()))){
-                    pId2ttMinPriceMap.put(ticketTypeDO.getProductId(), ticketTypeDO.getPrice());
-                }else {
-                    if (pId2ttMinPriceMap.get(ticketTypeDO.getProductId()).compareTo(ticketTypeDO.getPrice())>0){
-                        pId2ttMinPriceMap.put(ticketTypeDO.getProductId(), ticketTypeDO.getPrice());
-                    }
-                }
             }
             Set<String> finalProductIds = productIds;
             queryWapper.and(subQueryWapper -> {
@@ -216,13 +236,7 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
             }
         });
         List<ProductDO> similarProductList = Lists.newArrayList();
-        Boolean productPriceShow = switchControlService.getEnabledByName(SwitchControlConstants.PRODUCT_PRICE_SHOW);
         for(ProductDO product:productDOS){
-            if (productPriceShow){
-                if (Objects.nonNull(pId2ttMinPriceMap.get(product.getId()))){
-                    product.setPrice(pId2ttMinPriceMap.get(product.getId()));
-                }
-            }
             if(accurateProductIdSet.contains(product.getId())){
                 continue;
             }
@@ -251,12 +265,7 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
         List<ProductDO> allProductList = Lists.newArrayList();
         allProductList.addAll(accurateProducts);
         allProductList.addAll(similarProductList);
-        allProductList.forEach(productDO -> {
-            //将productDO的productimageliststore解析为map
-            List<Map> maps = JSON.parseArray(productDO.getProductimageliststore(), Map.class);
-            Map map = maps.get(0);
-            productDO.setImageFileKey(map.get("fileKey").toString());
-        });
+        suppleProducts(allProductList);
         return pager(productQueryVO,allProductList);
     }