SysUserServiceImpl.java 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562
  1. package com.ruoyi.system.service.impl;
  2. import cn.hutool.core.util.ObjectUtil;
  3. import com.baomidou.dynamic.datasource.annotation.DS;
  4. import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
  5. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  6. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  7. import com.ruoyi.common.annotation.DataScope;
  8. import com.ruoyi.common.annotation.TenantIgnore;
  9. import com.ruoyi.common.constant.BaseSourceConstant;
  10. import com.ruoyi.common.constant.UserConstants;
  11. import com.ruoyi.common.core.domain.entity.SysDept;
  12. import com.ruoyi.common.core.domain.entity.SysRole;
  13. import com.ruoyi.common.core.domain.entity.SysUser;
  14. import com.ruoyi.common.exception.ServiceException;
  15. import com.ruoyi.common.utils.SecurityUtils;
  16. import com.ruoyi.common.utils.StringUtils;
  17. import com.ruoyi.common.utils.bean.BeanValidators;
  18. import com.ruoyi.common.utils.spring.SpringUtils;
  19. import com.ruoyi.system.domain.SysPost;
  20. import com.ruoyi.system.domain.SysUserPost;
  21. import com.ruoyi.system.domain.SysUserRole;
  22. import com.ruoyi.system.mapper.*;
  23. import com.ruoyi.system.service.ISysConfigService;
  24. import com.ruoyi.system.service.ISysDeptService;
  25. import com.ruoyi.system.service.ISysUserService;
  26. import org.slf4j.Logger;
  27. import org.slf4j.LoggerFactory;
  28. import org.springframework.beans.factory.annotation.Autowired;
  29. import org.springframework.stereotype.Service;
  30. import org.springframework.transaction.annotation.Propagation;
  31. import org.springframework.transaction.annotation.Transactional;
  32. import org.springframework.util.CollectionUtils;
  33. import javax.validation.Validator;
  34. import java.util.*;
  35. import java.util.stream.Collectors;
  36. /**
  37. * 用户 业务层处理
  38. *
  39. * @author ruoyi
  40. */
  41. @Service
  42. public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> implements ISysUserService {
  43. private static final Logger log = LoggerFactory.getLogger(SysUserServiceImpl.class);
  44. @Autowired
  45. protected Validator validator;
  46. @Autowired
  47. private SysUserMapper userMapper;
  48. @Autowired
  49. private SysRoleMapper roleMapper;
  50. @Autowired
  51. private SysPostMapper postMapper;
  52. @Autowired
  53. private SysUserRoleMapper userRoleMapper;
  54. @Autowired
  55. private SysUserPostMapper userPostMapper;
  56. @Autowired
  57. private ISysConfigService configService;
  58. @Autowired
  59. private ISysDeptService sysDeptService;
  60. /**
  61. * 根据条件分页查询用户列表
  62. *
  63. * @param user 用户信息
  64. * @return 用户信息集合信息
  65. */
  66. @Override
  67. @DataScope(deptAlias = "d", userAlias = "u")
  68. public List<SysUser> selectUserList(SysUser user) {
  69. return userMapper.selectUserList(user);
  70. }
  71. /**
  72. * 根据条件分页查询已分配用户角色列表
  73. *
  74. * @param user 用户信息
  75. * @return 用户信息集合信息
  76. */
  77. @Override
  78. @DataScope(deptAlias = "d", userAlias = "u")
  79. public List<SysUser> selectAllocatedList(SysUser user) {
  80. return userMapper.selectAllocatedList(user);
  81. }
  82. /**
  83. * 根据条件分页查询未分配用户角色列表
  84. *
  85. * @param user 用户信息
  86. * @return 用户信息集合信息
  87. */
  88. @Override
  89. @DataScope(deptAlias = "d", userAlias = "u")
  90. public List<SysUser> selectUnallocatedList(SysUser user) {
  91. return userMapper.selectUnallocatedList(user);
  92. }
  93. /**
  94. * 通过用户名查询用户
  95. *
  96. * @param userName 用户名
  97. * @return 用户对象信息
  98. */
  99. @Override
  100. public SysUser selectUserByUserName(String tenantId, String userName) {
  101. return userMapper.selectUserByUserName(tenantId, userName);
  102. }
  103. /**
  104. * 通过用户ID查询用户
  105. *
  106. * @param userId 用户ID
  107. * @return 用户对象信息
  108. */
  109. @Override
  110. public SysUser selectUserById(Long userId) {
  111. return userMapper.selectUserById(userId);
  112. }
  113. /**
  114. * 查询用户所属角色组
  115. *
  116. * @param userName 用户名
  117. * @return 结果
  118. */
  119. @Override
  120. public String selectUserRoleGroup(String userName) {
  121. List<SysRole> list = roleMapper.selectRolesByUserName(userName);
  122. if (CollectionUtils.isEmpty(list)) {
  123. return StringUtils.EMPTY;
  124. }
  125. return list.stream().map(SysRole::getRoleName).collect(Collectors.joining(","));
  126. }
  127. /**
  128. * 查询用户所属岗位组
  129. *
  130. * @param userName 用户名
  131. * @return 结果
  132. */
  133. @Override
  134. public String selectUserPostGroup(String userName) {
  135. List<SysPost> list = postMapper.selectPostsByUserName(userName);
  136. if (CollectionUtils.isEmpty(list)) {
  137. return StringUtils.EMPTY;
  138. }
  139. return list.stream().map(SysPost::getPostName).collect(Collectors.joining(","));
  140. }
  141. /**
  142. * 校验用户名称是否唯一
  143. *
  144. * @param user 用户信息
  145. * @return 结果
  146. */
  147. @Override
  148. public boolean checkUserNameUnique(SysUser user) {
  149. Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
  150. SysUser info = userMapper.checkUserNameUnique(user);
  151. if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
  152. return UserConstants.NOT_UNIQUE;
  153. }
  154. return UserConstants.UNIQUE;
  155. }
  156. /**
  157. * 校验手机号码是否唯一
  158. *
  159. * @param user 用户信息
  160. * @return
  161. */
  162. @Override
  163. public boolean checkPhoneUnique(SysUser user) {
  164. Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
  165. SysUser info = userMapper.checkPhoneUnique(user.getPhonenumber());
  166. if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
  167. return UserConstants.NOT_UNIQUE;
  168. }
  169. return UserConstants.UNIQUE;
  170. }
  171. /**
  172. * 校验email是否唯一
  173. *
  174. * @param user 用户信息
  175. * @return
  176. */
  177. @Override
  178. public boolean checkEmailUnique(SysUser user) {
  179. Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
  180. SysUser info = userMapper.checkEmailUnique(user.getEmail());
  181. if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
  182. return UserConstants.NOT_UNIQUE;
  183. }
  184. return UserConstants.UNIQUE;
  185. }
  186. /**
  187. * 校验用户是否允许操作
  188. *
  189. * @param user 用户信息
  190. */
  191. @Override
  192. public void checkUserAllowed(SysUser user) {
  193. if (StringUtils.isNotNull(user.getUserId()) && user.isAdmin()) {
  194. throw new ServiceException("不允许操作超级管理员用户");
  195. }
  196. }
  197. /**
  198. * 校验用户是否有数据权限
  199. *
  200. * @param userId 用户id
  201. */
  202. @Override
  203. public void checkUserDataScope(Long userId) {
  204. if (!SysUser.isAdmin(SecurityUtils.getUserId())) {
  205. SysUser user = new SysUser();
  206. user.setUserId(userId);
  207. List<SysUser> users = SpringUtils.getAopProxy(this).selectUserList(user);
  208. if (StringUtils.isEmpty(users)) {
  209. throw new ServiceException("没有权限访问用户数据!");
  210. }
  211. }
  212. }
  213. /**
  214. * 新增保存用户信息
  215. *
  216. * @param user 用户信息
  217. * @return 结果
  218. */
  219. @Override
  220. @Transactional
  221. public int insertUser(SysUser user) {
  222. // 新增用户信息
  223. int rows = userMapper.insertUser(user);
  224. // 新增用户岗位关联
  225. insertUserPost(user);
  226. // 新增用户与角色管理
  227. insertUserRole(user);
  228. return rows;
  229. }
  230. /**
  231. * 注册用户信息
  232. *
  233. * @param user 用户信息
  234. * @return 结果
  235. */
  236. @Override
  237. public boolean registerUser(SysUser user) {
  238. return userMapper.insertUser(user) > 0;
  239. }
  240. /**
  241. * 修改保存用户信息
  242. *
  243. * @param user 用户信息
  244. * @return 结果
  245. */
  246. @Override
  247. @Transactional
  248. public int updateUser(SysUser user) {
  249. Long userId = user.getUserId();
  250. // 删除用户与角色关联
  251. userRoleMapper.deleteUserRoleByUserId(userId);
  252. // 新增用户与角色管理
  253. insertUserRole(user);
  254. // 删除用户与岗位关联
  255. userPostMapper.deleteUserPostByUserId(userId);
  256. // 新增用户与岗位管理
  257. insertUserPost(user);
  258. return userMapper.updateUser(user);
  259. }
  260. /**
  261. * 用户授权角色
  262. *
  263. * @param userId 用户ID
  264. * @param roleIds 角色组
  265. */
  266. @Override
  267. @Transactional
  268. public void insertUserAuth(Long userId, Long[] roleIds) {
  269. userRoleMapper.deleteUserRoleByUserId(userId);
  270. insertUserRole(userId, roleIds);
  271. }
  272. /**
  273. * 修改用户状态
  274. *
  275. * @param user 用户信息
  276. * @return 结果
  277. */
  278. @Override
  279. public int updateUserStatus(SysUser user) {
  280. return userMapper.updateUser(user);
  281. }
  282. /**
  283. * 修改用户基本信息
  284. *
  285. * @param user 用户信息
  286. * @return 结果
  287. */
  288. @TenantIgnore
  289. @Override
  290. public int updateUserProfile(SysUser user) {
  291. return userMapper.updateUser(user);
  292. }
  293. /**
  294. * 修改用户头像
  295. *
  296. * @param userName 用户名
  297. * @param avatar 头像地址
  298. * @return 结果
  299. */
  300. @Override
  301. public boolean updateUserAvatar(String userName, String avatar) {
  302. return userMapper.updateUserAvatar(userName, avatar) > 0;
  303. }
  304. /**
  305. * 重置用户密码
  306. *
  307. * @param user 用户信息
  308. * @return 结果
  309. */
  310. @Override
  311. public int resetPwd(SysUser user) {
  312. return userMapper.updateUser(user);
  313. }
  314. /**
  315. * 重置用户密码
  316. *
  317. * @param userName 用户名
  318. * @param password 密码
  319. * @return 结果
  320. */
  321. @Override
  322. public int resetUserPwd(String userName, String password) {
  323. return userMapper.resetUserPwd(userName, password);
  324. }
  325. /**
  326. * 新增用户角色信息
  327. *
  328. * @param user 用户对象
  329. */
  330. public void insertUserRole(SysUser user) {
  331. this.insertUserRole(user.getUserId(), user.getRoleIds());
  332. }
  333. /**
  334. * 新增用户岗位信息
  335. *
  336. * @param user 用户对象
  337. */
  338. public void insertUserPost(SysUser user) {
  339. Long[] posts = user.getPostIds();
  340. if (StringUtils.isNotEmpty(posts)) {
  341. // 新增用户与岗位管理
  342. List<SysUserPost> list = new ArrayList<SysUserPost>(posts.length);
  343. for (Long postId : posts) {
  344. SysUserPost up = new SysUserPost();
  345. up.setUserId(user.getUserId());
  346. up.setPostId(postId);
  347. list.add(up);
  348. }
  349. userPostMapper.batchUserPost(list);
  350. }
  351. }
  352. /**
  353. * 新增用户角色信息
  354. *
  355. * @param userId 用户ID
  356. * @param roleIds 角色组
  357. */
  358. public void insertUserRole(Long userId, Long[] roleIds) {
  359. if (StringUtils.isNotEmpty(roleIds)) {
  360. // 新增用户与角色管理
  361. List<SysUserRole> list = new ArrayList<SysUserRole>(roleIds.length);
  362. for (Long roleId : roleIds) {
  363. SysUserRole ur = new SysUserRole();
  364. ur.setUserId(userId);
  365. ur.setRoleId(roleId);
  366. list.add(ur);
  367. }
  368. userRoleMapper.batchUserRole(list);
  369. }
  370. }
  371. /**
  372. * 通过用户ID删除用户
  373. *
  374. * @param userId 用户ID
  375. * @return 结果
  376. */
  377. @Override
  378. @Transactional
  379. public int deleteUserById(Long userId) {
  380. // 删除用户与角色关联
  381. userRoleMapper.deleteUserRoleByUserId(userId);
  382. // 删除用户与岗位表
  383. userPostMapper.deleteUserPostByUserId(userId);
  384. return userMapper.deleteUserById(userId);
  385. }
  386. /**
  387. * 批量删除用户信息
  388. *
  389. * @param userIds 需要删除的用户ID
  390. * @return 结果
  391. */
  392. @Override
  393. @Transactional
  394. public int deleteUserByIds(Long[] userIds) {
  395. for (Long userId : userIds) {
  396. checkUserAllowed(new SysUser(userId));
  397. checkUserDataScope(userId);
  398. }
  399. // 删除用户与角色关联
  400. userRoleMapper.deleteUserRole(userIds);
  401. // 删除用户与岗位关联
  402. userPostMapper.deleteUserPost(userIds);
  403. return userMapper.deleteUserByIds(userIds);
  404. }
  405. /**
  406. * 导入用户数据
  407. *
  408. * @param userList 用户数据列表
  409. * @param isUpdateSupport 是否更新支持,如果已存在,则进行更新数据
  410. * @param operName 操作用户
  411. * @return 结果
  412. */
  413. @Override
  414. public String importUser(List<SysUser> userList, Boolean isUpdateSupport, String operName) {
  415. if (StringUtils.isNull(userList) || userList.size() == 0) {
  416. throw new ServiceException("导入用户数据不能为空!");
  417. }
  418. int successNum = 0;
  419. int failureNum = 0;
  420. StringBuilder successMsg = new StringBuilder();
  421. StringBuilder failureMsg = new StringBuilder();
  422. String password = configService.selectConfigByKey("sys.user.initPassword");
  423. for (SysUser user : userList) {
  424. try {
  425. // 验证是否存在这个用户
  426. SysUser u = userMapper.selectUserByUserName(user.getTenantId(), user.getUserName());
  427. if (StringUtils.isNull(u)) {
  428. BeanValidators.validateWithException(validator, user);
  429. user.setPassword(SecurityUtils.encryptPassword(password));
  430. user.setCreateBy(operName);
  431. this.insertUser(user);
  432. successNum++;
  433. successMsg.append("<br/>" + successNum + "、账号 " + user.getUserName() + " 导入成功");
  434. } else if (isUpdateSupport) {
  435. BeanValidators.validateWithException(validator, user);
  436. checkUserAllowed(user);
  437. checkUserDataScope(user.getUserId());
  438. user.setUpdateBy(operName);
  439. this.updateUser(user);
  440. successNum++;
  441. successMsg.append("<br/>" + successNum + "、账号 " + user.getUserName() + " 更新成功");
  442. } else {
  443. failureNum++;
  444. failureMsg.append("<br/>" + failureNum + "、账号 " + user.getUserName() + " 已存在");
  445. }
  446. } catch (Exception e) {
  447. failureNum++;
  448. String msg = "<br/>" + failureNum + "、账号 " + user.getUserName() + " 导入失败:";
  449. failureMsg.append(msg + e.getMessage());
  450. log.error(msg, e);
  451. }
  452. }
  453. if (failureNum > 0) {
  454. failureMsg.insert(0, "很抱歉,导入失败!共 " + failureNum + " 条数据格式不正确,错误如下:");
  455. throw new ServiceException(failureMsg.toString());
  456. } else {
  457. successMsg.insert(0, "恭喜您,数据已全部导入成功!共 " + successNum + " 条,数据如下:");
  458. }
  459. return successMsg.toString();
  460. }
  461. @Transactional(propagation = Propagation.REQUIRES_NEW)
  462. @DS(BaseSourceConstant.BASE)
  463. @Override
  464. public List<SysUser> getListByUserId(List<Long> userIdList) {
  465. return listByIds(userIdList);
  466. }
  467. @Override
  468. public List<SysUser> getListByPostId(Long postId) {
  469. return baseMapper.getListByPostId(postId);
  470. }
  471. @Override
  472. public List<SysUser> getListByRoleId(Long roleId) {
  473. return baseMapper.getListByRoleId(roleId);
  474. }
  475. @Override
  476. public List<SysUser> getAuthUserList() {
  477. DynamicDataSourceContextHolder.push(BaseSourceConstant.BASE);
  478. Long userId = SecurityUtils.getUserId();
  479. List<SysDept> sysDeptList = sysDeptService.list(Wrappers.<SysDept>lambdaQuery()
  480. .and(q -> q.eq(SysDept::getDirectorId, userId).or().eq(SysDept::getLeaderId, userId)));
  481. if (sysDeptList.size() == 0) {
  482. SysUser sysUser = getById(userId);
  483. sysUser.setPassword(null);
  484. return Collections.singletonList(sysUser);
  485. }
  486. Set<Long> deptSet = new HashSet<>();
  487. List<SysDept> list = sysDeptService.list();
  488. Map<Long, List<SysDept>> parentDeptMap = list.stream().collect(Collectors.groupingBy(SysDept::getParentId));
  489. for (SysDept sysDept : sysDeptList) {
  490. Long deptId = sysDept.getDeptId();
  491. if (Objects.equals(sysDept.getDirectorId(), userId)) {
  492. subordinateDeptId(deptId, deptSet, parentDeptMap);
  493. } else {
  494. deptSet.add(deptId);
  495. }
  496. }
  497. List<SysUser> sysUserList = list(Wrappers.<SysUser>lambdaQuery()
  498. .and(q -> q.in(SysUser::getDeptId, deptSet).or().eq(SysUser::getUserId, userId)));
  499. sysUserList.forEach(item -> item.setPassword(null));
  500. DynamicDataSourceContextHolder.poll();
  501. return sysUserList;
  502. }
  503. private void subordinateDeptId(Long deptId, Set<Long> deptSet, Map<Long, List<SysDept>> parentDeptMap) {
  504. deptSet.add(deptId);
  505. // 获取子部门
  506. List<SysDept> sysDeptList = parentDeptMap.get(deptId);
  507. if (ObjectUtil.isEmpty(sysDeptList)) {
  508. return;
  509. }
  510. for (SysDept sysDept : sysDeptList) {
  511. subordinateDeptId(sysDept.getDeptId(), deptSet, parentDeptMap);
  512. }
  513. }
  514. }