jtzx-crm/src/main/java/cc/uncarbon/module/sys/service/SysRoleMenuRelationService.java

package cc.uncarbon.module.sys.service;

import cc.uncarbon.framework.crud.service.impl.HelioBaseServiceImpl;
import cc.uncarbon.module.sys.entity.SysRoleMenuRelationEntity;
import cc.uncarbon.module.sys.mapper.SysRoleMenuRelationMapper;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.List;
import java.util.stream.Collectors;


/**
 * 后台角色-可见菜单关联
 *
 * @author Uncarbon
 */
@Slf4j
@Service
public class SysRoleMenuRelationService extends HelioBaseServiceImpl<SysRoleMenuRelationMapper, SysRoleMenuRelationEntity> {

    @Resource
    private SysMenuService sysMenuService;


    /**
     * 根据角色Ids取菜单Ids
     *
     * @param roleIds 角色Ids
     * @return 菜单Ids
     */
    public List<Long> listMenuIdByRoleIds(List<Long> roleIds) throws IllegalArgumentException {
        if (CollUtil.isEmpty(roleIds)) {
            throw new IllegalArgumentException("roleIds不能为空");
        }

        List<Long> menuIds = this.list(
                new QueryWrapper<SysRoleMenuRelationEntity>()
                        .select(" DISTINCT menu_id ")
                        .lambda()
                        .in(SysRoleMenuRelationEntity::getRoleId, roleIds)
        ).stream().map(SysRoleMenuRelationEntity::getMenuId).collect(Collectors.toList());

        return sysMenuService.filterDisabledIds(menuIds);
    }

    /**
     * 先清理角色ID所有关联关系, 再绑定角色ID与菜单ID
     */
    @Transactional(rollbackFor = Exception.class)
    public void cleanAndBind(Long roleId, List<Long> menuIds) {
        this.remove(
                new QueryWrapper<SysRoleMenuRelationEntity>()
                        .lambda()
                        .eq(SysRoleMenuRelationEntity::getRoleId, roleId)
        );

        if (CollUtil.isNotEmpty(menuIds)) {
            // 需要绑定菜单

            menuIds.forEach(
                    menuId -> this.save(
                            SysRoleMenuRelationEntity.builder()
                                    .roleId(roleId)
                                    .menuId(menuId)
                                    .build()
                    )
            );
        }
    }
}