/**
|
* Copyright (c) 2018 人人开源 All rights reserved.
|
*
|
* https://www.renren.io
|
*
|
* 版权所有，侵权必究！
|
*/
|
|
package com.zt.modules.sys.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.zt.common.constant.CacheKey;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.service.BaseService;
|
import com.zt.common.utils.CacheUtils;
|
import com.zt.core.context.UserContext;
|
import com.zt.modules.sys.dao.SysRoleMenuDao;
|
import com.zt.modules.sys.dto.RoleDto;
|
import com.zt.modules.sys.model.SysRoleMenu;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 角色与菜单对应关系
|
*
|
* @author hehz
|
*/
|
@Service
|
public class SysRoleMenuService extends BaseService<SysRoleMenuDao, SysRoleMenu> {
|
|
@Autowired
|
private SysRoleUserService sysRoleUserService;
|
|
@Transactional(rollbackFor = Exception.class)
|
public void saveOrUpdate(Long roleId, List<Long> menuIdList) {
|
List<SysRoleMenu> list = menuIdList.stream().map(menuId -> new SysRoleMenu(roleId, menuId))
|
.collect(Collectors.toList());
|
this.saveRelatedDatas("role_id", roleId, "menu_id", list);
|
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
}
|
|
/**
|
* 根据角色ID，获取菜单ID列表
|
*/
|
public List<Long> getRoleMenuIds(Long roleId) {
|
List<Long> list = baseDao.getByRoleId(roleId).stream().map(d -> d.getMenuId()).collect(Collectors.toList());
|
return list;
|
}
|
|
/**
|
* 根据角色id，删除角色菜单关系
|
*
|
* @param roleIds
|
* 角色ids
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void deleteByRoleIds(Long[] roleIds) {
|
this.deleteLogic(baseDao.selectList(new QueryWrapper<SysRoleMenu>().lambda()
|
.in(SysRoleMenu::getRoleId, roleIds)));
|
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
}
|
|
/**
|
* 根据菜单id，删除角色菜单关系
|
*
|
* @param menuId
|
* 菜单id
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void deleteByMenuId(Long menuId) {
|
baseDao.delete(new QueryWrapper<SysRoleMenu>().lambda().eq(SysRoleMenu::getMenuId, menuId));
|
|
CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
|
}
|
|
public List<SysRoleMenu> roleMenu() {
|
List<RoleDto> roles = sysRoleUserService.getUserRoles(UserContext.getUser().getId());
|
List<SysRoleMenu> sysRoleMenu = null;
|
for (RoleDto role : roles) {
|
RoleDto roleDto = new RoleDto();
|
roleDto.setId(role.getId());
|
sysRoleMenu = baseDao.roleMenu(roleDto);
|
}
|
return sysRoleMenu;
|
}
|
|
}
|
