如何为parallelStream设置ThreadLocal

Question

我有一个线程，它包含ThreadLocal变量。我需要在上面提到的线程中使用parallelStream()。需要调用使用线程局部变量的myService。在java8中使用parallelstream()时，是否有任何机制来设置ThreadLocal。

List<MyObject> result = myList.parallelStream().map(myObject -> {
    //call myService with the Threadlocal 
}).filter(...)
.....;

Answer 1

您可以在调用服务之前轻松地设置threadLocal变量。在map中，您可以从main threadlocal value或任何其他值中设置值。

 ThreadLocal<String> threadLocal= new ThreadLocal<>();

        IntStream.range(0, 8).parallel().forEach(n -> {
            threadLocal.set("MAIN");
            System.out.println("This is sequence access "+n);
            System.out.printf("Service used ThreadLocal - %d: %s\n", n, threadLocal.get());
        });

结果：

This is sequence access 5
This is sequence access 7
Parallel Consumer - 5: MAIN
Parallel Consumer - 7: MAIN
This is sequence access 4
This is sequence access 6
Parallel Consumer - 4: MAIN
Parallel Consumer - 6: MAIN
This is sequence access 2
This is sequence access 1
Parallel Consumer - 2: MAIN
This is sequence access 0
This is sequence access 3
Parallel Consumer - 0: MAIN
Parallel Consumer - 1: MAIN
Parallel Consumer - 3: MAIN