SysUserController.java 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787
  1. package com.xunmei.system.controller;
  2. import cn.hutool.core.util.ObjectUtil;
  3. import com.alibaba.excel.EasyExcel;
  4. import com.alibaba.excel.context.AnalysisContext;
  5. import com.alibaba.excel.event.AnalysisEventListener;
  6. import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
  7. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  8. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  9. import com.xunmei.common.core.domain.R;
  10. import com.xunmei.common.core.domain.message.domain.CoreAnnouncementNotificationToRole;
  11. import com.xunmei.common.core.domain.reminder.dto.CoreReminderUserQueryDto;
  12. import com.xunmei.common.core.utils.DateUtils;
  13. import com.xunmei.common.core.utils.StringUtils;
  14. import com.xunmei.common.core.vo.IdNameVo;
  15. import com.xunmei.common.core.web.controller.BaseController;
  16. import com.xunmei.common.core.web.domain.AjaxResult;
  17. import com.xunmei.common.core.web.page.TableDataInfo;
  18. import com.xunmei.common.log.annotation.Log;
  19. import com.xunmei.common.log.enums.BusinessType;
  20. import com.xunmei.common.security.annotation.InnerAuth;
  21. import com.xunmei.common.security.annotation.RequiresPermissions;
  22. import com.xunmei.common.security.utils.SecurityUtils;
  23. import com.xunmei.common.websocket.dto.WebSocketMessageDto;
  24. import com.xunmei.common.websocket.utils.WebSocketUtils;
  25. import com.xunmei.system.api.domain.SysOrg;
  26. import com.xunmei.system.api.domain.SysRole;
  27. import com.xunmei.system.api.domain.SysUser;
  28. import com.xunmei.system.api.domain.SysUserInformation;
  29. import com.xunmei.system.api.dto.SysPlanOrgDTO;
  30. import com.xunmei.system.api.model.LoginUser;
  31. import com.xunmei.system.domain.SysUserRole;
  32. import com.xunmei.system.domain.vo.SysMasterSubUserVO;
  33. import com.xunmei.system.domain.vo.SysUserListVo;
  34. import com.xunmei.system.domain.vo.UserRoleVo;
  35. import com.xunmei.system.dto.SysUserMapperDTO;
  36. import com.xunmei.system.mapper.SysOrgMapper;
  37. import com.xunmei.system.service.*;
  38. import com.xunmei.system.util.SecurityUserExport;
  39. import com.xunmei.system.util.UserExport;
  40. import com.xunmei.system.util.UserImport;
  41. import io.swagger.annotations.ApiOperation;
  42. import org.apache.commons.lang3.ArrayUtils;
  43. import org.springframework.beans.BeanUtils;
  44. import org.springframework.beans.factory.annotation.Autowired;
  45. import org.springframework.transaction.annotation.Transactional;
  46. import org.springframework.validation.annotation.Validated;
  47. import org.springframework.web.bind.annotation.*;
  48. import org.springframework.web.multipart.MultipartFile;
  49. import javax.servlet.http.HttpServletResponse;
  50. import java.io.File;
  51. import java.io.IOException;
  52. import java.net.URLEncoder;
  53. import java.util.*;
  54. import java.util.concurrent.atomic.AtomicInteger;
  55. import java.util.stream.Collectors;
  56. /**
  57. * 用户信息
  58. *
  59. * @author xunmei
  60. */
  61. @RestController
  62. @RequestMapping("/user")
  63. public class SysUserController extends BaseController {
  64. @Autowired
  65. private ISysUserService userService;
  66. @Autowired
  67. private ISysRoleService roleService;
  68. @Autowired
  69. private ISysPermissionService permissionService;
  70. @Autowired
  71. private ISysConfigService configService;
  72. @Autowired
  73. private ISysUserRoleService sysUserRoleService;
  74. @Autowired
  75. private ISysUserInformationService sysUserInformationService;
  76. @Autowired
  77. private SysOrgMapper orgMapper;
  78. /**
  79. * 获取用户列表
  80. */
  81. @RequiresPermissions("system:user:list")
  82. @GetMapping("/list")
  83. public AjaxResult list(SysUser user) {
  84. AjaxResult ajax = AjaxResult.success();
  85. com.xunmei.common.core.web.page.TableDataInfo<com.xunmei.system.api.domain.SysUser> sysUserTableDataInfo = userService.selectUserList(user);
  86. ajax.put("data", sysUserTableDataInfo);
  87. boolean admin = false;
  88. SysUser sysUser = userService.selectUserById(SecurityUtils.getUserId());
  89. if (null != sysUser && "Y".equals(sysUser.getIsManage())) {
  90. admin = true;
  91. }
  92. ajax.put("check", admin);
  93. String orgName = "福建农信";
  94. if (null != user.getOrgId()) {
  95. SysOrg sysOrg = orgMapper.selectSysOrgById(user.getOrgId());
  96. orgName = sysOrg.getShortName();
  97. }
  98. ajax.put("orgName", orgName);
  99. return ajax;
  100. }
  101. /**
  102. * 获取用户列表
  103. */
  104. @GetMapping("/userselector/list")
  105. public AjaxResult userSelectorList(SysUser user) {
  106. TableDataInfo<SysUser> sysUserTableDataInfo = userService.userSelectorList(user);
  107. return success(sysUserTableDataInfo);
  108. }
  109. @RequiresPermissions("system:user:list")
  110. @GetMapping("/dictionlist")
  111. public R<List<SysUser>> list2(SysUser user) {
  112. try {
  113. if (ObjectUtil.isEmpty(user.getOrgId())) {
  114. user.setOrgId(SecurityUtils.getLoginUser().getOrgId());
  115. }
  116. return R.ok(userService.selectSyncUserList(user));
  117. } catch (Exception e) {
  118. return R.fail(e.getMessage());
  119. }
  120. }
  121. // @RequiresPermissions("system:user:list")
  122. @GetMapping("/userListData")
  123. public R<List<SysUser>> userListData(SysUser user) {
  124. if (ObjectUtil.isEmpty(user.getOrgId())) {
  125. user.setOrgId(SecurityUtils.getLoginUser().getOrgId());
  126. }
  127. return R.ok(userService.selectUserListData(user));
  128. }
  129. /**
  130. * 获取当前用户信息
  131. */
  132. @InnerAuth
  133. @GetMapping("/info/{username}")
  134. public R<LoginUser> info(@PathVariable("username") String username) {
  135. SysUser sysUser = userService.selectUserByUserName(username);
  136. if (StringUtils.isNull(sysUser)) {
  137. return R.fail("用户名或密码错误");
  138. }
  139. //获取当前用户的角色数据。
  140. List<SysRole> sysRoles = roleService.selectRolesByUserId(sysUser.getId());
  141. sysUser.setRoles(sysRoles);
  142. // 角色集合
  143. Set<String> roles = permissionService.getRolePermission(sysUser);
  144. // 权限集合
  145. Set<String> permissions = permissionService.getMenuPermission(sysUser);
  146. LoginUser sysUserVo = new LoginUser();
  147. sysUserVo.setOrgId(sysUser.getOrgId());
  148. sysUserVo.setName(sysUser.getName());
  149. sysUserVo.setSource(sysUser.getSource());
  150. sysUserVo.setSysUser(sysUser);
  151. sysUserVo.setRoles(roles);
  152. sysUserVo.setPermissions(permissions);
  153. return R.ok(sysUserVo);
  154. }
  155. /**
  156. * 注册用户信息
  157. */
  158. @InnerAuth
  159. @PostMapping("/register")
  160. public R<Boolean> register(@RequestBody SysUser sysUser) {
  161. String username = sysUser.getUsername();
  162. if (!("true".equals(configService.selectConfigByKey("sys.account.registerUser")))) {
  163. return R.fail("当前系统没有开启注册功能!");
  164. }
  165. if (!userService.checkUserNameUnique(sysUser)) {
  166. return R.fail("保存用户'" + username + "'失败,注册账号已存在");
  167. }
  168. return R.ok(userService.registerUser(sysUser));
  169. }
  170. /**
  171. * 获取用户信息
  172. *
  173. * @return 用户信息
  174. */
  175. @GetMapping("getInfo")
  176. public AjaxResult getInfo() {
  177. SysUser user = userService.selectUserById(SecurityUtils.getUserId());
  178. SysOrg org = orgMapper.selectOne(new LambdaQueryWrapper<SysOrg>()
  179. .eq(SysOrg::getId, user.getOrgId())
  180. .select(SysOrg::getType, SysOrg::getShortName, SysOrg::getPath));
  181. if (ObjectUtil.isNotEmpty(org)) {
  182. user.setOrgType(org.getType());
  183. user.setOrgShortName(org.getShortName());
  184. user.setOrgPath(org.getPath());
  185. }
  186. // 角色集合
  187. Set<String> roles = permissionService.getRolePermission(user);
  188. // 权限集合
  189. Set<String> permissions = permissionService.getMenuPermission(user);
  190. List<UserRoleVo> userRoleVos = sysUserRoleService.selectUserRoleVoList(user.getId());
  191. AjaxResult ajax = AjaxResult.success();
  192. ajax.put("user", user);
  193. ajax.put("roles", roles);
  194. //用户角色信息
  195. ajax.put("roleList", userRoleVos);
  196. ajax.put("permissions", permissions);
  197. // 获取当前登录账号是否存在 关联子账号
  198. List<SysMasterSubUserVO> subUser = new ArrayList<>();
  199. if (ObjectUtil.isNotEmpty(SecurityUtils.getMasterUserId())) {
  200. subUser = userService.selectSwitchUserList(SecurityUtils.getMasterUserId(), user.getId());
  201. } else {
  202. subUser = userService.selectSwitchUserList(user.getId(), null);
  203. }
  204. ajax.put("subUserList", subUser);
  205. ajax.put("masterUserId", SecurityUtils.getMasterUserId());
  206. ajax.put("masterUserName", SecurityUtils.getMasterUsername());
  207. return ajax;
  208. }
  209. /**
  210. * 用户关联子用户
  211. */
  212. @RequiresPermissions("system:user:mapperUser")
  213. @Log(title = "关联子用户", businessType = BusinessType.OTHER)
  214. @PutMapping("/mapperuser")
  215. public AjaxResult mapperUser(@RequestBody SysUserMapperDTO userMapperDTO) {
  216. userService.insertUserMapper(userMapperDTO.getMasterUserId(), userMapperDTO.getMapperUserIds().toArray(new Long[0]));
  217. return success();
  218. }
  219. /**
  220. * 用户关联子用户
  221. */
  222. // @RequiresPermissions("system:user:mapperUser")
  223. @Log(title = "关联子用户", businessType = BusinessType.OTHER)
  224. @GetMapping("/mapperlist/{userId}")
  225. public R<List<SysMasterSubUserVO>> mapperUser(@PathVariable(value = "userId") Long userId) {
  226. return R.ok(userService.selectSwitchUserList(userId, null));
  227. }
  228. /**
  229. * 根据用户编号获取详细信息
  230. */
  231. @RequiresPermissions("system:user:query")
  232. @GetMapping(value = {"/", "/{userId}"})
  233. public AjaxResult getInfo(@PathVariable(value = "userId", required = false) Long userId) {
  234. userService.checkUserDataScope(userId);
  235. AjaxResult ajax = AjaxResult.success();
  236. List<SysRole> roles = roleService.selectRoleAll();
  237. ajax.put("roles", roles);
  238. ajax.put("posts", null);
  239. if (StringUtils.isNotNull(userId)) {
  240. SysUserInformation informationByUserId = sysUserInformationService.getInformationByUserId(userId);
  241. if (null == informationByUserId) {
  242. informationByUserId = new SysUserInformation();
  243. }
  244. SysUser sysUser = userService.selectUserById(userId);
  245. // sysUser.setOrgId(sysUser.getOriginalOrgId());
  246. List<SysUserRole> userRoles = sysUserRoleService.list(new QueryWrapper<SysUserRole>().eq("user_id", userId));
  247. sysUser.setRoles(roles);
  248. String sex = "-";
  249. String status = "禁用";
  250. if (null != sysUser) {
  251. if (null != sysUser.getGender()) {
  252. if (sysUser.getGender().equals("1")) {
  253. sex = "女";
  254. } else {
  255. sex = "男";
  256. }
  257. }
  258. if (sysUser.getIsLock().equals("0")) {
  259. status = "启用";
  260. }
  261. }
  262. ajax.put(AjaxResult.DATA_TAG, sysUser);
  263. ajax.put("information", informationByUserId);
  264. ajax.put("postIds", null);
  265. ajax.put("roleIds", userRoles.stream().map(SysUserRole::getRoleId).collect(Collectors.toList()));
  266. ajax.put("roleName", roleService.getRoleName(userId));
  267. ajax.put("sex", sex);
  268. ajax.put("status", status);
  269. ajax.put("source", sysUser.getSource());
  270. }
  271. return ajax;
  272. }
  273. /**
  274. * 新增用户
  275. */
  276. @RequiresPermissions("system:user:add")
  277. @Log(title = "用户管理", businessType = BusinessType.INSERT)
  278. @PostMapping
  279. @Transactional(rollbackFor = Exception.class)
  280. public AjaxResult add(@Validated @RequestBody SysUser user) {
  281. if (!userService.checkUserNameUnique(user)) {
  282. return error("新增用户'" + user.getUsername() + "'失败,账号已存在");
  283. } else if (StringUtils.isNotEmpty(user.getPhone()) && !userService.checkPhoneUnique(user)) {
  284. return error("新增用户'" + user.getUsername() + "'失败,手机号码已存在");
  285. }
  286. user.setCreateBy(SecurityUtils.getUsername());
  287. user.setSource(0);
  288. user.setCreateTime(new Date());
  289. final String salt = SecurityUtils.salt();
  290. user.setSalt(salt);
  291. user.setPassword(SecurityUtils.encryptPassword(user.getPassword(), salt));
  292. //user.setOriginalOrgId(user.getOrgId());
  293. userService.insertUser(user);
  294. return AjaxResult.success();
  295. }
  296. @GetMapping("/selectrolesByOrgId")
  297. public AjaxResult selectrolesByOrgId(Long orgId) {
  298. return AjaxResult.success(userService.selectrolesByOrgId(orgId));
  299. }
  300. /**
  301. * 修改用户
  302. */
  303. @RequiresPermissions("system:user:edit")
  304. @Log(title = "用户管理", businessType = BusinessType.UPDATE)
  305. @PutMapping
  306. public AjaxResult edit(@Validated @RequestBody SysUser user) {
  307. userService.checkUserAllowed(user);
  308. userService.checkUserDataScope(user.getId());
  309. if (!userService.checkUserNameUnique(user)) {
  310. return error("修改用户'" + user.getUsername() + "'失败,账号已存在");
  311. } else if (StringUtils.isNotEmpty(user.getPhone()) && !userService.checkPhoneUnique(user)) {
  312. return error("修改用户'" + user.getUsername() + "'失败,手机号码已存在");
  313. }
  314. if (user.getRoleIds() != null && user.getRoleIds().length > 0) {
  315. sysUserRoleService.remove(new QueryWrapper<SysUserRole>().eq("user_id", user.getId()));
  316. for (Long roleId :
  317. user.getRoleIds()) {
  318. SysUserRole sysUserRole = new SysUserRole();
  319. sysUserRole.setUserId(user.getId());
  320. sysUserRole.setRoleId(roleId);
  321. sysUserRole.setCreateTime(new Date());
  322. sysUserRoleService.insertSysUserRole(sysUserRole);
  323. }
  324. }
  325. // user.setOriginalOrgId(user.getOrgId());
  326. user.setUpdateBy(SecurityUtils.getUsername());
  327. return toAjax(userService.updateUser(user));
  328. }
  329. /**
  330. * 删除用户
  331. */
  332. @RequiresPermissions("system:user:remove")
  333. @Log(title = "用户管理", businessType = BusinessType.DELETE)
  334. @DeleteMapping("/{userIds}")
  335. public AjaxResult remove(@PathVariable Long[] userIds) {
  336. if (ArrayUtils.contains(userIds, SecurityUtils.getUserId())) {
  337. return error("当前用户不能删除");
  338. }
  339. return toAjax(userService.deleteUserByIds(userIds));
  340. }
  341. /**
  342. * 重置密码
  343. */
  344. @RequiresPermissions("system:user:edit")
  345. @Log(title = "用户管理", businessType = BusinessType.UPDATE)
  346. @PutMapping("/resetPwd")
  347. public AjaxResult resetPwd(@RequestBody SysUser user) {
  348. userService.checkUserAllowed(user);
  349. userService.checkUserDataScope(user.getId());
  350. final String salt = SecurityUtils.salt();
  351. user.setSalt(salt);
  352. user.setPassword(SecurityUtils.encryptPassword(user.getPassword(), salt));
  353. user.setUpdateBy(SecurityUtils.getUsername());
  354. return toAjax(userService.resetPwd(user));
  355. }
  356. /**
  357. * 状态修改
  358. */
  359. @RequiresPermissions("system:user:edit")
  360. @Log(title = "用户管理", businessType = BusinessType.UPDATE)
  361. @PutMapping("/changeStatus")
  362. public AjaxResult changeStatus(@RequestBody SysUser user) {
  363. userService.checkUserAllowed(user);
  364. userService.checkUserDataScope(user.getId());
  365. user.setUpdateBy(SecurityUtils.getUsername());
  366. return toAjax(userService.updateUserStatus(user));
  367. }
  368. /**
  369. * 根据用户编号获取授权角色
  370. */
  371. @RequiresPermissions("system:user:query")
  372. @GetMapping("/authRole/{userId}")
  373. public AjaxResult authRole(@PathVariable("userId") Long userId) {
  374. AjaxResult ajax = AjaxResult.success();
  375. SysUser user = userService.selectUserById(userId);
  376. List<SysRole> roles = roleService.selectRolesByUserId(userId);
  377. List<SysUserRole> userRoles = sysUserRoleService.selectSysUserRoleListByUserId(userId);
  378. ajax.put("user", user);
  379. ajax.put("roles", roles);
  380. ajax.put("userRoles", userRoles);
  381. return ajax;
  382. }
  383. /**
  384. * 用户授权角色
  385. */
  386. @RequiresPermissions("system:user:edit")
  387. @Log(title = "用户管理", businessType = BusinessType.GRANT)
  388. @PutMapping("/authRole")
  389. public AjaxResult insertAuthRole(Long userId, Long[] roleIds) {
  390. userService.checkUserDataScope(userId);
  391. userService.insertUserAuth(userId, roleIds);
  392. return success();
  393. }
  394. /**
  395. * 获取用户列表
  396. */
  397. @InnerAuth
  398. @PostMapping("/sync/list")
  399. public R<List<SysUser>> syncUserList(@RequestBody SysUser user) {
  400. try {
  401. return R.ok(userService.selectSyncUserList(user));
  402. } catch (Exception e) {
  403. return R.fail(e.getMessage());
  404. }
  405. }
  406. @InnerAuth
  407. @GetMapping("/getUserById")
  408. public SysUser getUserById(Long id) {
  409. return userService.selectUserById(id);
  410. }
  411. /**
  412. * 查询机构列表
  413. */
  414. @ApiOperation(value = "批量保存同步用户数据")
  415. @InnerAuth
  416. @PostMapping("/sync/batch")
  417. public R<Boolean> batchSaveSyncSysOrgs(@RequestBody List<SysUser> sysUserList) {
  418. userService.batchSaveOrUpdate(sysUserList);
  419. return R.ok(true);
  420. }
  421. /**
  422. * 查询机构列表
  423. */
  424. @ApiOperation(value = "记录用户最后一次登录")
  425. @InnerAuth
  426. @PostMapping("/userLoginInfo")
  427. public R<Boolean> userLoginInfo(Long id, Date loginTime, String loginIp) {
  428. SysUser sysUser = userService.selectUserById(id);
  429. sysUser.setLastIp(loginIp);
  430. if (null == loginTime) {
  431. loginTime = DateUtils.getNowDate();
  432. }
  433. sysUser.setLastTime(loginTime);
  434. userService.saveOrUpdate(sysUser);
  435. return R.ok(true);
  436. }
  437. @ApiOperation(value = "根据机构id查询用户")
  438. @GetMapping("/getByOrgId/{orgId}")
  439. public R<List<SysUserListVo>> getByOrgId(@PathVariable Long orgId) {
  440. List<SysUserListVo> sysUsers = userService.selectUserByOrgId(orgId);
  441. return R.ok(sysUsers);
  442. }
  443. @InnerAuth
  444. @PostMapping("/allUserList")
  445. public List<SysUser> allUserList() {
  446. return userService.list();
  447. }
  448. @ApiOperation(value = "通过角色获取用户")
  449. @InnerAuth
  450. @PostMapping("/userListByRole")
  451. public List<SysUser> userLoginInfo(@RequestBody List<CoreAnnouncementNotificationToRole> notificationToRoles) {
  452. return userService.selectUserListByRole(notificationToRoles);
  453. }
  454. @GetMapping("/selectAllUser")
  455. public AjaxResult selectAllUser(@PathVariable(value = "userName", required = false) String userName) {
  456. return success(userService.selectAllUser(userName));
  457. }
  458. @ApiOperation(value = "通过角色和机构获取用户")
  459. @InnerAuth
  460. @PostMapping("/userListByRoleAndOrg")
  461. public List<SysUser> userListByRoleAndOrg(@RequestBody SysPlanOrgDTO sysRoleOrgDTO) {
  462. return userService.selectUserListByRoleAndOrg(sysRoleOrgDTO);
  463. }
  464. /**
  465. * 获取用户的角色id及名称
  466. *
  467. * @param
  468. * @return
  469. */
  470. @ApiOperation(value = "获取用户的角色id及名称")
  471. @InnerAuth
  472. @GetMapping("/getroleinfoofuser")
  473. public List<IdNameVo> getRoleInfoOfUser(@RequestParam Long userId) {
  474. List<UserRoleVo> vos = sysUserRoleService.selectUserRoleVoList(userId);
  475. return vos.stream().map(v -> {
  476. IdNameVo vo = new IdNameVo();
  477. vo.setId(v.getRoleId());
  478. vo.setName(v.getRoleName());
  479. return vo;
  480. }).collect(Collectors.toList());
  481. }
  482. /**
  483. * 导出用户数据
  484. */
  485. @RequiresPermissions("system:user:export")
  486. @PostMapping("/export")
  487. public void export(SysUser user, HttpServletResponse response) {
  488. List<UserExport> list = null;
  489. Set<String> excludeColumnFiledNames = new HashSet<String>();
  490. //如果导出模板
  491. if (user.getTemplate() != null && user.getTemplate()) {
  492. excludeColumnFiledNames.add("lastIp");
  493. excludeColumnFiledNames.add("lastTime");
  494. list = new ArrayList<>();
  495. } else {
  496. list = userService.down(user);
  497. if (ObjectUtil.isEmpty(list)) {
  498. throw new RuntimeException("导出数据为空!");
  499. }
  500. AtomicInteger xh = new AtomicInteger();
  501. xh.getAndIncrement();
  502. list.forEach(e -> {
  503. if (e.getIsManage() == null) {
  504. e.setIsManage("N");
  505. }
  506. e.setXh(String.valueOf(xh.getAndIncrement()));
  507. });
  508. if (list.size() > 10000) {
  509. throw new RuntimeException("导出数据量过大(单次导出限量10000条数据),请填写条件分批导出");
  510. }
  511. }
  512. try {
  513. // 设置响应头
  514. response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("用户数据", "utf-8"));
  515. response.setContentType("application/octet-stream;charset=UTF-8");
  516. response.setCharacterEncoding("utf-8");
  517. // 数据导出
  518. EasyExcel.write(response.getOutputStream(), UserExport.class).excludeColumnFiledNames(excludeColumnFiledNames)
  519. .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).sheet("用户数据").doWrite(list);
  520. } catch (Exception e) {
  521. // 重置response
  522. response.reset();
  523. response.setContentType("application/json");
  524. response.setCharacterEncoding("utf-8");
  525. }
  526. }
  527. @RequiresPermissions("system:user:exportSecurity")
  528. @PostMapping("/exportSecurity")
  529. public void exportSecurity(SysUser user, HttpServletResponse response) {
  530. List<SecurityUserExport> securityUserExports = userService.downSecurityUser();
  531. if (ObjectUtil.isEmpty(securityUserExports)) {
  532. throw new RuntimeException("导出数据为空!");
  533. }
  534. AtomicInteger xh = new AtomicInteger();
  535. xh.getAndIncrement();
  536. securityUserExports.forEach(e -> {
  537. e.setXh(String.valueOf(xh.getAndIncrement()));
  538. });
  539. if (securityUserExports.size() > 10000) {
  540. throw new RuntimeException("导出数据量过大(单次导出限量10000条数据),请填写条件分批导出");
  541. }
  542. try {
  543. // 设置响应头
  544. response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("用户数据", "utf-8"));
  545. response.setContentType("application/octet-stream;charset=UTF-8");
  546. response.setCharacterEncoding("utf-8");
  547. // 数据导出
  548. EasyExcel.write(response.getOutputStream(), SecurityUserExport.class)
  549. .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).sheet("用户数据").doWrite(securityUserExports);
  550. } catch (Exception e) {
  551. // 重置response
  552. response.reset();
  553. response.setContentType("application/json");
  554. response.setCharacterEncoding("utf-8");
  555. }
  556. }
  557. /**
  558. * 导出管理人员数据
  559. */
  560. @RequiresPermissions("system:user:exportManager")
  561. @PostMapping("/exportManager")
  562. public void exportManager(SysUser user, HttpServletResponse response) {
  563. List<UserExport> list = userService.down(user);
  564. if (ObjectUtil.isEmpty(list)) {
  565. throw new RuntimeException("导出数据为空!");
  566. }
  567. list = list.stream().filter(e -> "Y".equals(e.getIsManage())).collect(Collectors.toList());
  568. AtomicInteger xh = new AtomicInteger();
  569. xh.getAndIncrement();
  570. list.forEach(e -> {
  571. if (e.getIsManage() == null) {
  572. e.setIsManage("N");
  573. }
  574. e.setXh(String.valueOf(xh.getAndIncrement()));
  575. });
  576. if (list.size() > 10000) {
  577. throw new RuntimeException("导出数据量过大(单次导出限量10000条数据),请填写条件分批导出");
  578. }
  579. try {
  580. // 设置响应头
  581. response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("用户数据", "utf-8"));
  582. response.setContentType("application/octet-stream;charset=UTF-8");
  583. response.setCharacterEncoding("utf-8");
  584. // 数据导出
  585. EasyExcel.write(response.getOutputStream(), UserExport.class)
  586. .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).sheet("用户数据").doWrite(list);
  587. } catch (Exception e) {
  588. // 重置response
  589. response.reset();
  590. response.setContentType("application/json");
  591. response.setCharacterEncoding("utf-8");
  592. }
  593. }
  594. @ApiOperation("检查用户的按钮权限")
  595. @InnerAuth
  596. @PostMapping("/checkuserpermission")
  597. public List<Long> checkUserPermission(@RequestParam List<Long> menuIds, @RequestParam Long userId) {
  598. return userService.checkUserPermission(menuIds, userId);
  599. }
  600. @InnerAuth
  601. @GetMapping("/selectUserRoleMenu")
  602. List<Long> selectUserRoleMenu(Long userId) {
  603. return userService.selectUserRoleMenu(userId);
  604. }
  605. /**
  606. * 导入
  607. */
  608. @RequiresPermissions("system:user:import")
  609. @ResponseBody
  610. @PostMapping("/importData")
  611. @Transactional(rollbackFor = {RuntimeException.class, Exception.class})
  612. public AjaxResult importData(MultipartFile file, boolean updateSupport) {
  613. List<UserImport> list = new ArrayList<>();
  614. // 读取excel
  615. EasyExcel.read(multipartFileToFile(file), UserImport.class, new AnalysisEventListener<UserImport>() {
  616. // 每解析一行数据,该方法会被调用一次
  617. @Override
  618. public void invoke(UserImport user, AnalysisContext analysisContext) {
  619. // System.out.println("解析数据为:" + user.toString());
  620. if (user.getName() != null) {
  621. list.add(user);
  622. }
  623. }
  624. // 全部解析完成被调用
  625. @Override
  626. public void doAfterAllAnalysed(AnalysisContext analysisContext) {
  627. List<String> names = list.stream().map(UserImport::getName).collect(Collectors.toList());
  628. List<String> exist = userService.selectCountByNames(names);
  629. if (exist.size() > 0) {
  630. throw new RuntimeException("用户" + exist + "已存在");
  631. }
  632. // 将解析的数据保存到数据库
  633. for (UserImport u : list) {
  634. //构建基础user对象
  635. SysUser user = new SysUser();
  636. BeanUtils.copyProperties(u, user);
  637. user.setDeleted(0);
  638. user.setIsLock("正常".equals(u.getIsLock()) ? "0" : "1");
  639. user.setGender("男".equals(u.getGender()) ? "0" : "1");
  640. user.setIsManage("是".equals(u.getIsManage()) ? "Y" : "N");
  641. SysOrg org = orgMapper.selectOne(new LambdaQueryWrapper<SysOrg>().eq(SysOrg::getShortName, u.getOrgName()).eq(SysOrg::getDeleted, 0));
  642. if (org == null) {
  643. throw new RuntimeException("机构" + u.getOrgName() + "不存在");
  644. }
  645. user.setOrgName(org.getName());
  646. user.setOrgPath(org.getPath());
  647. user.setOrgId(org.getId());
  648. user.setOrgType(org.getType());
  649. user.setCreateBy(SecurityUtils.getUsername());
  650. user.setSource(0);
  651. user.setCreateTime(new Date());
  652. //isp的加密方式
  653. final String salt = SecurityUtils.salt();
  654. user.setSalt(salt);
  655. user.setPassword(SecurityUtils.encryptPassword("Admin1234", salt));
  656. user.setOriginalOrgId(user.getOrgId());
  657. userService.insertUser(user);
  658. //构建用户角色关联关系
  659. String[] roleNames = user.getRoleNames().split(",");
  660. List<SysRole> roles = roleService.list(new LambdaQueryWrapper<SysRole>().in(SysRole::getRoleName, roleNames));
  661. if (roles.size() != roleNames.length) {
  662. throw new RuntimeException("角色" + Arrays.toString(roleNames) + "存在异常");
  663. }
  664. for (SysRole r :
  665. roles) {
  666. SysUserRole sysUserRole = new SysUserRole();
  667. sysUserRole.setUserId(user.getId());
  668. sysUserRole.setRoleId(r.getId());
  669. sysUserRoleService.insertSysUserRole(sysUserRole);
  670. }
  671. }
  672. }
  673. }).sheet().doRead();
  674. return success("导入成功!");
  675. }
  676. /***
  677. * 功能描述:
  678. * MultipartFile 转 File
  679. * @return: java.io.File
  680. */
  681. public static File multipartFileToFile(MultipartFile multipartFile) {
  682. // 获取文件名
  683. String fileName = multipartFile.getOriginalFilename();
  684. // 获取文件后缀
  685. assert fileName != null;
  686. String suffix = fileName.substring(fileName.lastIndexOf("."));
  687. // 若需要防止临时文件重复,需要在文件名后加上UUID
  688. try {
  689. File file = File.createTempFile(fileName.substring(0, fileName.lastIndexOf(".")) + UUID.randomUUID(), suffix);
  690. multipartFile.transferTo(file);
  691. return file;
  692. } catch (IOException e) {
  693. e.printStackTrace();
  694. }
  695. return null;
  696. }
  697. @ApiOperation(value = "通过角色和机构获取用户")
  698. @InnerAuth
  699. @PostMapping("/findReminderUser")
  700. public List<SysUser> findReminderUser(@RequestBody CoreReminderUserQueryDto userQueryDto) {
  701. return userService.findReminderUser(userQueryDto);
  702. }
  703. /**
  704. * 修改用户签名鹰印章
  705. */
  706. //@Log(title = "用户管理", businessType = BusinessType.UPDATE)
  707. @PostMapping("/updateSignatureSeal")
  708. public AjaxResult updateSignatureSeal(@RequestBody SysUser user) {
  709. if (null == user || StringUtils.isEmpty(user.getSignatureSeal())){
  710. return error("签名印章不能为空!");
  711. }
  712. SysUser updateUser = new SysUser();
  713. updateUser.setId(SecurityUtils.getUserId());
  714. updateUser.setSignatureSeal(user.getSignatureSeal());
  715. return toAjax(userService.updateById(updateUser));
  716. }
  717. @RequiresPermissions("system:user:import")
  718. @PostMapping("/importRole")
  719. @Transactional(rollbackFor = {RuntimeException.class, Exception.class})
  720. public void importRole(MultipartFile file, HttpServletResponse response) {
  721. userService.importRole(file, response);
  722. }
  723. }