/**
* @ClassName: FunctionService
* @Description: 功能菜单
* @author: yaozhenhua
* @date: 2018/12/26 12:29
*/
@Service
public class MenuService {
@Resource
MenuDao menuDao;
/**
*查询可用且上线的菜单
*
* @param
* @author: yaozhenhua 2019/3/27 17:24
*/
public List<MenuVO> listMenu() throws InstantiationException, IllegalAccessException, AobpException {
//结果处理容器
List<MenuVO> menuVOS = new ArrayList<>();
//查询一级菜单
List<MenuBO> menuBOS = menuDao.listParentMenu();
if(!EmptyUtils.isEmpty(menuBOS)){
//一级菜单排序
Collections.sort(menuBOS, new Comparator<MenuBO>() {
@Override
public int compare(MenuBO o1, MenuBO o2) {
int ret = 0;
ret = o1.getParentSequence().compareTo(o2.getParentSequence());
return ret;
}
});
menuVOS = BeanCopyUtils.copyList(menuBOS,MenuVO.class);
for(int i=0; i<menuBOS.size(); i++){
MenuBO menuBO = menuBOS.get(i);
MenuVO menuVO = menuVOS.get(i);
//查询子菜单并排序
List<MenuVO> subMenuVOS = listSubMenu(menuBO.getId(),1);
menuVO.setSubs(subMenuVOS);
}
}
return menuVOS;
}
/**
*根据父id查询子菜单
*
* @param parentId
* @author: yaozhenhua 2019/3/27 17:24
*/
public List<MenuVO> listSubMenu(Long parentId, int alarm) throws InstantiationException, IllegalAccessException, AobpException {
//计数器 递归20次抛出异常,菜单层级最大没有超过20层的。
if(alarm>20){
throw new AobpException(ExceptionConstant.INTERNAL_ERR,"递归次数超过20次异常");
}
//查询出此父菜单下的子菜单
List<MenuBO> menuBOS= menuDao.listMenuByParentId(parentId);
//结果容器
List<MenuVO> menuVOS = new ArrayList<>();
//查看子菜单是否是还有下一级菜单(断定条件 用id做为父id去查 )
if (!EmptyUtils.isEmpty(menuBOS)){
Collections.sort(menuBOS, new Comparator<MenuBO>() {
@Override
public int compare(MenuBO o1, MenuBO o2) {
int ret = 0;
ret = o1.getChildSequence().compareTo(o2.getChildSequence());
return ret;
}
});
menuVOS = BeanCopyUtils.copyList(menuBOS,MenuVO.class);
for (int i = 0; i < menuBOS.size(); i++) {
MenuBO menuBO = menuBOS.get(i);
MenuVO menuVO = menuVOS.get(i);
List<MenuVO> nextMenuVOS = listSubMenu(menuBO.getId(),alarm+1);
if(!EmptyUtils.isEmpty(nextMenuVOS)){
menuVO.setSubs(nextMenuVOS);
}
}
}
return menuVOS;
}
}
另外一种方式:
/**
* @ClassName: MenuService
* @author: yaozhenhua
* @date: 2019/3/27 15:59
*/
public class MenuService {
@Transactional(readOnly = true)
public List<Menu> selectDocMenuList() {
// 查询所有文档
List<Menu> rootMenuList = docDao.selectListExpUnable();
// 最终的菜单
List<Menu> docList = new ArrayList<Menu>();
// 没有父ID的文档, 做为一级菜单
for (Menu rootMenu : rootMenuList) {
if (rootMenu.getDocPid() == null) {
docList.add(rootMenu);
}
}
// 为一级菜单添加子菜单
for (Menu menu : docList) {
menu.setChildList(getChild(menu.getDocId(), rootMenuList));
}
if (docList != null && docList.size() > 0) {
// 为list排序
this.mySort(docList, "docOrder", null);
}
return docList;
}
private List<Menu> getChild(String docId, List<Menu> rootMenu) {
if (StringUtils.isBlank(docId)) {
throw new BusinessException("文档id不能为空");
}
// 子菜单
List<Menu> childList = new ArrayList<>();
for (Menu menu : rootMenu) {
// 遍历全部节点,将父菜单id与传过来的type比较
if (menu.getDocPid() != null) {
if (menu.getDocPid().equals(docId)) {
childList.add(menu);
}
}
}
// 把子菜单的子菜单再循环一遍
for (Menu menu : childList) {// 没有内容的子菜单还有子菜单
try {
if (menu.getDocContentHref() != null) {
menu.setDocContent(new String(menu.getDocContentHref(), "UTF-8"));
}
if (StringUtils.isBlank(menu.getDocContent())) {
// 递归
menu.setChildList(getChild(menu.getDocId(), rootMenu));
}
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
}
} // 递归退出条件
if (childList != null && childList.size() > 0) {
this.mySort(childList, "docOrder", null);
} else {
return null;
}
return childList;
}
}
public
class
Menu {
// 菜单id
private
String id;
// 菜单名称
private
String name;
// 父菜单id
private
String parentId;
// 菜单url
private
String url;
// 菜单图标
private
String icon;
// 菜单顺序
private
int
order;
// 子菜单
private
List<Menu> children;
// ... 省去getter和setter方法以及toString方法
}
|
菜单通常须要排序,咱们根据Menu的order字段进行排序:java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
/*
* 排序,根据order排序
*/
public
Comparator<Menu> order(){
Comparator<Menu> comparator =
new
Comparator<Menu>() {
@Override
public
int
compare(Menu o1, Menu o2) {
if
(o1.getOrder() != o2.getOrder()){
return
o1.getOrder() - o2.getOrder();
}
return
0
;
}
};
return
comparator;
}
|
生成树的方法:ide
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
public
Map<String,Object> findTree(){
Map<String,Object> data =
new
HashMap<String,Object>();
try
{
//查询全部菜单
List<Menu> allMenu = menuDao.findTree();
//根节点
List<Menu> rootMenu =
new
ArrayList<Menu>();
for
(Menu nav : allMenu) {
if
(nav.getParentId().equals(
"0"
)){
//父节点是0的,为根节点。
rootMenu.add(nav);
}
}
/* 根据Menu类的order排序 */
Collections.sort(rootMenu, order());
//为根菜单设置子菜单,getClild是递归调用的
for (Menu nav : rootMenu) {
/* 获取根节点下的全部子节点 使用getChild方法*/
List<Menu> childList = getChild(nav.getId(), allMenu);
nav.setChildren(childList);//给根节点设置子节点
}
/**
* 输出构建好的菜单数据。
*
*/
data.put(
"success"
,
"true"
);
data.put(
"list"
, rootMenu);
return
data;
}
catch
(Exception e) {
data.put(
"success"
,
"false"
);
data.put(
"list"
,
new
ArrayList());
return
data;
}
}
|
获取子菜单:this
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
/**
* 获取子节点
* @param id 父节点id
* @param allMenu 全部菜单列表
* @return 每一个根节点下,全部子菜单列表
*/
public
List<Menu> getChild(String id,List<Menu> allMenu){
//子菜单
List<Menu> childList =
new
ArrayList<Menu>();
for
(Menu nav : allMenu) {
// 遍历全部节点,将全部菜单的父id与传过来的根节点的id比较
//相等说明:为该根节点的子节点。
if
(nav.ParentId().equals(id)){
childList.add(nav);
}
}
//递归
for
(Menu nav : childList) {
nav.setChildren(getChild(nav.getId(), allMenu));
}
Collections.sort(childList,order());
//排序
//若是节点下没有子节点,返回一个空List(递归退出)
if
(childList.size() ==
0
){
return
new
ArrayList<Menu>();
}
return
childList;
}
|