如何优雅地将Collection转为Map?
将Collection转换为Map是常见的需求,尤其是在处理数据时需要快速查找或去重。以下是几种常见的方法,包括使用谷歌的Maps.uniqueIndex、Hutool的CollUtil.toMap和Java Stream API的Collectors.toMap三种方法。
谷歌的Maps.uniqueIndex
/**
* 使用com.google.common.collect.Maps#uniqueIndex(java.lang.Iterable, com.google.common.base.Function)方法
*
* @author 付聪
* @time 2025-03-13 09:29:51
*/
@Test
public void testGoogleMapsUniqueIndex() {
User user1 = new User();
user1.setId(1);
user1.setName("张三");
User user2 = new User();
user2.setId(2);
user2.setName("李四");
User user3 = new User();
// 重复id(2)
user3.setId(2);
user3.setName("王五");
List<User> userList = CollUtil.newArrayList();
userList.add(user1);
userList.add(user2);
//userList.add(user3);
Set<User> userSet = CollUtil.newHashSet();
userSet.add(user1);
userSet.add(user2);
//userSet.add(user3);
// 如果key重复,会报异常。
Map<Integer, User> userListMap = Maps.uniqueIndex(userList, user -> user.getId());
System.out.println(StrUtil.format("userListMap是:{}", userListMap.toString()));
Map<Integer, User> userSetMap = Maps.uniqueIndex(userSet, user -> user.getId());
System.out.println(StrUtil.format("userSetMap是:{}", userSetMap.toString()));
}
———————————————————————— 开始测试单个方法 ————————————————————————
userListMap是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=李四, remark=null)}
userSetMap是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=李四, remark=null)}
———————————————————————— 结束测试单个方法 ————————————————————————
注意:如果key重复,会报异常。
Hutool的CollUtil.toMap
/**
* 使用cn.hutool.core.collection.CollUtil#toMap(java.lang.Iterable, java.util.Map, cn.hutool.core.lang.func.Func1, cn.hutool.core.lang.func.Func1)
*
* @author 付聪
* @time 2025-03-12 09:29:35
*/
@Test
public void testHutoolCollUtilToMap() {
User user1 = new User();
user1.setId(1);
user1.setName("张三");
User user2 = new User();
user2.setId(2);
user2.setName("李四");
User user3 = new User();
// 重复id(2)
user3.setId(2);
user3.setName("王五");
List<User> userList = CollUtil.newArrayList();
userList.add(user1);
userList.add(user2);
userList.add(user3);
Set<User> userSet = CollUtil.newHashSet();
userSet.add(user1);
userSet.add(user3);
userSet.add(user2);
// 如果key重复,会取其中一个,具体哪一个,我暂不确定。
Map<Integer, User> userListMap = CollUtil.toMap(userList, MapUtil.newHashMap(), user -> user.getId(), user -> user);
System.out.println(StrUtil.format("userListMap是:{}", userListMap.toString()));
// 如果key重复,会取其中一个,具体哪一个,我暂不确定。
Map<Integer, User> userSetMap = CollUtil.toMap(userSet, MapUtil.newHashMap(), user -> user.getId(), user -> user);
System.out.println(StrUtil.format("userSetMap是:{}", userSetMap.toString()));
}
———————————————————————— 开始测试单个方法 ————————————————————————
userListMap是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=王五, remark=null)}
userSetMap是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=王五, remark=null)}
———————————————————————— 结束测试单个方法 ————————————————————————
注意:如果key重复,会取其中一个,具体哪一个,我暂不确定。
Java Stream API的Collectors.toMap
/**
* 使用java.util.stream.Collectors#toMap(java.util.function.Function, java.util.function.Function, java.util.function.BinaryOperator)
*
* @author 付聪
* @time 2025-03-12 09:29:35
*/
@Test
public void testJavaStreamAPICollectorsToMap() {
User user1 = new User();
user1.setId(1);
user1.setName("张三");
User user2 = new User();
user2.setId(2);
user2.setName("李四");
User user3 = new User();
// 重复id(2)
user3.setId(2);
user3.setName("王五");
List<User> userList = CollUtil.newArrayList();
userList.add(user1);
userList.add(user2);
userList.add(user3);
Set<User> userSet = CollUtil.newHashSet();
userSet.add(user1);
userSet.add(user2);
userSet.add(user3);
// 如果key重复,会报异常。
//Map<Integer, User> userListMap1 = userList.stream().collect(Collectors.toMap(user -> user.getId(), user -> user));
//System.out.println(StrUtil.format("userListMap1是:{}", userListMap1.toString()));
// (key1, key2) -> key1:如果key重复,取第一个。
Map<Integer, User> userListMap2 = userList.stream().collect(Collectors.toMap(user -> user.getId(), user -> user, (key1, key2) -> key1));
System.out.println(StrUtil.format("userListMap2是:{}", userListMap2.toString()));
// 如果key重复,会报异常。
//Map<Integer, User> userSetMap1 = userSet.stream().collect(Collectors.toMap(user -> user.getId(), user -> user));
//System.out.println(StrUtil.format("userSetMap1是:{}", userSetMap1.toString()));
// (key1, key2) -> key2:如果key重复,取第二个。
Map<Integer, User> userSetMap2 = userSet.stream().collect(Collectors.toMap(user -> user.getId(), user -> user, (key1, key2) -> key2));
System.out.println(StrUtil.format("userSetMap2是:{}", userSetMap2.toString()));
}
———————————————————————— 开始测试单个方法 ————————————————————————
userListMap2是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=李四, remark=null)}
userSetMap2是:{1=User(id=1, name=张三, remark=null), 2=User(id=2, name=王五, remark=null)}
———————————————————————— 结束测试单个方法 ————————————————————————
区别对比
是否支持处理重复key | 如果key重复会怎样 | |
---|---|---|
谷歌的Maps.uniqueIndex | 否 | 报错 |
Hutool的CollUtil.toMap | 否 | 取其中一个 |
Java Stream API的Collectors.toMap | 是 | 可根据合并函数取值 |
其他说明
如果不存在将多个字段值拼接成key的情况,也可以把【user -> user.getId()】改为【User::getId】,但是如果存在将多个字段值拼接成key的情况,就可以使用【->】的形式进行拼接。
Map<String, User> userListMap2 = userList.stream().collect(Collectors.toMap(user -> StrUtil.format("{}_{}", user.getId(), user.getName()), user -> user, (key1, key2) -> key1));