Java8 中 List 转 Map(Collectors.toMap) 使用技巧（转）

Java8 中 List 转 Map(Collectors.toMap) 使用技巧
发表于 2019-12-05 | 阅读次数： 4322

在实际项目中我们经常会用到 List 转 Map 操作，在过去我们可能使用的是 for 循环遍历的方式。举个例子：

先定义类：

// 简单对象
@Accessors(chain = true) // 链式方法
@lombok.Data
class User {
private String id;
private String name;
}
然后有这样一个 List：

List userList = Lists.newArrayList(
new User().setId(“A”).setName(“张三”),
new User().setId(“B”).setName(“李四”),
new User().setId(“C”).setName(“王五”)
);
我们希望转成 Map 的格式为：

A-> 张三
B-> 李四
C-> 王五
过去的做法（循环）：

Map<String, String> map = new HashMap<>();
for (User user : userList) {
map.put(user.getId(), user.getName());
}
使用 Java8 特性
Java8 中新增了 Stream 特性，使得我们在处理集合操作时更方便了。

以上述例子为例，我们可以一句话搞定：

userList.stream().collect(Collectors.toMap(User::getId, User::getName));
当然，如果希望得到 Map 的 value 为对象本身时，可以这样写：

userList.stream().collect(Collectors.toMap(User::getId, t -> t));
或：
userList.stream().collect(Collectors.toMap(User::getId, Function.identity()));
关于 Collectors.toMap 方法
Collectors.toMap 有三个重载方法：

toMap(Function< super T, extends K> keyMapper, Function< super T, extends U> valueMapper);
toMap(Function< super T, extends K> keyMapper, Function< super T, extends U> valueMapper,
BinaryOperator mergeFunction);
toMap(Function< super T, extends K> keyMapper, Function< super T, extends U> valueMapper,
BinaryOperator mergeFunction, Supplier mapSupplier);
参数含义分别是：

keyMapper：Key 的映射函数

valueMapper：Value 的映射函数

mergeFunction：当 Key 冲突时，调用的合并方法

mapSupplier：Map 构造器，在需要返回特定的 Map 时使用

还是用上面的例子，如果 List 中 userId 有相同的，使用上面的写法会抛异常：

List userList = Lists.newArrayList(
new User().setId(“A”).setName(“张三”),
new User().setId(“A”).setName(“李四”), // Key 相同
new User().setId(“C”).setName(“王五”)
);
userList.stream().collect(Collectors.toMap(User::getId, User::getName));

// 异常：
java.lang.IllegalStateException: Duplicate key 张三
at java.util.stream.Collectors.lambda$throwingMerger 114 ( C o l l e c t o r s . j a v a : 133 ) a t j a v a . u t i l . H a s h M a p . m e r g e ( H a s h M a p . j a v a : 1245 ) a t j a v a . u t i l . s t r e a m . C o l l e c t o r s . l a m b d a 114(Collectors.java:133) at java.util.HashMap.merge(HashMap.java:1245) at java.util.stream.Collectors.lambda 114(Collectors.java:133)atjava.util.HashMap.merge(HashMap.java:1245)atjava.util.stream.Collectors.lambdatoMap $172 (C o l l e c t o r s . j a v a : 1320) a t j a v a . u t i l . s t r e a m . R e d u c e O p s 3 R e d u c i n g S i n k . a c c e p t (R e d u c e O p s . j a v a : 169) a t j a v a . u t i l . A r r a y L i s t 3 R e d u c i n g S i n k . a c c e p t (R e d u c e O p s . j a v a : 169) a t j a v a . u t i l . A r r a y L i s t 3 R e d u c i n g S i n k . a c c e p t (R e d u c e O p s . j a v a : 169) a t j a v a . u t i l . A r r a y L i s t A r r a y L i s t S p l i t e r a t o r . f o r E a c h R e m a i n i n g (A r r a y L i s t . j a v a : 1374) a t j a v a . u t i l . s t r e a m . A b s t r a c t P i p e l i n e . c o p y I n t o (A b s t r a c t P i p e l i n e . j a v a : 481) a t j a v a . u t i l . s t r e a m . A b s t r a c t P i p e l i n e . w r a p A n d C o p y I n t o (A b s t r a c t P i p e l i n e . j a v a : 471) a t j a v a . u t i l . s t r e a m . R e d u c e O p s$ ReduceOp.evaluateSequential(ReduceOps.java:708)
at java.util.stream.AbstractPipeline.evaluate(AbstractPipeline.java:234)
at java.util.stream.ReferencePipeline.collect(ReferencePipeline.java:499)
at Test.toMap(Test.java:17)
…
这时就需要调用第二个重载方法，传入合并函数，如：

userList.stream().collect(Collectors.toMap(User::getId, User::getName, (n1, n2) -> n1 + n2));

// 输出结果：
A-> 张三李四
C-> 王五
第四个参数（mapSupplier）用于自定义返回 Map 类型，比如我们希望返回的 Map 是根据 Key 排序的，可以使用如下写法：

List userList = Lists.newArrayList(
new User().setId(“B”).setName(“张三”),
new User().setId(“A”).setName(“李四”),
new User().setId(“C”).setName(“王五”)
);
userList.stream().collect(
Collectors.toMap(User::getId, User::getName, (n1, n2) -> n1, TreeMap::new)
);

// 输出结果：
A-> 李四
B-> 张三
C-> 王五

Java8 中 List 转 Map(Collectors.toMap) 使用技巧（转）

猜你喜欢