Parallel stream 과 Map

1. parallelStream + forEach 로  HashMap 을 collect 하려할 경우 원하는 결과가 안나올 수 있다.

Map<Long, Data> dataIdMap = new HashMap<>();

events
    .parallelStream()
    .forEach(data -> {
        dataIdMap.put(data.getId(), data);
    });

HashMap 은 Thread safe 하지 않다.

 

해결법 1 : ConcurrentHashMap 을 사용한다.

해결법 2 : stream 을 사용한다.

Map<Long, Data> dataIdMap = new HashMap<>();

events
    .stream()
    .forEach(data -> {
        dataIdMap.put(data.getId(), data);
    });

해결법 3 : 자바 8의 collect 기능을 이용한다.

Map<Long, Data> dataIdMap = events
    .parallelStream()
    .collect(Collectors.toMap(
        data -> data.getId(),
        data -> data));

 

 

 

 

 

2. parallelStream 으로 TreeMap 을 collect 하려 할 경우 무한 루프를 만들 수도 있다.

TreeMap<Long, List<Data>> dataGroupByGroupId = events
    .parallelStream()
    .collect(Collectors.groupingBy(Data::getGroupId, TreeMap::new, toList()));

https://ivoanjo.me/blog/2018/07/21/writing-to-a-java-treemap-concurrently-can-lead-to-an-infinite-loop-during-reads/

Writing to a Java TreeMap concurrently can lead to an infinite loop during reads - ivo's awfully random tech blog