所有代码

无关风月

2024-12-09 2053b8fe0e98d4b4449bc756a93ced78f42277c4

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
package com.jilongda.manage.security;
 
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.jilongda.manage.authority.mapper.SecResourcesMapper;
import com.jilongda.manage.authority.mapper.SecRoleMapper;
import com.jilongda.manage.authority.mapper.SecUserMapper;
import com.jilongda.manage.authority.model.SecRole;
import com.jilongda.manage.authority.model.SecUser;
import com.jilongda.manage.authority.service.SecUserService;
import com.jilongda.manage.authority.vo.SecResourceVO;
import com.jilongda.common.basic.Constant;
import com.jilongda.common.utils.SpringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;
 
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
 
 
/**
 * 用户登录认证信息查询
 *
 * @author xiaochen
 * @date Jun 29, 2019
 */
@Component("loadUserDetailsService")
public class SysUserDetailsService implements UserDetailsService {
    @Autowired
    private SecUserMapper secUserMapper;
    @Autowired
    private SecRoleMapper secRoleMapper;
    @Autowired
    private SecResourcesMapper secResourcesMapper;
    @Autowired
    private SecUserService secUserService;
 
 
    @Override
    public SecurityUserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
        SecUser user = secUserMapper.selectOne(Wrappers.lambdaQuery(SecUser.class).eq(SecUser::getAccount, username).last(" LIMIT 1"));
 
        if (Objects.isNull(user)) {
            throw new UsernameNotFoundException("该用户不存在");
        }
 
        if (user.getRoleType()==2){
            throw  new RuntimeException("该用户暂无权限");
        }
 
        SecurityUserDetails vo = SpringUtils.beanCopy(user, SecurityUserDetails.class);
        if (user.getAccount().equals(Constant.ADMIN)) {
            getAdminPermission(vo);
        } else {
            List<SecRole> roles = secRoleMapper.selectRolesByUid(user.getId());
            List<Long> roleIds = roles.stream().map(SecRole::getId).collect(Collectors.toList());
            List<SecResourceVO> resources;
            if (CollectionUtils.isEmpty(roleIds)) {
                resources = new ArrayList<>();
            } else {
                resources = secUserService.getResourceTag(user.getId());
            }
            vo.setRoles(roles);
            vo.setResources(resources);
        }
        // 更新登录时间
        SecurityUserDetails securityUserDetails = new SecurityUserDetails();
        securityUserDetails.setId(user.getId());
        securityUserDetails.setLastLoginTime(LocalDateTime.now());
        secUserMapper.updateById(securityUserDetails);
        return vo;
    }
 
 
    /**
     * 为管理员赋所有权限
     *
     * @param
     * @return
     */
    private SecurityUserDetails getAdminPermission(SecurityUserDetails details) {
        List<SecRole> roles = secRoleMapper.selectList(Wrappers.emptyWrapper());
        List<SecResourceVO> resources = secUserService.selectUserResourcesTag(details.getId());
        details.setRoles(roles);
        details.setResources(resources);
        return details;
    }
}