CountDownLatch 线程计数器
CountDownLatch 线程计数器,可以实现:一个主线程等到N个子线程全部执行完成后再执行主线程代码。
final CountDownLatch latch = new CountDownLatch(2);
new Thread() {
public void run() {
System.out.println("子线程" + Thread.currentThread().getName() + "正在执行");
try {
Thread.sleep(3000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("子线程" + Thread.currentThread().getName() + "执行完毕");
latch.countDown();
}
}.start();
new Thread() {
public void run() {
System.out.println("子线程" + Thread.currentThread().getName() + "正在执行");
try {
Thread.sleep(3000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("子线程" + Thread.currentThread().getName() + "执行完毕");
latch.countDown();
}
}.start();
System.out.println("等待 2 个子线程执行完毕...");
latch.await();
System.out.println("2 个子线程已经执行完毕");
System.out.println("继续执行主线程");
执行结果
CyclicBarrier 回环栅栏
CyclicBarrier 回环栅栏。通过它可以实现让一组线程等待至某个状态之后再全部同时执行。叫做回环是因为当所有等待线程都被释放以后,CyclicBarrier 可以被重用。我们把这个状态就叫做barrier,当调用 await()方法之后,线程就处于 barrier 了。
CyclicBarrier 中最重要的方法就是 await 方法,它有 2 个重载版本:
- public int await():用来挂起当前线程,直至所有线程都到达 barrier 状态再同时执行后续任务;
- public int await(long timeout, TimeUnit unit):让这些线程等待至一定的时间,如果还有线程没有到达 barrier 状态就直接让到达 barrier 的线程执行后续任务。
需求:把100M的excel用户数据保存到数据库中分析统计。
int N = 3;
CyclicBarrier barrier = new CyclicBarrier(N);
for (int i = 0; i < N; i++) {
new Thread(() -> {
try {
System.out.println("子线程" + Thread.currentThread().getName() + "开始导入数据");
Thread.sleep(1 + (int) (Math.random() * 20000));
System.out.println("子线程" + Thread.currentThread().getName() + "完成导入数据");
barrier.await();
System.out.println("子线程" + Thread.currentThread().getName() + "开始分析数据");
Thread.sleep(1 + (int) (Math.random() * 20000));
System.out.println("子线程" + Thread.currentThread().getName() + "完成分析数据");
barrier.await();
System.out.println("子线程" + Thread.currentThread().getName() + "保存分析数据");
Thread.sleep(1 + (int) (Math.random() * 20000));
System.out.println("子线程" + Thread.currentThread().getName() + "任务结束");
} catch (Exception e) {
e.printStackTrace();
}
}).start();
运行结果
Semaphore 信号量-控制同时访问的线程个数
Semaphore 翻译成字面意思为 信号量,Semaphore 可以控制同时访问的线程个数,通过acquire() 获取一个许可,如果没有就等待,而 release() 释放一个许可。
Semaphore 类中比较重要的几个方法:
- public void acquire(): 用来获取一个许可,若无许可能够获得,则会一直等待,直到获得许可。
- public void acquire(int permits):获取 permits 个许可
- public void release() { } :释放许可。注意,在释放许可之前,必须先获获得许可。
- public void release(int permits) { }:释放 permits 个许可
例子:若一个工厂有 3 台机器,但是有 8 个工人,一台机器同时只能被一个工人使用,只有使用完了,其他工人才能继续使用。那么我们就可以通过 Semaphore 来实现:
Semaphore semaphore = new Semaphore(3);
for (int i = 0; i < 8; i++) {
new Thread(() -> {
try {
semaphore.acquire();
System.out.println("工人" + Thread.currentThread().getName() + "开始使用机器");
Thread.sleep(20000);
System.out.println("工人" + Thread.currentThread().getName() + "结束使用机器");
} catch (Exception e) {
e.printStackTrace();
} finally {
semaphore.release();
}
}).start();
}
运行结果
上面 4 个方法都会被阻塞,如果想立即得到执行结果,可以使用下面几个方法
6. public boolean tryAcquire():尝试获取一个许可,若获取成功,则立即返回 true,若获取失败,则立即返回 false
7. public boolean tryAcquire(long timeout, TimeUnit unit):尝试获取一个许可,若在指定的时间内获取成功,则立即返回 true,否则则立即返回 false
8. public boolean tryAcquire(int permits):尝试获取 permits 个许可,若获取成功,则立即返回 true,若获取失败,则立即返回 false
9. public boolean tryAcquire(int permits, long timeout, TimeUnit unit): 尝试获取 permits个许可,若在指定的时间内获取成功,则立即返回true,否则则立即返回 false