SysUserServiceImpl.java 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502
  1. package com.ruoyi.system.service.impl;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.util.ObjectUtil;
  4. import com.baomidou.mybatisplus.core.conditions.Wrapper;
  5. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  6. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  7. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  8. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  9. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  10. import com.ruoyi.common.constant.UserConstants;
  11. import com.ruoyi.common.core.domain.PageQuery;
  12. import com.ruoyi.common.core.domain.entity.SysDept;
  13. import com.ruoyi.common.core.domain.entity.SysRole;
  14. import com.ruoyi.common.core.domain.entity.SysUser;
  15. import com.ruoyi.common.core.page.TableDataInfo;
  16. import com.ruoyi.common.exception.ServiceException;
  17. import com.ruoyi.common.helper.DataBaseHelper;
  18. import com.ruoyi.common.helper.LoginHelper;
  19. import com.ruoyi.common.utils.StringUtils;
  20. import com.ruoyi.system.domain.SysPost;
  21. import com.ruoyi.system.domain.SysUserPost;
  22. import com.ruoyi.system.domain.SysUserRole;
  23. import com.ruoyi.system.mapper.*;
  24. import com.ruoyi.system.service.ISysUserService;
  25. import lombok.RequiredArgsConstructor;
  26. import lombok.extern.slf4j.Slf4j;
  27. import org.springframework.stereotype.Service;
  28. import org.springframework.transaction.annotation.Transactional;
  29. import java.util.ArrayList;
  30. import java.util.Arrays;
  31. import java.util.List;
  32. import java.util.Map;
  33. import java.util.stream.Collectors;
  34. /**
  35. * 用户 业务层处理
  36. *
  37. * @author Lion Li
  38. */
  39. @Slf4j
  40. @RequiredArgsConstructor
  41. @Service
  42. public class SysUserServiceImpl implements ISysUserService {
  43. private final SysUserMapper baseMapper;
  44. private final SysDeptMapper deptMapper;
  45. private final SysRoleMapper roleMapper;
  46. private final SysPostMapper postMapper;
  47. private final SysUserRoleMapper userRoleMapper;
  48. private final SysUserPostMapper userPostMapper;
  49. @Override
  50. public TableDataInfo<SysUser> selectPageUserList(SysUser user, PageQuery pageQuery) {
  51. Page<SysUser> page = baseMapper.selectPageUserList(pageQuery.build(), this.buildQueryWrapper(user));
  52. return TableDataInfo.build(page);
  53. }
  54. /**
  55. * 根据条件分页查询用户列表
  56. *
  57. * @param user 用户信息
  58. * @return 用户信息集合信息
  59. */
  60. @Override
  61. public List<SysUser> selectUserList(SysUser user) {
  62. return baseMapper.selectUserList(this.buildQueryWrapper(user));
  63. }
  64. private Wrapper<SysUser> buildQueryWrapper(SysUser user) {
  65. Map<String, Object> params = user.getParams();
  66. QueryWrapper<SysUser> wrapper = Wrappers.query();
  67. wrapper.eq("u.del_flag", UserConstants.USER_NORMAL)
  68. .eq(ObjectUtil.isNotNull(user.getUserId()), "u.user_id", user.getUserId())
  69. .like(StringUtils.isNotBlank(user.getUserName()), "u.user_name", user.getUserName())
  70. .eq(StringUtils.isNotBlank(user.getStatus()), "u.status", user.getStatus())
  71. .like(StringUtils.isNotBlank(user.getPhonenumber()), "u.phonenumber", user.getPhonenumber())
  72. .between(params.get("beginTime") != null && params.get("endTime") != null,
  73. "u.create_time", params.get("beginTime"), params.get("endTime"))
  74. .and(ObjectUtil.isNotNull(user.getDeptId()), w -> {
  75. List<SysDept> deptList = deptMapper.selectList(new LambdaQueryWrapper<SysDept>()
  76. .select(SysDept::getDeptId)
  77. .apply(DataBaseHelper.findInSet(user.getDeptId(), "ancestors")));
  78. List<Long> ids = deptList.stream().map(SysDept::getDeptId).collect(Collectors.toList());
  79. ids.add(user.getDeptId());
  80. w.in("u.dept_id", ids);
  81. });
  82. return wrapper;
  83. }
  84. /**
  85. * 根据条件分页查询已分配用户角色列表
  86. *
  87. * @param user 用户信息
  88. * @return 用户信息集合信息
  89. */
  90. @Override
  91. public TableDataInfo<SysUser> selectAllocatedList(SysUser user, PageQuery pageQuery) {
  92. QueryWrapper<SysUser> wrapper = Wrappers.query();
  93. wrapper.eq("u.del_flag", UserConstants.USER_NORMAL)
  94. .eq(ObjectUtil.isNotNull(user.getRoleId()), "r.role_id", user.getRoleId())
  95. .like(StringUtils.isNotBlank(user.getUserName()), "u.user_name", user.getUserName())
  96. .eq(StringUtils.isNotBlank(user.getStatus()), "u.status", user.getStatus())
  97. .like(StringUtils.isNotBlank(user.getPhonenumber()), "u.phonenumber", user.getPhonenumber());
  98. Page<SysUser> page = baseMapper.selectAllocatedList(pageQuery.build(), wrapper);
  99. return TableDataInfo.build(page);
  100. }
  101. /**
  102. * 根据条件分页查询未分配用户角色列表
  103. *
  104. * @param user 用户信息
  105. * @return 用户信息集合信息
  106. */
  107. @Override
  108. public TableDataInfo<SysUser> selectUnallocatedList(SysUser user, PageQuery pageQuery) {
  109. List<Long> userId = userRoleMapper.selectUserIdsByRoleId(user.getRoleId());
  110. QueryWrapper<SysUser> wrapper = Wrappers.query();
  111. wrapper.eq("u.del_flag", UserConstants.USER_NORMAL)
  112. .and(w -> w.ne("r.role_id", user.getRoleId()).or().isNull("r.role_id"))
  113. .notIn("u.user_id", userId)
  114. .like(StringUtils.isNotBlank(user.getUserName()), "u.user_name", user.getUserName())
  115. .like(StringUtils.isNotBlank(user.getPhonenumber()), "u.phonenumber", user.getPhonenumber());
  116. Page<SysUser> page = baseMapper.selectUnallocatedList(pageQuery.build(), wrapper);
  117. return TableDataInfo.build(page);
  118. }
  119. /**
  120. * 通过用户名查询用户
  121. *
  122. * @param userName 用户名
  123. * @return 用户对象信息
  124. */
  125. @Override
  126. public SysUser selectUserByUserName(String userName) {
  127. return baseMapper.selectUserByUserName(userName);
  128. }
  129. /**
  130. * 通过手机号查询用户
  131. *
  132. * @param phonenumber 手机号
  133. * @return 用户对象信息
  134. */
  135. @Override
  136. public SysUser selectUserByPhonenumber(String phonenumber) {
  137. return baseMapper.selectUserByPhonenumber(phonenumber);
  138. }
  139. /**
  140. * 通过用户ID查询用户
  141. *
  142. * @param userId 用户ID
  143. * @return 用户对象信息
  144. */
  145. @Override
  146. public SysUser selectUserById(Long userId) {
  147. return baseMapper.selectUserById(userId);
  148. }
  149. /**
  150. * 查询用户所属角色组
  151. *
  152. * @param userName 用户名
  153. * @return 结果
  154. */
  155. @Override
  156. public String selectUserRoleGroup(String userName) {
  157. List<SysRole> list = roleMapper.selectRolesByUserName(userName);
  158. if (CollUtil.isEmpty(list)) {
  159. return StringUtils.EMPTY;
  160. }
  161. return list.stream().map(SysRole::getRoleName).collect(Collectors.joining(","));
  162. }
  163. /**
  164. * 查询用户所属岗位组
  165. *
  166. * @param userName 用户名
  167. * @return 结果
  168. */
  169. @Override
  170. public String selectUserPostGroup(String userName) {
  171. List<SysPost> list = postMapper.selectPostsByUserName(userName);
  172. if (CollUtil.isEmpty(list)) {
  173. return StringUtils.EMPTY;
  174. }
  175. return list.stream().map(SysPost::getPostName).collect(Collectors.joining(","));
  176. }
  177. /**
  178. * 校验用户名称是否唯一
  179. *
  180. * @param userName 用户名称
  181. * @return 结果
  182. */
  183. @Override
  184. public String checkUserNameUnique(String userName) {
  185. boolean exist = baseMapper.exists(new LambdaQueryWrapper<SysUser>().eq(SysUser::getUserName, userName));
  186. if (exist) {
  187. return UserConstants.NOT_UNIQUE;
  188. }
  189. return UserConstants.UNIQUE;
  190. }
  191. /**
  192. * 校验手机号码是否唯一
  193. *
  194. * @param user 用户信息
  195. * @return
  196. */
  197. @Override
  198. public String checkPhoneUnique(SysUser user) {
  199. boolean exist = baseMapper.exists(new LambdaQueryWrapper<SysUser>()
  200. .eq(SysUser::getPhonenumber, user.getPhonenumber())
  201. .ne(ObjectUtil.isNotNull(user.getUserId()), SysUser::getUserId, user.getUserId()));
  202. if (exist) {
  203. return UserConstants.NOT_UNIQUE;
  204. }
  205. return UserConstants.UNIQUE;
  206. }
  207. /**
  208. * 校验email是否唯一
  209. *
  210. * @param user 用户信息
  211. * @return
  212. */
  213. @Override
  214. public String checkEmailUnique(SysUser user) {
  215. boolean exist = baseMapper.exists(new LambdaQueryWrapper<SysUser>()
  216. .eq(SysUser::getEmail, user.getEmail())
  217. .ne(ObjectUtil.isNotNull(user.getUserId()), SysUser::getUserId, user.getUserId()));
  218. if (exist) {
  219. return UserConstants.NOT_UNIQUE;
  220. }
  221. return UserConstants.UNIQUE;
  222. }
  223. /**
  224. * 校验用户是否允许操作
  225. *
  226. * @param user 用户信息
  227. */
  228. @Override
  229. public void checkUserAllowed(SysUser user) {
  230. if (ObjectUtil.isNotNull(user.getUserId()) && user.isAdmin()) {
  231. throw new ServiceException("不允许操作超级管理员用户");
  232. }
  233. }
  234. /**
  235. * 校验用户是否有数据权限
  236. *
  237. * @param userId 用户id
  238. */
  239. @Override
  240. public void checkUserDataScope(Long userId) {
  241. if (!LoginHelper.isAdmin()) {
  242. SysUser user = new SysUser();
  243. user.setUserId(userId);
  244. List<SysUser> users = this.selectUserList(user);
  245. if (CollUtil.isEmpty(users)) {
  246. throw new ServiceException("没有权限访问用户数据!");
  247. }
  248. }
  249. }
  250. /**
  251. * 新增保存用户信息
  252. *
  253. * @param user 用户信息
  254. * @return 结果
  255. */
  256. @Override
  257. @Transactional(rollbackFor = Exception.class)
  258. public int insertUser(SysUser user) {
  259. // 新增用户信息
  260. int rows = baseMapper.insert(user);
  261. // 新增用户岗位关联
  262. insertUserPost(user);
  263. // 新增用户与角色管理
  264. insertUserRole(user);
  265. return rows;
  266. }
  267. /**
  268. * 注册用户信息
  269. *
  270. * @param user 用户信息
  271. * @return 结果
  272. */
  273. @Override
  274. public boolean registerUser(SysUser user) {
  275. user.setCreateBy(user.getUserName());
  276. user.setUpdateBy(user.getUserName());
  277. return baseMapper.insert(user) > 0;
  278. }
  279. /**
  280. * 修改保存用户信息
  281. *
  282. * @param user 用户信息
  283. * @return 结果
  284. */
  285. @Override
  286. @Transactional(rollbackFor = Exception.class)
  287. public int updateUser(SysUser user) {
  288. Long userId = user.getUserId();
  289. // 删除用户与角色关联
  290. userRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, userId));
  291. // 新增用户与角色管理
  292. insertUserRole(user);
  293. // 删除用户与岗位关联
  294. userPostMapper.delete(new LambdaQueryWrapper<SysUserPost>().eq(SysUserPost::getUserId, userId));
  295. // 新增用户与岗位管理
  296. insertUserPost(user);
  297. return baseMapper.updateById(user);
  298. }
  299. /**
  300. * 用户授权角色
  301. *
  302. * @param userId 用户ID
  303. * @param roleIds 角色组
  304. */
  305. @Override
  306. @Transactional(rollbackFor = Exception.class)
  307. public void insertUserAuth(Long userId, Long[] roleIds) {
  308. userRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>()
  309. .eq(SysUserRole::getUserId, userId));
  310. insertUserRole(userId, roleIds);
  311. }
  312. /**
  313. * 修改用户状态
  314. *
  315. * @param user 用户信息
  316. * @return 结果
  317. */
  318. @Override
  319. public int updateUserStatus(SysUser user) {
  320. return baseMapper.updateById(user);
  321. }
  322. /**
  323. * 修改用户基本信息
  324. *
  325. * @param user 用户信息
  326. * @return 结果
  327. */
  328. @Override
  329. public int updateUserProfile(SysUser user) {
  330. return baseMapper.updateById(user);
  331. }
  332. /**
  333. * 修改用户头像
  334. *
  335. * @param userName 用户名
  336. * @param avatar 头像地址
  337. * @return 结果
  338. */
  339. @Override
  340. public boolean updateUserAvatar(String userName, String avatar) {
  341. return baseMapper.update(null,
  342. new LambdaUpdateWrapper<SysUser>()
  343. .set(SysUser::getAvatar, avatar)
  344. .eq(SysUser::getUserName, userName)) > 0;
  345. }
  346. /**
  347. * 重置用户密码
  348. *
  349. * @param user 用户信息
  350. * @return 结果
  351. */
  352. @Override
  353. public int resetPwd(SysUser user) {
  354. return baseMapper.updateById(user);
  355. }
  356. /**
  357. * 重置用户密码
  358. *
  359. * @param userName 用户名
  360. * @param password 密码
  361. * @return 结果
  362. */
  363. @Override
  364. public int resetUserPwd(String userName, String password) {
  365. return baseMapper.update(null,
  366. new LambdaUpdateWrapper<SysUser>()
  367. .set(SysUser::getPassword, password)
  368. .eq(SysUser::getUserName, userName));
  369. }
  370. /**
  371. * 新增用户角色信息
  372. *
  373. * @param user 用户对象
  374. */
  375. public void insertUserRole(SysUser user) {
  376. Long[] roles = user.getRoleIds();
  377. if (ObjectUtil.isNotNull(roles)) {
  378. // 新增用户与角色管理
  379. List<SysUserRole> list = new ArrayList<SysUserRole>();
  380. for (Long roleId : roles) {
  381. SysUserRole ur = new SysUserRole();
  382. ur.setUserId(user.getUserId());
  383. ur.setRoleId(roleId);
  384. list.add(ur);
  385. }
  386. if (list.size() > 0) {
  387. userRoleMapper.insertBatch(list);
  388. }
  389. }
  390. }
  391. /**
  392. * 新增用户岗位信息
  393. *
  394. * @param user 用户对象
  395. */
  396. public void insertUserPost(SysUser user) {
  397. Long[] posts = user.getPostIds();
  398. if (ObjectUtil.isNotNull(posts)) {
  399. // 新增用户与岗位管理
  400. List<SysUserPost> list = new ArrayList<SysUserPost>();
  401. for (Long postId : posts) {
  402. SysUserPost up = new SysUserPost();
  403. up.setUserId(user.getUserId());
  404. up.setPostId(postId);
  405. list.add(up);
  406. }
  407. if (list.size() > 0) {
  408. userPostMapper.insertBatch(list);
  409. }
  410. }
  411. }
  412. /**
  413. * 新增用户角色信息
  414. *
  415. * @param userId 用户ID
  416. * @param roleIds 角色组
  417. */
  418. public void insertUserRole(Long userId, Long[] roleIds) {
  419. if (ObjectUtil.isNotNull(roleIds)) {
  420. // 新增用户与角色管理
  421. List<SysUserRole> list = new ArrayList<SysUserRole>();
  422. for (Long roleId : roleIds) {
  423. SysUserRole ur = new SysUserRole();
  424. ur.setUserId(userId);
  425. ur.setRoleId(roleId);
  426. list.add(ur);
  427. }
  428. if (list.size() > 0) {
  429. userRoleMapper.insertBatch(list);
  430. }
  431. }
  432. }
  433. /**
  434. * 通过用户ID删除用户
  435. *
  436. * @param userId 用户ID
  437. * @return 结果
  438. */
  439. @Override
  440. @Transactional(rollbackFor = Exception.class)
  441. public int deleteUserById(Long userId) {
  442. // 删除用户与角色关联
  443. userRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, userId));
  444. // 删除用户与岗位表
  445. userPostMapper.delete(new LambdaQueryWrapper<SysUserPost>().eq(SysUserPost::getUserId, userId));
  446. return baseMapper.deleteById(userId);
  447. }
  448. /**
  449. * 批量删除用户信息
  450. *
  451. * @param userIds 需要删除的用户ID
  452. * @return 结果
  453. */
  454. @Override
  455. @Transactional(rollbackFor = Exception.class)
  456. public int deleteUserByIds(Long[] userIds) {
  457. for (Long userId : userIds) {
  458. checkUserAllowed(new SysUser(userId));
  459. checkUserDataScope(userId);
  460. }
  461. List<Long> ids = Arrays.asList(userIds);
  462. // 删除用户与角色关联
  463. userRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>().in(SysUserRole::getUserId, ids));
  464. // 删除用户与岗位表
  465. userPostMapper.delete(new LambdaQueryWrapper<SysUserPost>().in(SysUserPost::getUserId, ids));
  466. return baseMapper.deleteBatchIds(ids);
  467. }
  468. }