|
@@ -50,13 +50,14 @@ public class MenuService extends ServiceImpl<MenuMapper, Menu> implements IMenuS
|
|
|
|
|
|
@Override
|
|
|
public RestResponse all() {
|
|
|
- return RestResponse.ok(this.list(), ApiConstants.RESULT_SUCCESS, "获取所有菜单信息");
|
|
|
+ List<Menu> allMenus = this.list();
|
|
|
+ return RestResponse.ok(getMenuTree(allMenus), ApiConstants.RESULT_SUCCESS, "获取所有菜单信息");
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public RestResponse list(MenuRequest menuRequest) {
|
|
|
List<Menu> menuList = this.list(new QueryWrapper<Menu>().eq("company_id", menuRequest.getCompanyId()));
|
|
|
- return RestResponse.ok(menuList, ApiConstants.RESULT_SUCCESS, "获取企业所有菜单信息");
|
|
|
+ return RestResponse.ok(getMenuTree(menuList), ApiConstants.RESULT_SUCCESS, "获取企业所有菜单信息");
|
|
|
}
|
|
|
|
|
|
@Override
|
|
@@ -80,6 +81,11 @@ public class MenuService extends ServiceImpl<MenuMapper, Menu> implements IMenuS
|
|
|
return RestResponse.ok(null, ApiConstants.RESULT_SUCCESS, "分配菜单成功");
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
+ public List<Menu> getMenuTree(List<Menu> menuList) {
|
|
|
+ return TreeUtil.listToTree(menuList, ApiConstants.MENU_PARENT_ATTR, ApiConstants.ROOT_MENU_PARENT_ID, "id");
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* @param menuTree 菜单树,
|
|
|
* @param sb 路径StringBuilder对象
|