SysUserController.java 26 KB

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