gongchunyi
12 小时以前 241c256ff1ba033d2c1d4f958534956ee37cd57a
src/main/java/com/ruoyi/framework/security/service/TokenService.java
@@ -159,16 +159,21 @@
    {
        loginUser.setLoginTime(System.currentTimeMillis());
        loginUser.setExpireTime(loginUser.getLoginTime() + expireTime * MILLIS_MINUTE);
        LambdaQueryWrapper<SysUserDept> sysUserDeptLambdaQueryWrapper = new LambdaQueryWrapper<>();
        sysUserDeptLambdaQueryWrapper.eq(SysUserDept::getUserId, loginUser.getUserId());
        List<SysUserDept> sysUserDept = sysUserDeptMapper.selectList(sysUserDeptLambdaQueryWrapper);
        if(!CollectionUtils.isEmpty(sysUserDept)){
            List<Long> collect = sysUserDept.stream().map(SysUserDept::getDeptId).collect(Collectors.toList());
            loginUser.setDeptId(collect.toArray(new Long[0]));
        }
        loginUser.setDeptIds(getDeptIdsByUserId(loginUser.getUserId()));
        loginUser.setCurrentDeptId(loginUser.getDeptIds()[0]);
        // 根据uuid将loginUser缓存
        String userKey = getTokenKey(loginUser.getToken());
        redisCache.setCacheObject(userKey, loginUser, expireTime, TimeUnit.MINUTES);
    }
    public Long[] getDeptIdsByUserId(Long userId){
        LambdaQueryWrapper<SysUserDept> sysUserDeptLambdaQueryWrapper = new LambdaQueryWrapper<>();
        sysUserDeptLambdaQueryWrapper.eq(SysUserDept::getUserId, userId);
        List<SysUserDept> sysUserDept = sysUserDeptMapper.selectList(sysUserDeptLambdaQueryWrapper);
        if(CollectionUtils.isEmpty(sysUserDept)){
            return null;
        }
        return sysUserDept.stream().map(SysUserDept::getDeptId).toArray(Long[]::new);
    }
    /**
@@ -244,6 +249,6 @@
    private String getTokenKey(String uuid)
    {
        return CacheConstants.LOGIN_TOKEN_KEY + uuid;
        return CacheConstants.PREFIX + CacheConstants.LOGIN_TOKEN_KEY + uuid;
    }
}