jinlin
2024-01-10 cdf29a8f6fb9990f82ccc1d3f47bfba48cfaf4fc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/**
 * 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.sys.model.SysUser;
import com.zt.modules.sys.dao.SysRoleUserDao;
import com.zt.modules.sys.dto.RoleDto;
import com.zt.modules.sys.dto.UserDto;
import com.zt.modules.sys.model.SysRoleUser;
import org.springframework.stereotype.Service;
 
import java.util.List;
import java.util.stream.Collectors;
 
/**
 * 角色用户关系
 *
 * @author hehz
 * @since 1.0.0
 */
@Service
public class SysRoleUserService extends BaseService<SysRoleUserDao, SysRoleUser> {
 
    /**
     * 用户角色列表
     *
     * @param userId
     *            用户ID
     */
    public List<RoleDto> getUserRoles(Long userId) {
        return baseDao.getUserRoles(userId);
    }
 
 
    /**
     * 角色用户列表
     *
     * @param roleId
     *            角色ID
     */
    public List<SysUser> getRoleUsers(Long roleID) {
        return baseDao.getRoleUsers(roleID);
    }
    /**
     * 角色用户列表
     *
     * @param roleId
     *            角色ID
     */
    public List<SysRoleUser> getSingleRoleUsers(Long roleID,Long userID) {
        return baseDao.getSingleRoleUsers(roleID,userID);
    }
 
    /**
     * 保存用户角色
     *
     * @param userId
     * @param roleIdList
     */
    public void saveOrUpdate(Long userId, List<Long> roleIdList) {
        List<SysRoleUser> list = roleIdList.stream().map(roleId -> new SysRoleUser(roleId, userId))
                .collect(Collectors.toList());
        this.saveRelatedDatas("user_id", userId, "role_id", list);
 
        CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
    }
 
    /**
     * 根据角色ids,删除角色用户关系
     *
     * @param roleIds
     *            角色ids
     */
    public void deleteByRoleIds(Long[] roleIds) {
        baseDao.delete(new QueryWrapper<SysRoleUser>().lambda().in(SysRoleUser::getRoleId, roleIds));
 
        CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
    }
 
    /**
     * 根据用户id,删除角色用户关系
     *
     * @param userIds
     *            用户ids
     */
    public void deleteByUserIds(Long[] userIds) {
        baseDao.delete(new QueryWrapper<SysRoleUser>().lambda().in(SysRoleUser::getUserId, userIds));
 
        CacheUtils.removeAll(Constant.Cache.TOKEN);// 清理用户权限缓存
    }
 
    public  void  addRoleUser(Long roleId,List<Long> userIds){
        List<SysUser> list =  baseDao.getRoleUsers(roleId);
        List<SysRoleUser> sysRoleUsers = userIds.stream().map(userId -> new SysRoleUser(roleId, userId)).collect(Collectors.toList());
        for (SysRoleUser data : sysRoleUsers) {
            if(list.stream().filter(p-> p.getId().equals(data.getUserId())).count()==0){
                data.setDelete(false);
                baseDao.insert(data);
            }
            else {
                data.setDelete(false);
                baseDao.updateById(data);
            }
        }
        CacheUtils.remove(Constant.Cache.TOKEN, CacheKey.TOKEN_PERMISSIONS);// 清理用户权限缓存
    }
}