快捷導(dǎo)航

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

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

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

谷歌的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ù)，會報異常。
    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)}

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

注意：如果key重復(fù)，會報異常。

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

    // 如果key重復(fù)，會取其中一個，具體哪一個，我暫不確定。
    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)}

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

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

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ù)，會報異常。
    //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ù)，取第一個。
    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ù)，會報異常。
    //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ù)，取第二個。
    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)}

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

區(qū)別對比

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

其他說明

如果不存在將多個字段值拼接成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));