目录

如何优雅地将Collection转为Map

JAY.LIN 收录于数据结构和算法

2025-03-13 约 2470 字预计阅读 5 分钟

https://bing.ee123.net/img/rand?artid=146224591

目录

如何优雅地将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));