|
@@ -4,6 +4,7 @@ package com.huaxu.service;
|
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
import com.huaxu.dao.MenuMapper;
|
|
|
import com.huaxu.entity.MenuEntity;
|
|
|
+import org.springframework.beans.BeanUtils;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
@@ -13,6 +14,9 @@ import org.springframework.transaction.annotation.Transactional;
|
|
|
import java.util.Map;
|
|
|
import java.util.List;
|
|
|
import java.util.Arrays;
|
|
|
+
|
|
|
+import static com.google.common.collect.Lists.newArrayList;
|
|
|
+
|
|
|
/**
|
|
|
*
|
|
|
* 菜单Service接口
|
|
@@ -36,9 +40,34 @@ public class MenuService extends ServiceImpl<MenuMapper,MenuEntity> {
|
|
|
* 查列表
|
|
|
*/
|
|
|
public List<MenuEntity> findList(MenuEntity menuEntity) {
|
|
|
- return menuMapper.findList(menuEntity);
|
|
|
+ List<MenuEntity> menuEntities = menuMapper.findList(menuEntity);
|
|
|
+ List<MenuEntity> menuTree = getMenuTree(menuEntities, Long.valueOf(0), 1);
|
|
|
+ return menuTree;
|
|
|
+ }
|
|
|
+
|
|
|
+ public static List<MenuEntity> getMenuTree(List<MenuEntity> list, Long id, Integer leve) {
|
|
|
+ List<MenuEntity> temList = newArrayList();
|
|
|
+ if (list != null) {
|
|
|
+ for (MenuEntity menuEntity : list) {
|
|
|
+ if (id.equals(menuEntity.getParentMenuId())) {
|
|
|
+ List<MenuEntity> chidren = getMenuTree(list, menuEntity.getId(), ++leve);
|
|
|
+ MenuEntity temMeanu = new MenuEntity();
|
|
|
+ BeanUtils.copyProperties(menuEntity, temMeanu);
|
|
|
+ temMeanu.setChildren(chidren);
|
|
|
+ temList.add(temMeanu);
|
|
|
+ leve--;
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if(temList.size() > 0){
|
|
|
+ return temList;
|
|
|
+ }else {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
+
|
|
|
/**
|
|
|
* 按父级节点查列表
|
|
|
*/
|