package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.whyc.dto.Response;
|
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 Response updateMenuList(List<Menu> menuList) {
|
List<Menu> allMenu = new LinkedList<>();
|
for (Menu menu : menuList){
|
allMenu.add(menu);
|
if (menu.getChildrens().size()!=0){
|
allMenu.addAll(menu.getChildrens());
|
}
|
}
|
//todo 国网菜单需要进行审核
|
|
//先删除所有再添加
|
mapper.deleteMenuAll();
|
int flag = mapper.insertBatchSomeColumn(allMenu);
|
return new Response().set(1,flag>0?true:false,"更新成功");
|
}
|
}
|
