ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

处理了CollStreamUtil.toIdentityMap中，如果value为空时，发生NPE异常问题

Browse Source
This commit is contained in:
VampireAchao 2021-12-06 21:13:26 +08:00 committed by achao
parent a2331bf5c9
commit 1b28ba13e4
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
hutool-core/src/main/java/cn/hutool/core/collection/CollStreamUtil.java
View File

@ -1,6 +1,7 @@
package cn.hutool.core.collection; package cn.hutool.core.collection;
import cn.hutool.core.lang.Opt;
import cn.hutool.core.map.MapUtil; import cn.hutool.core.map.MapUtil;
import cn.hutool.core.stream.StreamUtil; import cn.hutool.core.stream.StreamUtil;
@ -48,7 +49,7 @@ public class CollStreamUtil {
return Collections.emptyMap(); return Collections.emptyMap();
} }
return StreamUtil.of(collection, isParallel) return StreamUtil.of(collection, isParallel)
.collect(Collectors.toMap(key, Function.identity(), (l, r) -> l)); .collect(HashMap::new, (HashMap<K, V> m, V v) -> m.put(Opt.ofNullable(v).map(key).get(), v), HashMap::putAll);
} }
/** /**

5
hutool-core/src/test/java/cn/hutool/core/collection/CollStreamUtilTest.java
View File

@ -28,6 +28,11 @@ public class CollStreamUtilTest {
Assert.assertEquals(map.get(2L).getName(), "李四"); Assert.assertEquals(map.get(2L).getName(), "李四");
Assert.assertEquals(map.get(3L).getName(), "王五"); Assert.assertEquals(map.get(3L).getName(), "王五");
Assert.assertNull(map.get(4L)); Assert.assertNull(map.get(4L));
// 测试value为空时
list.add(null);
map = CollStreamUtil.toIdentityMap(list, Student::getStudentId);
Assert.assertNull(map.get(4L));
} }
@Test @Test