将Collection转换为Map是常见的需求,尤其是在处理数据时需要快速查找或去重。以下是几种常见的方法,包括使用谷歌的Maps.uniqueIndex、Hutool的CollUtil.toMap和Java Stream API的Collectors.toMap三种方法。
谷歌的Maps.uniqueIndex
java">/*** 使用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
java">/*** 使用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">/*** 使用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));