package com.example.springboot.service.impl;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.example.springboot.entity.vo.MenuListVo;
|
import com.example.springboot.mapper.MenuListMapper;
|
import com.example.springboot.mapper.RoleMenuListMapper;
|
import com.example.springboot.security.util.SecurityUtil;
|
import com.example.springboot.service.MenuListService;
|
import com.example.springboot.service.RoleService;
|
import com.example.springboot.entity.MenuList;
|
import com.example.springboot.entity.Role;
|
import com.example.springboot.entity.RoleMenuList;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@Service
|
@Slf4j
|
public class MenuListServiceImpl extends ServiceImpl<MenuListMapper, MenuList> implements MenuListService {
|
@Autowired
|
private RoleService roleService;
|
@Autowired
|
private RoleMenuListMapper roleMenuListDao;
|
|
@Override
|
public IPage<MenuList> selectPage(MenuListVo menuListVO) {
|
return lambdaQuery()
|
.like(StrUtil.isNotBlank(menuListVO.getName()), MenuList::getName, menuListVO.getName())
|
.eq(null != menuListVO.getState(), MenuList::getState, menuListVO.getState())
|
.between(null != menuListVO.getStartTime() && null != menuListVO.getEndTime(), MenuList::getCreateTime, menuListVO.getStartTime(), menuListVO.getEndTime())
|
.orderByAsc(MenuList::getParentId).last(",create_time desc")
|
.page(new Page<>(menuListVO.getPageNum(), menuListVO.getPageSize()));
|
}
|
|
@Override
|
public List<MenuList> selectList(Long roleId) {
|
List<Role> roles = roleService.selectRoles(roleId == null ? SecurityUtil.getCurrentUser().getRoleId() : roleId, true);
|
List<Long> ids = roles.stream().map(Role::getId).collect(Collectors.toList());
|
List<RoleMenuList> roleMenuLists = roleMenuListDao.selectList(new QueryWrapper<RoleMenuList>().lambda()
|
.in(RoleMenuList::getRoleId, ids));
|
List<Long> collect = roleMenuLists.stream().map(RoleMenuList::getMenuListId).collect(Collectors.toList());
|
if (CollectionUtil.isEmpty(collect)) return new ArrayList<>();
|
List<MenuList> menuLists = lambdaQuery()
|
.or(menuListLambdaQueryWrapper -> menuListLambdaQueryWrapper.isNull(MenuList::getParentId).or().eq(MenuList::getParentId, ""))
|
.in(MenuList::getId, collect)
|
.list();
|
for (MenuList menuList : menuLists) {
|
List<MenuList> list = selectByParentId(menuList.getId(), collect);
|
menuList.setMenuLists(list);
|
}
|
return menuLists;
|
}
|
|
@Override
|
public List<MenuList> selectByParentId(Long parentId, List<Long> ids) {
|
List<MenuList> list = lambdaQuery().eq(MenuList::getParentId, parentId).in(MenuList::getId, ids).list();
|
if (CollectionUtil.isEmpty(list)) return null;
|
for (MenuList menuList : list) {
|
List<MenuList> menuLists = selectByParentId(menuList.getId(), ids);
|
menuList.setMenuLists(menuLists);
|
}
|
return list;
|
}
|
}
|
