fix: 二级分类递归查一级分类

boyang
1 parent a11bc16b
Showing 1 changed file with 66 additions and 23 deletions
shop/src/main/java/com/canrd/shop/service/impl/ProductServiceImpl.java
@@ -137,36 +137,49 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
                 .eq(Productcategoryrelation::getProductId, productQueryVO.getId())
                 .list();
         List<String> categoryIds = productcategoryrelations.stream().map(Productcategoryrelation::getCategoryId).collect(Collectors.toList());
+
         if (CollectionUtils.isNotEmpty(categoryIds)) {
+            // 查询第一级分类
             List<ProductCategoryDO> productCategories = categoryService.lambdaQuery()
                     .in(ProductCategoryDO::getId, categoryIds)
                     .eq(ProductCategoryDO::getParentId, "33ba6943c8bf4ca082614f299865341c")
                     .list();
-            ProductCategoryDO firstCategory = productCategories.get(0);
-            List<ProductCategoryDO> productCategories2 = categoryService.lambdaQuery()
-                    .in(ProductCategoryDO::getId, categoryIds)
-                    .eq(ProductCategoryDO::getParentId, firstCategory.getId())
-                    .list();
-//        ProductCategoryDO secondCategory = productCategories2.get(0);
-            ProductCategoryDO secondCategory = productCategories2.stream()
-                    .filter(category -> !"Not specified".equals(category.getName()))
-                    .findFirst()
-                    .orElse(productCategories2.get(0)); // 如果找不到，使用列表第一个元素
-            // 从 functionService 获取 functionId
-            List<ProductFunctionDO> functions = functionService.getFirstFunction(productVO.getId());
-            ProductFunctionDO selectedFunction = null;
-            if (functions != null && functions.size() > 0) {
-                selectedFunction = functions.stream()
-                        .filter(function -> !"Not specified".equals(function.getName()))
+            ProductCategoryDO firstCategory = null;
+            if (productCategories.isEmpty()) {
+                firstCategory = findParentCategory(categoryIds, "33ba6943c8bf4ca082614f299865341c");
+            } else {
+                firstCategory = productCategories.get(0);
+            }
+
+            if (firstCategory != null) {
+                // 查询第二级分类
+                List<ProductCategoryDO> productCategories2 = categoryService.lambdaQuery()
+                        .in(ProductCategoryDO::getId, categoryIds)
+                        .eq(ProductCategoryDO::getParentId, firstCategory.getId())
+                        .list();
+                ProductCategoryDO secondCategory = productCategories2.stream()
+                        .filter(category -> !"Not specified".equals(category.getName()))
                         .findFirst()
-                        .orElse(functions.get(0)); // 如果找不到，使用列表第一个元素
+                        .orElse(productCategories2.isEmpty() ? null : productCategories2.get(0)); // 如果找不到，使用列表第一个元素或 null
+
+                // 获取功能信息
+                List<ProductFunctionDO> functions = functionService.getFirstFunction(productVO.getId());
+                ProductFunctionDO selectedFunction = null;
+                if (CollectionUtils.isNotEmpty(functions)){
+                    selectedFunction = functions.stream()
+                            .filter(function -> !"Not specified".equals(function.getName()))
+                            .findFirst()
+                            .orElse(functions.get(0)); // 如果找不到，使用列表第一个元素
+                }
+
+                // 构建面包屑信息
+                ProductCrumbsVO productCrumbsVO = ProductCrumbsVO.builder()
+                        .category1(firstCategory.getName())
+                        .category2(secondCategory != null ? secondCategory.getName() : null)
+                        .function(selectedFunction != null ? selectedFunction.getName() : null)
+                        .build();
+                productVO.setProductCrumbsVO(productCrumbsVO);
             }
-            ProductCrumbsVO productCrumbsVO = ProductCrumbsVO.builder()
-                    .category1(firstCategory.getName())   //
-                    .category2(secondCategory.getName())
-                    .function(selectedFunction.getName())
-                    .build();
-            productVO.setProductCrumbsVO(productCrumbsVO);
         }
         List<EbCategorysRelation> cateRelations = ebCategorysRelationService.lambdaQuery()
@@ -198,6 +211,36 @@ public class ProductServiceImpl extends ServiceImpl&lt;ProductMapper, ProductDO&gt; im
         return ServerResult.success(productVO);
     }
+    //查找一级分类
+    private ProductCategoryDO findParentCategory(List<String> categoryIds, String targetParentId) {
+        for (String categoryId : categoryIds) {
+            // 查询当前 categoryId 的数据
+            ProductCategoryDO currentCategory = categoryService.lambdaQuery()
+                    .eq(ProductCategoryDO::getId, categoryId)
+                    .one();
+            if (currentCategory != null) {
+                // 如果当前分类的 parentId 等于目标 parentId，返回当前分类
+                if (targetParentId.equals(currentCategory.getParentId())) {
+                    return currentCategory;
+                }
+                // 否则递归查找 parentId 的分类
+                ProductCategoryDO parentCategory = categoryService.lambdaQuery()
+                        .eq(ProductCategoryDO::getId, currentCategory.getParentId())
+                        .one();
+                if (parentCategory != null) {
+                    ProductCategoryDO foundCategory = findParentCategory(
+                            Collections.singletonList(parentCategory.getId()),
+                            targetParentId
+                    );
+                    if (foundCategory != null) {
+                        return foundCategory;
+                    }
+                }
+            }
+        }
+        return null; // 如果未找到符合条件的分类，返回 null
+    }
+
     /**
      * 分页查询
      *