快捷導(dǎo)航

如何優(yōu)雅的實(shí)現(xiàn)將Collection轉(zhuǎn)為Map

更新時(shí)間：2025年03月25日 09:45:13 作者：付聰1210

這篇文章主要介紹了如何優(yōu)雅的實(shí)現(xiàn)將Collection轉(zhuǎn)為Map,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

谷歌的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();
    // 重復(fù)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重復(fù)，會(huì)報(bào)異常。
    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()));

}

———————————————————————— 開(kāi)始測(cè)試單個(gè)方法 ————————————————————————

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)}

———————————————————————— 結(jié)束測(cè)試單個(gè)方法 ————————————————————————

注意：如果key重復(fù)，會(huì)報(bào)異常。

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();
    // 重復(fù)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重復(fù)，會(huì)取其中一個(gè)，具體哪一個(gè)，我暫不確定。
    Map<Integer, User> userListMap = CollUtil.toMap(userList, MapUtil.newHashMap(), user -> user.getId(), user -> user);
    System.out.println(StrUtil.format("userListMap是：{}", userListMap.toString()));

    // 如果key重復(fù)，會(huì)取其中一個(gè)，具體哪一個(gè)，我暫不確定。
    Map<Integer, User> userSetMap = CollUtil.toMap(userSet, MapUtil.newHashMap(), user -> user.getId(), user -> user);
    System.out.println(StrUtil.format("userSetMap是：{}", userSetMap.toString()));

}

———————————————————————— 開(kāi)始測(cè)試單個(gè)方法 ————————————————————————

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)}

———————————————————————— 結(jié)束測(cè)試單個(gè)方法 ————————————————————————

注意：如果key重復(fù)，會(huì)取其中一個(gè)，具體哪一個(gè)，我暫不確定。

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();
    // 重復(fù)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重復(fù)，會(huì)報(bào)異常。
    //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重復(fù)，取第一個(gè)。
    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重復(fù)，會(huì)報(bào)異常。
    //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重復(fù)，取第二個(gè)。
    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()));

}

———————————————————————— 開(kāi)始測(cè)試單個(gè)方法 ————————————————————————

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)}

———————————————————————— 結(jié)束測(cè)試單個(gè)方法 ————————————————————————

區(qū)別對(duì)比

	是否支持處理重復(fù)key	如果key重復(fù)會(huì)怎樣
谷歌的Maps.uniqueIndex	否	報(bào)錯(cuò)
Hutool的CollUtil.toMap	否	取其中一個(gè)
Java Stream API的Collectors.toMap	是	可根據(jù)合并函數(shù)取值

其他說(shuō)明

如果不存在將多個(gè)字段值拼接成key的情況，也可以把【user -> user.getId()】改為【User::getId】，但是如果存在將多個(gè)字段值拼接成key的情況，就可以使用【->】的形式進(jìn)行拼接。

Map<String, User> userListMap2 = userList.stream().collect(Collectors.toMap(user -> StrUtil.format("{}_{}", user.getId(), user.getName()), user -> user, (key1, key2) -> key1));