SysUserController.java 28 KB

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