소스 검색

:zap: 优化菜单新增逻辑

smallchill 5 년 전
부모
커밋
9e36be1049
1개의 변경된 파일6개의 추가작업 그리고 0개의 파일을 삭제
  1. 6 0
      src/main/java/org/springblade/modules/system/service/impl/MenuServiceImpl.java

+ 6 - 0
src/main/java/org/springblade/modules/system/service/impl/MenuServiceImpl.java

@@ -235,6 +235,12 @@ public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements IM
 		if (menu.getParentId() == null && menu.getId() == null) {
 			menu.setParentId(BladeConstant.TOP_PARENT_ID);
 		}
+		if (menu.getParentId() != null && menu.getId() == null) {
+			Menu parentMenu = baseMapper.selectById(menu.getParentId());
+			if (parentMenu.getCategory() != 1) {
+				throw new ServiceException("父节点只可选择菜单类型!");
+			}
+		}
 		menu.setIsDeleted(BladeConstant.DB_NOT_DELETED);
 		return saveOrUpdate(menu);
 	}