diff --git a/crmeb-service/src/main/java/com/zbkj/service/service/impl/CategoryServiceImpl.java b/crmeb-service/src/main/java/com/zbkj/service/service/impl/CategoryServiceImpl.java index 86afeb6..08b8028 100644 --- a/crmeb-service/src/main/java/com/zbkj/service/service/impl/CategoryServiceImpl.java +++ b/crmeb-service/src/main/java/com/zbkj/service/service/impl/CategoryServiceImpl.java @@ -107,7 +107,16 @@ public class CategoryServiceImpl extends ServiceImpl impl public List getByPIds(List idList) { LambdaQueryWrapper lambdaQueryWrapper = new LambdaQueryWrapper<>(); lambdaQueryWrapper.in(Category::getPid, idList); - return dao.selectList(lambdaQueryWrapper); + List two = dao.selectList(lambdaQueryWrapper); + if (CollUtil.isEmpty(two)){ + return two; + } + List twoIds=two.stream().map(Category::getId).collect(Collectors.toList()); + LambdaQueryWrapper lambdaQueryWrapper2 = new LambdaQueryWrapper<>(); + lambdaQueryWrapper2.in(Category::getPid, twoIds); + List three = dao.selectList(lambdaQueryWrapper2); + two.addAll(three); + return two; } /**