diff --git a/yshop-shop/src/main/java/co/yixiang/modules/shop/service/impl/YxStoreCategoryServiceImpl.java b/yshop-shop/src/main/java/co/yixiang/modules/shop/service/impl/YxStoreCategoryServiceImpl.java index 6b5c1bea..13b34b60 100644 --- a/yshop-shop/src/main/java/co/yixiang/modules/shop/service/impl/YxStoreCategoryServiceImpl.java +++ b/yshop-shop/src/main/java/co/yixiang/modules/shop/service/impl/YxStoreCategoryServiceImpl.java @@ -128,6 +128,7 @@ public class YxStoreCategoryServiceImpl implements YxStoreCategoryService { YxStoreCategoryDTO categoryDTO = new YxStoreCategoryDTO(); Boolean isChild; + List categories = yxStoreCategoryRepository.findAll(); for (YxStoreCategoryDTO deptDTO : categoryDTOS) { isChild = false; if ("0".equals(deptDTO.getPid().toString())) { @@ -144,8 +145,11 @@ public class YxStoreCategoryServiceImpl implements YxStoreCategoryService { } if(isChild) cates.add(deptDTO); - else if(!deptNames.contains(yxStoreCategoryRepository.findNameById(deptDTO.getPid()))) - cates.add(deptDTO); + for (YxStoreCategory category : categories) { + if(category.getId()==deptDTO.getPid()&&!deptNames.contains(category.getCateName())){ + cates.add(deptDTO); + } + } } @@ -164,4 +168,4 @@ public class YxStoreCategoryServiceImpl implements YxStoreCategoryService { return map; //return null; } -} \ No newline at end of file +} diff --git a/yshop-system/src/main/java/co/yixiang/modules/system/service/impl/DeptServiceImpl.java b/yshop-system/src/main/java/co/yixiang/modules/system/service/impl/DeptServiceImpl.java index 93696d6b..20eea935 100644 --- a/yshop-system/src/main/java/co/yixiang/modules/system/service/impl/DeptServiceImpl.java +++ b/yshop-system/src/main/java/co/yixiang/modules/system/service/impl/DeptServiceImpl.java @@ -72,6 +72,7 @@ public class DeptServiceImpl implements DeptService { Set depts= new LinkedHashSet<>(); List deptNames = deptDtos.stream().map(DeptDTO::getName).collect(Collectors.toList()); boolean isChild; + List deptList = deptRepository.findAll(); for (DeptDTO deptDTO : deptDtos) { isChild = false; if ("0".equals(deptDTO.getPid().toString())) { @@ -88,8 +89,11 @@ public class DeptServiceImpl implements DeptService { } if(isChild) { depts.add(deptDTO); - } else if(!deptNames.contains(deptRepository.findNameById(deptDTO.getPid()))) { - depts.add(deptDTO); + for (Dept dept : deptList) { + if(dept.getId() == deptDTO.getPid() && !deptNames.contains(dept.getName())){ + depts.add(deptDTO); + } + } } } @@ -158,4 +162,4 @@ public class DeptServiceImpl implements DeptService { } return deptDtos; } -} \ No newline at end of file +}