Java Stream使用注意事项

本文讨论了在Java中使用并行流操作共享资源时可能遇到的线程安全问题,特别是HashMap在多线程环境下的风险。通过案例展示了在并行流中对HashMap进行写操作可能导致的结果错误,并提出了使用ConcurrentHashMap作为线程安全的解决方案。同时,文章提醒在使用Stream汇总到Map时应指定key冲突的处理方式,避免运行时异常。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.并行流的线程安全

如果在并行流中涉及到对共享资源的写的操作,需要注意多线程带来的问题。如下代码中,在并行流中操作共享资源result,由于HashMap非线程安全,代码执行结束之后可能会得到错误的结果

public Map<String, String> foo(List<String> demoList, String arg) {
        
        if (CollectionUtils.isEmpty(demoList)) {
            return Collections.emptyMap();
        }
        
        Map<String, String> result = new HashMap<>();
        
        demoList.parallelStream().forEach(text -> {
            try {
                result.put(text, doSomething(arg));
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
        
        return result;
    }

解决方案

保证共享资源的线程安全性,容器类可以改用对应的线程安全的类型,如HashMap改用ConcurrentHashMap,其他同理,伪代码如下:

        Map<String, String> result = new ConcurrentHashMap<>();

        demoList.parallelStream().forEach(text -> {
            try {
                result.put(text, doSomething(arg));
            } catch (Exception e) {
                e.printStackTrace();
            }
        });

2.流(Stream)汇总为Map的时候建议指定key重复时的处理方式,若不指定,将会抛出异常

正向用例伪代码如下:

public Map<String, String> bar(List<String> demoList) {

        if (CollectionUtils.isEmpty(demoList)) {
            return Collections.emptyMap();
        }

        return demoList.stream().collect(Collectors.toMap(k -> k,
                Function.identity(), (v1, v2) -> v1));
    }

对应Collectors的API为:

Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
                                    Function<? super T, ? extends U> valueMapper,
                                    BinaryOperator<U> mergeFunction)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值