From f9129efee277a67a2dcc1c87e01d733dd122765a Mon Sep 17 00:00:00 2001
From: qinjie <463333974@qq.com>
Date: Thu, 21 Nov 2024 00:51:30 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=94=A8=E6=88=B7=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/isu/gaswellwatch/dto/UserDTO.java          | 2 ++
 .../com/isu/gaswellwatch/service/impl/UserServiceImpl.java   | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java b/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java
index 1d7db85..3cb23f0 100644
--- a/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java
+++ b/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java
@@ -29,5 +29,7 @@ public class UserDTO {
     private String isEnable;
     @NotNull(message = "角色不能为空")
     private List<RoleVO> roles;
+    //部门ID
+    private Long departmentId;
 
 }
diff --git a/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java b/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java
index e54003b..8c29fff 100644
--- a/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java
@@ -132,6 +132,11 @@ public class UserServiceImpl extends ServiceImpl<UserDao, User> implements UserS
         save(user);
         //关联新增
         addRelation(userDTO.getRoles(), user);
+
+        if(userDTO.getDepartmentId()!=null){
+            userDepartmentService.remove(new LambdaQueryWrapper<UserDepartment>().eq(UserDepartment::getUserId, user.getId()));
+            addDepartment(userDTO.getDepartmentId(), user);
+        }
     }
 
     @Override