jiaosong
2023-08-11 c2e4ed1a6e3a14406db471304360cd45ac8177d1
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
package doumeemes.service.system.impl;
 
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import doumeemes.core.model.PageData;
import doumeemes.core.model.PageWrap;
import doumeemes.dao.system.SystemUserMapper;
import doumeemes.dao.system.dto.QuerySystemUserDTO;
import doumeemes.dao.system.model.SystemUser;
import doumeemes.dao.system.vo.SystemUserListVO;
import doumeemes.service.system.SystemDepartmentService;
import doumeemes.service.system.SystemPositionService;
import doumeemes.service.system.SystemRoleService;
import doumeemes.service.system.SystemUserService;
import doumeemes.dao.system.vo.SystemDepartmentListVO;
import doumeemes.service.aware.DepartmentDataPermissionAware;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
 
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
 
/**
 * 系统用户Service实现
 * @author Eva.Caesar Liu
 * @date 2022/04/18 18:12
 */
@Service
public class SystemUserServiceImpl implements SystemUserService {
 
    @Autowired
    private SystemUserMapper systemUserMapper;
 
    @Autowired
    private SystemRoleService systemRoleService;
 
    @Autowired
    private SystemDepartmentService systemDepartmentService;
 
    @Autowired
    private SystemPositionService systemPositionService;
 
    @Autowired
    private DepartmentDataPermissionAware departmentDataPermissionAware;
 
    @Override
    public Integer create(SystemUser systemUser) {
        systemUserMapper.insert(systemUser);
        return systemUser.getId();
    }
 
    @Override
    public void deleteById(Integer id) {
        SystemUser systemUser = new SystemUser();
        systemUser.setId(id);
        systemUser.setDeleted(Boolean.TRUE);
        this.updateById(systemUser);
    }
 
    @Override
    @Transactional
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) return;
        for (Integer id : ids) {
            this.deleteById(id);
        }
    }
 
    @Override
    public void updateById(SystemUser systemUser) {
        systemUserMapper.updateById(systemUser);
    }
 
    @Override
    @Transactional
    public void updateByIdInBatch(List<SystemUser> systemUsers) {
        if (CollectionUtils.isEmpty(systemUsers)) return;
        for (SystemUser systemUser: systemUsers) {
            this.updateById(systemUser);
        }
    }
 
    @Override
    public SystemUser findById(Integer id) {
        return systemUserMapper.selectById(id);
    }
 
    @Override
    public SystemUser findOne(SystemUser systemUser) {
        QueryWrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectOne(wrapper.last("limit 1"));
    }
 
    @Override
    public List<SystemUser> findList(SystemUser systemUser) {
        Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectList(wrapper);
    }
  
    @Override
    public PageData<SystemUserListVO> findPage(PageWrap<QuerySystemUserDTO> pageWrap) {
        // 根部门条件处理(需查询根部门下所有部门的用户)
        if (pageWrap.getModel().getRootDeptId() != null) {
            pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(pageWrap.getModel().getRootDeptId()));
        } else {
            pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(null));
        }
        // 执行查询
        PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
        List<SystemUserListVO> userList = systemUserMapper.selectManageList(pageWrap.getModel(), pageWrap.getOrderByClause());
        for (SystemUserListVO user : userList) {
            // 查询用户角色列表
            user.setRoles(systemRoleService.findByUserId(user.getId()));
            // 查询用户岗位列表
            user.setPositions(systemPositionService.findByUserId(user.getId()));
        }
        return PageData.from(new PageInfo<>(userList));
    }
 
 
    @Override
    public List<SystemUserListVO> findSystemUsers(Integer type) {
        QueryWrapper<SystemUser> wrapper = new QueryWrapper<>();
        wrapper.lambda().eq(SystemUser::getType,type);
//                        .like(StringUtils.isEmpty(querySystemUserDTO.getUsername()),SystemUser::getUsername,querySystemUserDTO.getUsername());
 
        List<SystemUser> systemUsers = systemUserMapper.selectList(wrapper);
        List<SystemUserListVO> collect = systemUsers.stream().map(s -> {
            SystemUserListVO systemUserListVO = new SystemUserListVO();
            systemUserListVO.setId(s.getId());
            systemUserListVO.setUsername(s.getUsername());
            systemUserListVO.setRealname(s.getRealname());
            systemUserListVO.setEmpNo(s.getEmpNo());
            systemUserListVO.setBirthday(s.getBirthday());
            systemUserListVO.setSex(s.getSex());
            systemUserListVO.setEmail(s.getEmail());
            systemUserListVO.setMobile(s.getMobile());
            return systemUserListVO;
 
        }).collect(Collectors.toList());
        return collect;
    }
 
    @Override
    public long count(SystemUser systemUser) {
        Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectCount(wrapper);
    }
 
    /**
     * 获取用户权限内允许查询的部门ID
     */
    private List<Integer> getAllowedDeptIds(Integer rootDeptId) {
        List<SystemDepartmentListVO> allowedDepartments = departmentDataPermissionAware.execute();
        List<Integer> allowedDeptIds = new ArrayList<>();
        for (SystemDepartmentListVO listVO : allowedDepartments) {
            injectIds(allowedDeptIds, listVO);
        }
        // 没有允许的部门
        if (allowedDeptIds.size() == 0) {
            allowedDeptIds.add(-1);
            return allowedDeptIds;
        }
        if (rootDeptId == null) {
            return allowedDeptIds;
        }
        List<Integer> departmentIds = systemDepartmentService.findChildren(rootDeptId);
        departmentIds.add(rootDeptId);
        departmentIds.removeIf(deptId -> !allowedDeptIds.contains(deptId));
        return departmentIds;
    }
 
    /**
     * 递归注入用户权限内的部门ID
     */
    private void injectIds (List<Integer> pool, SystemDepartmentListVO listVO) {
        pool.add(listVO.getId());
        if (CollectionUtils.isEmpty(listVO.getChildren())) {
            return;
        }
        for (SystemDepartmentListVO child : listVO.getChildren()) {
            injectIds(pool, child);
        }
    }
}