From 2f640b1038fa331954f78ed1f4317212cf5bb34d Mon Sep 17 00:00:00 2001
From: chenlu <1320612696@qq.com>
Date: 星期五, 07 六月 2024 16:56:06 +0800
Subject: [PATCH] Merge branch 'master' of http://10.153.19.25:10101/r/ERP_override

---
 north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java |   41 +++++++++++++++++++++++++++++++++++++++++
 1 files changed, 41 insertions(+), 0 deletions(-)

diff --git a/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java
new file mode 100644
index 0000000..f968e5b
--- /dev/null
+++ b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java
@@ -0,0 +1,41 @@
+package com.example.erp.service.userInfo;
+
+import com.baomidou.dynamic.datasource.annotation.DS;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.example.erp.entity.userInfo.UserRole;
+import com.example.erp.mapper.userInfo.UserRoleMapper;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+@Service
+@DS("user_info")
+public class UserRoleService {
+    private final UserRoleMapper userRoleMapper;
+
+    public UserRoleService(UserRoleMapper userRoleMapper) {
+        this.userRoleMapper = userRoleMapper;
+    }
+
+    public  Boolean updateUserRole(Map<String, Object> params) {
+        Integer userId = (Integer) params.get("userId");
+        List<Integer> list = (List<Integer>) params.get("roles");
+        List<UserRole> userRoleList = new ArrayList<>();
+        if(!list.isEmpty()) {
+            for (Integer roleId : list) {
+                UserRole userRole = new UserRole();
+                userRole.setUserId(userId);
+                userRole.setRoleId(roleId);
+                userRoleList.add(userRole);
+            }
+        }
+        userRoleMapper.delete(new QueryWrapper<UserRole>().eq("user_id", userId));
+        userRoleList.forEach(userRole -> {
+            userRoleMapper.insert(userRole);
+        });
+        return true;
+    }
+}

--
Gitblit v1.8.0