有两个SysRoleAuth的List集合
- 一个集合只有roleId、perms数据（可能有多个重复roleId，perms不重复）
- 另一个只有roleId、resourceCode数据
两个集合roleId并不是都一一对应的
问：
如何把两集合相同roleId数据合并添加
若有不同roleId，也添加进去
最后类型为： Map<Long, SysRoleList>
画了个图，方便大家理解
@Data
@AllArgsConstructor
@NoArgsConstructor
public class SysRoleAuth {
/**
* 角色id
*/
Long roleId;
/**
* 权限字符
*/
String perms;
/**
* 资源编码
*/
String resourceCode;
}
SysRoleList：
@Data
@AllArgsConstructor
@NoArgsConstructor
public class SysRoleList {
/**
* 角色id
*/
Long roleID;
/**
* 权限字符
*/
Set<String> perms;
/**
* 资源编码
*/
Set<String> resourceCode;
}
这是我写的代码，只是实现功能，有没有好的解决方案
package com.oddfar.test.role;
import java.util.*;
import java.util.stream.Collectors;
public class TestList {
public static void main(String[] args) {
Map<Long, SysRoleList> resMap = new HashMap<>();
Map<Long, List<SysRoleAuth>> list1 = creatList1().stream().collect(Collectors.groupingBy(SysRoleAuth::getRoleID));
for (Long roleId : list1.keySet()) {
List<SysRoleAuth> list = list1.get(roleId);
//把关于roleId的perms数据建立成set集合
Set<String> perms = new HashSet<>();
list.stream().forEach(r -> perms.add(r.getPerms()));
//如果resMap包含roleId
if (resMap.containsKey(roleId)) {
SysRoleList sysRoleList = resMap.get(roleId);
if (sysRoleList.getPerms() != null) {
//如果存在perms，则添加数据
sysRoleList.getPerms().addAll(perms);
} else {
//无数据则直接set
sysRoleList.setPerms(perms);
}
} else {
//不包含roleId重新生成
SysRoleList sysRoleList = new SysRoleList(roleId, perms, null);
resMap.put(roleId, sysRoleList);
}
}
//list2集合操作同理list1
Map<Long, List<SysRoleAuth>> list2 = creatList2().stream().collect(Collectors.groupingBy(SysRoleAuth::getRoleID));
for (Long roleId : list2.keySet()) {
List<SysRoleAuth> list = list2.get(roleId);
//把关于roleId的resource数据建立成set集合
Set<String> resourceSet = new HashSet<>();
list.stream().forEach(r -> resourceSet.add(r.getResourceCode()));
//如果map包含roleId
if (resMap.containsKey(roleId)) {
SysRoleList sysRoleList = resMap.get(roleId);
if (sysRoleList.getResourceCode() != null) {
sysRoleList.getResourceCode().addAll(resourceSet);
} else {
sysRoleList.setResourceCode(resourceSet);
}
} else {
SysRoleList sysRoleList = new SysRoleList(roleId, null, resourceSet);
resMap.put(roleId, sysRoleList);
}
}
System.out.println(resMap);
}
/**
* 创建第1个集合
*/
public static List<SysRoleAuth> creatList1() {
List<SysRoleAuth> list = new ArrayList<>();
//id为1的权限字符列表
list.add(new SysRoleAuth(1L, "system:user:list", null));
list.add(new SysRoleAuth(1L, "system:user:add", null));
list.add(new SysRoleAuth(1L, "system:user:edit", null));
//id为2的权限字符列表
list.add(new SysRoleAuth(2L, "system:config:query", null));
//id为3的权限字符列表
list.add(new SysRoleAuth(3L, "system:notice:query", null));
return list;
}
/**
* 创建第2个集合
*/
public static List<SysRoleAuth> creatList2() {
List<SysRoleAuth> list = new ArrayList<>();
//id为1的资源编码列表
list.add(new SysRoleAuth(1L, null, "campus.test.test1"));
list.add(new SysRoleAuth(1L, null, "campus.test.test2"));
//id为2的资源编码列表
list.add(new SysRoleAuth(2L, null, "campus.test.test1"));
//id为6的资源编码列表
list.add(new SysRoleAuth(6L, null, "campus.test.test1"));
return list;
}
}
最后返回数据：
{1=SysRoleList(roleID=1, perms=[system:user:list, system:user:edit, system:user:add], resourceCode=[campus.test.test1, campus.test.test2]), 2=SysRoleList(roleID=2, perms=[system:config:query], resourceCode=[campus.test.test1]), 3=SysRoleList(roleID=3, perms=[system:notice:query], resourceCode=null), 6=SysRoleList(roleID=6, perms=null, resourceCode=[campus.test.test1])}
发表于 2022-11-15 14:12
|
发表于 2022-11-15 14:36
