package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.whyc.mapper.MenuMapper;
|
import com.whyc.pojo.Menu;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.Collection;
|
import java.util.LinkedList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class MenuService {
|
|
@Resource
|
private MenuMapper mapper;
|
|
public List<Menu> getList() {
|
List<Menu> resultMenuList = new LinkedList<>();
|
|
QueryWrapper<Menu> wrapper = Wrappers.query();
|
wrapper.eq("enableduse",1);
|
List<Menu> menuList = mapper.selectList(wrapper);
|
Map<Integer, List<Menu>> collect = menuList.stream().collect(Collectors.groupingBy(Menu::getMenuId));
|
Collection<List<Menu>> values = collect.values();
|
for (List<Menu> listTemp:values){
|
Menu menu = new Menu();
|
List<Menu> subMenuList = new LinkedList<>();
|
for (Menu temp:listTemp){
|
if(temp.getLevel()==1){
|
menu = temp;
|
}else{
|
subMenuList.add(temp);
|
}
|
}
|
menu.setChildrens(subMenuList);
|
resultMenuList.add(menu);
|
|
}
|
|
return resultMenuList;
|
}
|
|
public void updateMenuList(List<Menu> menuList) {
|
mapper.updateMenuList(menuList);
|
}
|
}
|
