刘光辉
6 小时以前 06f314545ac3514d7ce28939920866ae8188d31c
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
package jnpf.dmsPermission;
 
import jnpf.base.UserInfo;
import jnpf.dmsEntity.permission.model.DmsSubjectSet;
import jnpf.permission.RoleRelationApi;
import jnpf.permission.entity.RoleRelationEntity;
import jnpf.permission.model.rolerelaiton.RoleRelationModel;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
 
import java.util.Collections;
import java.util.List;
 
@Component
@RequiredArgsConstructor
public class DmsSubjectResolver {
    private final DmsSubjectHierarchyResolver hierarchyResolver;
    private final RoleRelationApi roleRelationApi;
 
    public DmsSubjectSet resolve(UserInfo user) {
        DmsSubjectSet subjects = new DmsSubjectSet();
        subjects.setUserId(user.getUserId());
        subjects.getDirectOrgIds().addAll(orEmpty(user.getOrganizeIds()));
        subjects.getPositionIds().addAll(orEmpty(user.getPositionIds()));
        subjects.getRoleIds().addAll(orEmpty(user.getRoleIds()));
        subjects.getPlatformGroupIds().addAll(orEmpty(user.getGroupIds()));
        hierarchyResolver.resolve(Collections.singletonList(subjects));
        resolveDirectUserRoles(subjects);
        return subjects;
    }
 
    private void resolveDirectUserRoles(DmsSubjectSet subjects) {
        String userId = subjects.getUserId();
        if (userId == null || userId.trim().isEmpty()) {
            return;
        }
        List<RoleRelationEntity> relations = roleRelationApi.getListByObjectId(
                new RoleRelationModel(Collections.singletonList(userId), "user"));
        if (relations == null) {
            return;
        }
        for (RoleRelationEntity relation : relations) {
            if (relation != null && userId.equals(relation.getObjectId())
                    && "user".equalsIgnoreCase(relation.getObjectType())
                    && relation.getRoleId() != null && !relation.getRoleId().trim().isEmpty()) {
                subjects.getDirectUserRoleIds().add(relation.getRoleId().trim());
            }
        }
    }
 
    private List<String> orEmpty(List<String> values) {
        return values == null ? Collections.<String>emptyList() : values;
    }
}