当前位置: 首页 > article >正文

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

http://www.kler.cn/a/585191.html

相关文章:

  • 平安养老险广西分公司2025年“3∙15”金融消费者权益教育宣传活动暨南湖公园健步行活动
  • 【C语言】编译和链接详解
  • Redis的缓存雪崩、缓存击穿、缓存穿透与缓存预热、缓存降级
  • 2025-03-15 学习记录--C/C++-PTA 练习3-4 统计字符
  • 【3D视觉学习笔记2】摄像机的标定、畸变的建模、2D/3D变换
  • python如何获取三个小时之前的时间并输出
  • MATLAB 控制系统设计与仿真 - 26
  • python画图文字显示不全+VScode新建jupyter文件
  • 构建分类树(ElementPlus的二级数据模型)
  • [S32K]SPI
  • Python 语言因其广泛的库与框架资源,诸如 `requests`、`BeautifulSoup
  • 证券交易系统的流程
  • pytorch lightning ddp 逆天分配显存方式
  • 关于重构分析查询界面的思考(未完)
  • 基于Hadoop的城市道路交通数据的可视化分析-Flask
  • 前端技巧第五期JavaScript函数
  • C++ 内存管理
  • NFC碰一碰发视频-nfc碰一碰发视频拓客系统 实体商家碰一碰发视频引流获客
  • AI辅助工具-通义灵码
  • 【机器学习】基于t-SNE的MNIST数据集可视化探索