客户详情主营类别和主营商品回显返回值格式化封装bug修复

yudao-module-customer/yudao-module-customer-rest/src/main/java/cn/iocoder/yudao/module/customer/controller/admin/customer/CustomerController.java

@@ -438,29 +438,23 @@ public class CustomerController {
                 respVO.setPromoterName(customer1.getName());
             }
         }
-        if (CollectionUtil.isNotEmpty(respVO.getProductList())) {
-            // 对商品数据进行分组返回
-            Map<Long, List<ProductRespVO>> productRespVOMap = respVO.getProductList().stream().collect(Collectors.groupingBy(ProductRespVO::getTypeId));
-            List<CustomerProductTypeGroupVO> products = productRespVOMap.entrySet().stream().map(entry -> {
-                CustomerProductTypeGroupVO productTypeGroupVO = new CustomerProductTypeGroupVO();
-                productTypeGroupVO.setTypeId(entry.getKey());
-                productTypeGroupVO.setProductIds(entry.getValue().stream().map(ProductRespVO::getId).collect(Collectors.toList()));
-                return productTypeGroupVO;
-            }).collect(Collectors.toList());
+        // 主营类别是必须的
         if (CollectionUtil.isNotEmpty(respVO.getProductTypeList())){
             // 允许选择了主营类别，但是该主营类别下面没有对应的商品
+            List<CustomerProductTypeGroupVO> products = new ArrayList<>();
             for (ProductTypeRespVO typeRespVO : respVO.getProductTypeList()) {
-                    if (!productRespVOMap.containsKey(typeRespVO.getId())) {
                 CustomerProductTypeGroupVO productTypeGroupVO = new CustomerProductTypeGroupVO();
                 productTypeGroupVO.setTypeId(typeRespVO.getId());
+                if (CollectionUtil.isNotEmpty(respVO.getProductList())) {
+                    List<Long> prodIds = respVO.getProductList().stream().filter(p -> p.getTypeId() == typeRespVO.getId()).map(ProductRespVO::getId).collect(Collectors.toList());
+                    productTypeGroupVO.setProductIds(CollectionUtil.isEmpty(prodIds) ? new ArrayList<>() : prodIds);
+                } else {
                     productTypeGroupVO.setProductIds(new ArrayList<>());
-                        products.add(productTypeGroupVO);
-                    }
                 }
+                products.add(productTypeGroupVO);
             }
             respVO.setProducts(products);
         }
-
         //设置银行账号
         CustomerBankQueryVO customerBankQueryVO = new CustomerBankQueryVO();
         customerBankQueryVO.setCustomerId(id);
@@ -501,25 +495,20 @@ public class CustomerController {
         if (Objects.isNull(respVO)) {
             throw exception(CUSTOMER_NOT_EXISTS);
         }
-        if (CollectionUtil.isNotEmpty(respVO.getProductList())) {
-            // 对商品数据进行分组返回
-            Map<Long, List<ProductRespVO>> productRespVOMap = respVO.getProductList().stream().collect(Collectors.groupingBy(ProductRespVO::getTypeId));
-            List<CustomerProductTypeGroupVO> products = productRespVOMap.entrySet().stream().map(entry -> {
-                CustomerProductTypeGroupVO productTypeGroupVO = new CustomerProductTypeGroupVO();
-                productTypeGroupVO.setTypeId(entry.getKey());
-                productTypeGroupVO.setProductIds(entry.getValue().stream().map(ProductRespVO::getId).collect(Collectors.toList()));
-                return productTypeGroupVO;
-            }).collect(Collectors.toList());
+        // 主营类别是必须的
         if (CollectionUtil.isNotEmpty(respVO.getProductTypeList())){
             // 允许选择了主营类别，但是该主营类别下面没有对应的商品
+            List<CustomerProductTypeGroupVO> products = new ArrayList<>();
             for (ProductTypeRespVO typeRespVO : respVO.getProductTypeList()) {
-                    if (!productRespVOMap.containsKey(typeRespVO.getId())) {
                 CustomerProductTypeGroupVO productTypeGroupVO = new CustomerProductTypeGroupVO();
                 productTypeGroupVO.setTypeId(typeRespVO.getId());
+                if (CollectionUtil.isNotEmpty(respVO.getProductList())) {
+                    List<Long> prodIds = respVO.getProductList().stream().filter(p -> p.getTypeId() == typeRespVO.getId()).map(ProductRespVO::getId).collect(Collectors.toList());
+                    productTypeGroupVO.setProductIds(CollectionUtil.isEmpty(prodIds) ? new ArrayList<>() : prodIds);
+                } else {
                     productTypeGroupVO.setProductIds(new ArrayList<>());
-                        products.add(productTypeGroupVO);
-                    }
                 }
+                products.add(productTypeGroupVO);
             }
             respVO.setProducts(products);
         }