diff --git a/src/main/java/com/gunshi/project/xyt/controller/CctvBMenuController.java b/src/main/java/com/gunshi/project/xyt/controller/CctvBMenuController.java index 4590997..de27f03 100644 --- a/src/main/java/com/gunshi/project/xyt/controller/CctvBMenuController.java +++ b/src/main/java/com/gunshi/project/xyt/controller/CctvBMenuController.java @@ -43,6 +43,15 @@ public class CctvBMenuController { throw new IllegalArgumentException("当前名称已存在"); } dto.setId(IdWorker.getId()); + if (Objects.isNull(dto.getOrderIndex())){ + CctvBMenu one = service.lambdaQuery() + .select(CctvBMenu::getOrderIndex) + .orderByDesc(CctvBMenu::getOrderIndex) + .one(); + if (Objects.nonNull(one)){ + dto.setOrderIndex(one.getOrderIndex() + 1); + } + } boolean result = service.save(dto); return R.ok(result ? dto : null); } diff --git a/src/main/java/com/gunshi/project/xyt/service/CctvBMenuService.java b/src/main/java/com/gunshi/project/xyt/service/CctvBMenuService.java index bd61654..341ae38 100644 --- a/src/main/java/com/gunshi/project/xyt/service/CctvBMenuService.java +++ b/src/main/java/com/gunshi/project/xyt/service/CctvBMenuService.java @@ -29,6 +29,7 @@ public class CctvBMenuService extends ServiceImpl if (CollectionUtils.isEmpty(list)){ return list; } + Map> listMap = list.stream().collect(Collectors.groupingBy(CctvBMenu::getParentId)); list.forEach(o -> o.setChildren(listMap.get(o.getId())));