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