From b9efed08b3cf24f3f30b915b17557efd93195940 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期四, 22 二月 2024 13:54:23 +0800 Subject: [PATCH] # --- zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/UserController.java | 43 +++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 41 insertions(+), 2 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 2d39a81..b7a9423 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 @@ -4,15 +4,19 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.zy.asrs.framework.common.Cools; import com.zy.asrs.framework.common.R; +import com.zy.asrs.framework.exception.CoolException; import com.zy.asrs.wcs.common.annotation.OperationLog; import com.zy.asrs.wcs.common.domain.BaseParam; import com.zy.asrs.wcs.common.domain.KeyValVo; import com.zy.asrs.wcs.common.domain.PageParam; import com.zy.asrs.wcs.system.entity.User; +import com.zy.asrs.wcs.system.entity.UserRole; +import com.zy.asrs.wcs.system.service.UserRoleService; import com.zy.asrs.wcs.system.service.UserService; import com.zy.asrs.wcs.utils.ExcelUtil; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.transaction.annotation.Transactional; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; @@ -26,6 +30,8 @@ @Autowired private UserService userService; + @Autowired + private UserRoleService userRoleService; @PreAuthorize("hasAuthority('system:user:list')") @PostMapping("/user/page") @@ -59,6 +65,18 @@ @OperationLog("娣诲姞鐢ㄦ埛") @PostMapping("/user/save") public R save(@RequestBody User user) { + if (!Cools.isEmpty(user.getUsername()) + && userService.count(new LambdaQueryWrapper<User>().eq(User::getUsername, user.getUsername())) > 0) { + return R.error("鐧诲綍璐﹀彿宸插瓨鍦�"); + } + if (!Cools.isEmpty(user.getNickname()) + && userService.count(new LambdaQueryWrapper<User>().eq(User::getNickname, user.getNickname())) > 0) { + return R.error("鐢ㄦ埛鍚嶇О宸插瓨鍦�"); + } + if (!Cools.isEmpty(user.getPhone()) + && userService.count(new LambdaQueryWrapper<User>().eq(User::getPhone, user.getPhone())) > 0) { + return R.error("鎵嬫満鍙峰凡瀛樺湪"); + } user.setPassword(userService.encodePassword(COMMON_PASSWORD)); @@ -77,6 +95,21 @@ @OperationLog("淇敼鐢ㄦ埛") @PostMapping("/user/update") public R update(@RequestBody User user) { + if (!Cools.isEmpty(user.getUsername()) && userService.count(new LambdaQueryWrapper<User>() + .eq(User::getUsername, user.getUsername()) + .ne(User::getId, user.getId())) > 0) { + return R.error("鐧诲綍璐﹀彿宸插瓨鍦�"); + } + if (!Cools.isEmpty(user.getNickname()) && userService.count(new LambdaQueryWrapper<User>() + .eq(User::getNickname, user.getNickname()) + .ne(User::getId, user.getId())) > 0) { + return R.error("鐢ㄦ埛鍚嶇О宸插瓨鍦�"); + } + if (!Cools.isEmpty(user.getPhone()) && userService.count(new LambdaQueryWrapper<User>() + .eq(User::getPhone, user.getPhone()) + .ne(User::getId, user.getId())) > 0) { + return R.error("鎵嬫満鍙峰凡瀛樺湪"); + } if (!Cools.isEmpty(user.getPassword())) { user.setPassword(userService.encodePassword(user.getPassword())); @@ -94,9 +127,15 @@ @PreAuthorize("hasAuthority('system:user:remove')") @OperationLog("鍒犻櫎鐢ㄦ埛") @PostMapping("/user/remove/{ids}") + @Transactional public R remove(@PathVariable Long[] ids) { - if (!userService.removeByIds(Arrays.asList(ids))) { - return R.error("鍒犻櫎澶辫触"); + for (Long userId : ids) { + if (!userService.removeById(userId)) { + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); + } + if (!userRoleService.remove(new LambdaQueryWrapper<UserRole>().eq(UserRole::getUserId, userId))) { + throw new CoolException("鏈嶅姟鍣ㄥ唴閮ㄩ敊璇�"); + } } return R.ok("鍒犻櫎鎴愬姛"); } -- Gitblit v1.9.1