CyclicBarrier是Java中多线程并发编程中的一个工具类,它可以让一组线程到达一个同步点后再同时继续执行。CyclicBarrier可以用于多线程计算数据,最后合并计算结果的场景。
CyclicBarrier的特点:
- 可以重复使用:CyclicBarrier可以被重复使用,每次重复使用都需要重新调用reset()方法进行重置。
- 等待线程数可变:在创建CyclicBarrier的时候可以指定一个等待线程数,但是在等待的时候,如果有新的线程想要加入等待,可以通过await()方法加入,这使得CyclicBarrier的等待线程数可以动态变化。
使用CyclicBarrier步骤:
- 创建CyclicBarrier对象,指定等待线程数和响应动作
- 创建一组线程,并启动
- 等待所有线程执行完毕后合并结果
下面是一个使用CyclicBarrier的示例代码,假设有10个线程需要并行处理数据,处理完毕后需要等待所有线程完成后再进行下一步操作:
import java.util.concurrent.CyclicBarrier;
public class CyclicBarrierDemo {
private static final int THREAD_NUM = 10;
private static final CyclicBarrier BARRIER = new CyclicBarrier(THREAD_NUM, new Runnable(){
@Override
public void run(){
System.out.println("All threads have finished execution!");
}
});
public static void main(String[] args){
for(int i = 0; i < THREAD_NUM; i++){
new Thread(new Task()).start();
}
}
static class Task implements Runnable {
@Override
public void run(){
try{
System.out.println("Thread " + Thread.currentThread().getName() + " is now processing data.");
Thread.sleep(1000); // 模拟数据处理时间
System.out.println("Thread " + Thread.currentThread().getName() + " has finished processing data.");
BARRIER.await(); // 等待其他线程执行完毕
System.out.println("Thread " + Thread.currentThread().getName() + " is now doing next task.");
}catch(Exception ex){
ex.printStackTrace();
}
}
}
}
在上面的示例代码中,我们首先创建了一个Thread类的子类Task,它实现了Runnable接口。在任务执行时,我们需要模拟数据处理时间,以便更好地演示CyclicBarrier的效果。
在main函数中,我们创建了10个线程,并启动它们。每个线程执行Task类的实例,并模拟数据处理时间,然后通过await()方法等待其他线程处理完毕。当所有线程都处理完毕后,执行BARRIER中设置的响应动作。
你也可以通过以下方式重置CyclicBarrier:
BARRIER.reset();
这会使得CyclicBarrier重新变为初始状态,并使得之前等待中的线程抛出BrokenBarrierException异常。