From 3392e903116c7a754298618ce7b75cfd9c0b2a41 Mon Sep 17 00:00:00 2001
From: linzhijie <19970921lzj>
Date: 星期一, 12 七月 2021 11:43:32 +0800
Subject: [PATCH] 多角色

---
 src/main/java/com/ots/project/system/role/controller/RoleController.java |    2 ++
 src/main/java/com/ots/project/system/role/service/RoleServiceImpl.java   |   20 +++++++++++++-------
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/ots/project/system/role/controller/RoleController.java b/src/main/java/com/ots/project/system/role/controller/RoleController.java
index 4c14095..0da24fc 100644
--- a/src/main/java/com/ots/project/system/role/controller/RoleController.java
+++ b/src/main/java/com/ots/project/system/role/controller/RoleController.java
@@ -182,6 +182,7 @@
     @ResponseBody
     public TableDataInfo unallocatedList(User user) {
         startPage();
+        //鏌ヨ鍙垎閰嶇敤鎴�
         List<User> list = userService.selectUnallocatedList(user);
         return getDataTable(list);
     }
@@ -190,6 +191,7 @@
     @PostMapping("/authUser/selectAll")
     @ResponseBody
     public AjaxResult selectAuthUserAll(Long roleId, String userIds) {
+        //鍒嗛厤鐢ㄦ埛
         return toAjax(roleService.insertAuthUsers(roleId, userIds));
     }
 }
diff --git a/src/main/java/com/ots/project/system/role/service/RoleServiceImpl.java b/src/main/java/com/ots/project/system/role/service/RoleServiceImpl.java
index ace982a..a0e4174 100644
--- a/src/main/java/com/ots/project/system/role/service/RoleServiceImpl.java
+++ b/src/main/java/com/ots/project/system/role/service/RoleServiceImpl.java
@@ -18,6 +18,7 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import java.util.*;
+import java.util.stream.Collectors;
 
 @Service
 public class RoleServiceImpl implements IRoleService {
@@ -191,20 +192,25 @@
         return userRoleMapper.deleteUserRoleInfo(userRole);
     }
     
+    @Override
     public int deleteAuthUsers(Long roleId, String userIds) {
         return userRoleMapper.deleteUserRoleInfos(roleId, Convert.toLongArray(userIds));
     }
-    
+
+    @Override
     public int insertAuthUsers(Long roleId, String userIds) {
         Long[] users = Convert.toLongArray(userIds);
-        
-        List<UserRole> list = new ArrayList<UserRole>();
-        for (Long userId : users) {
+
+        /**
+         * 閬嶅巻鍒嗛厤鐢ㄦ埛闆嗗悎 澶氳鑹�
+         */
+        List<UserRole> list = Arrays.stream(users).map(u -> {
             UserRole ur = new UserRole();
-            ur.setUserId(userId);
+            ur.setUserId(u);
             ur.setRoleId(roleId);
-            list.add(ur);
-        }
+            return  ur;
+        }).collect(Collectors.toList());
+
         return userRoleMapper.batchUserRole(list);
     }
 }

--
Gitblit v1.9.1