public class MenuService implements IMenuService {
@Autowired
private SysMenuMapper sysMenuMapper;
public List<SysMenuDTO> queryAll(){
String lang = LocaleContextHolder.getLocale().getLanguage();
List<SysMenuDTO> rspMenuDTOList = new ArrayList();
List<SysMenu> sysMenuAllList = sysMenuMapper.queryAll();
if( sysMenuAllList == null || sysMenuAllList.size() == 0 ){
return rspMenuDTOList;
}
//根节点 ,父节点分类
Iterator iterator = sysMenuAllList.iterator();
Map<Long,List<SysMenuDTO>> map = new HashMap<Long,List<SysMenuDTO>>();
while(iterator.hasNext()){
SysMenu sysMenu = (SysMenu)iterator.next();
Long parentId = sysMenu.getParentId();
List<SysMenuDTO> list = map.get(parentId)==null?new ArrayList<SysMenuDTO>():map.get(parentId);
SysMenuDTO sysMenuDTO = new SysMenuDTO();
BeanUtility.copyBean(sysMenuDTO,sysMenu);
if(lang.equals("zh")){
sysMenuDTO.setName(sysMenu.getCnName());
}else {
sysMenuDTO.setName(sysMenu.getEnName());
}
list.add(sysMenuDTO);
map.put(parentId,list);
}
List<SysMenuDTO> rootList = map.get(0L);
//查找子节点
for(int i=0; i<rootList.size();i++){
findChildren(map,rootList.get(i));
}
return rootList;
}
public void findChildren( Map<Long,List<SysMenuDTO>> allMap,SysMenuDTO parentSysMenu){
List<SysMenuDTO> childList = allMap.get(parentSysMenu.getId());
if(childList!=null && childList.size()>0){
parentSysMenu.setChildren(childList);
for(int i=0;i<childList.size();i++){
findChildren(allMap,childList.get(i));
}
}
}
}
菜单树的生成代码
猜你喜欢
转载自blog.csdn.net/zhefudexiaojiahuo/article/details/81455402
今日推荐
周排行