瀏覽代碼

: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);
 	}