whyczh
2021-05-14 fd2d92448a1fa0031daa0e05c739109093acdc3f
src/main/java/com/whyc/service/RoleService.java
@@ -26,8 +26,7 @@
    private RoleMapper mapper;
    @Resource
    private UserRoleMapper userRoleMapper;
    @Resource
    private UserMapper userMapper;
    public List<Role> getAll() {
        return mapper.selectList(null);
@@ -73,37 +72,7 @@
        }
    }
    //绑定用户
    public Response getUsersByRoleId(int roleId){
        List<User> userList = userMapper.selectList(null);
        QueryWrapper<UserRole> queryWrapper= new QueryWrapper();
        queryWrapper.eq("role_id",roleId);
        List<UserRole> userRoleList = userRoleMapper.selectList(queryWrapper);
        for (UserRole ur:userRoleList) {
            for (User u:userList) {
                if(ur.getUserId().equals(u.getId())){
                    u.setSelected(true);
                    break;
                }
            }
        }
        return new Response().set(1,userList);
    }
    public void updateRoleUser(int roleId,List<User> users){
        QueryWrapper<UserRole> queryWrapper = new QueryWrapper<UserRole>().eq("role_id",roleId);
        userRoleMapper.delete(queryWrapper);
        for (User user:users) {
            if(user.isSelected()){
                UserRole userRole = new UserRole();
                userRole.setUserId(user.getId());
                userRole.setRoleId(roleId);
                userRoleMapper.insert(userRole);
            }
        }
    }
@@ -112,19 +81,7 @@
    }
    public Response<PageInfo<User>> getUserByRoleId(int pageNum,int pageSize,int roleId){
        PageHelper.startPage(pageNum,pageSize);
        QueryWrapper queryWrapper = new QueryWrapper();
        queryWrapper.select("user_id").eq("role_id",roleId);
        List<UserRole> userIds = userRoleMapper.selectList(queryWrapper);
        List<User> userList = new ArrayList<>();
        for (UserRole userRole:userIds) {
            User user = userMapper.selectById(userRole.getUserId());
            userList.add(user);
        }
        PageInfo<User> pageInfo = new PageInfo<>(userList);
        return new Response<PageInfo<User>>().set(1,pageInfo);
    }
    public Response updateRoleState(int roleId,int state){
        Role role = mapper.selectById(roleId);