From 1904cc522c211d7963eaf91af66598e6687b770f Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期一, 26 二月 2024 15:50:30 +0800 Subject: [PATCH] # --- zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java | 29 +++++++++++++++++++++++++---- 1 files changed, 25 insertions(+), 4 deletions(-) diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java index b7a9423..da19514 100644 --- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java +++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java @@ -20,7 +20,10 @@ import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; -import java.util.*; +import java.util.ArrayList; +import java.util.Date; +import java.util.List; +import java.util.Map; @RestController @RequestMapping("/api") @@ -38,7 +41,7 @@ public R page(@RequestBody Map<String, Object> map) { BaseParam baseParam = buildParam(map, BaseParam.class); PageParam<User, BaseParam> pageParam = new PageParam<>(baseParam, User.class); - return R.ok().add(userService.page(pageParam, pageParam.buildWrapper(true))); + return R.ok().add(userService.page(pageParam, pageParam.buildWrapper(true, wrapper -> wrapper.orderByAsc("create_time")))); } @PreAuthorize("hasAuthority('system:user:list')") @@ -64,6 +67,7 @@ @PreAuthorize("hasAuthority('system:user:save')") @OperationLog("娣诲姞鐢ㄦ埛") @PostMapping("/user/save") + @Transactional public R save(@RequestBody User user) { if (!Cools.isEmpty(user.getUsername()) && userService.count(new LambdaQueryWrapper<User>().eq(User::getUsername, user.getUsername())) > 0) { @@ -86,14 +90,23 @@ user.setUpdateTime(new Date()); if (!userService.save(user)) { - return R.error("娣诲姞澶辫触"); + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); } + + userRoleService.remove(new LambdaQueryWrapper<UserRole>().eq(UserRole::getUserId, user.getId())); + for (Long roleId : user.getRoleIds()) { + if (!userRoleService.save(new UserRole(user.getId(), roleId))) { + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); + } + } + return R.ok("娣诲姞鎴愬姛"); } @PreAuthorize("hasAuthority('system:user:update')") @OperationLog("淇敼鐢ㄦ埛") @PostMapping("/user/update") + @Transactional public R update(@RequestBody User user) { if (!Cools.isEmpty(user.getUsername()) && userService.count(new LambdaQueryWrapper<User>() .eq(User::getUsername, user.getUsername()) @@ -119,8 +132,16 @@ user.setUpdateTime(new Date()); if (!userService.updateById(user)) { - return R.error("淇敼澶辫触"); + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); } + + userRoleService.remove(new LambdaQueryWrapper<UserRole>().eq(UserRole::getUserId, user.getId())); + for (Long roleId : user.getRoleIds()) { + if (!userRoleService.save(new UserRole(user.getId(), roleId))) { + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); + } + } + return R.ok("淇敼鎴愬姛"); } -- Gitblit v1.9.1