wait 和 notify 方法可用于多个线程等待一个线程。有时候我们需要在主线程中发起多个线程,然后等待它们全部执行完再继续。最后面是一个简单例子。
如何判断这些线程是否都执行完毕,这里有一点讲究。如果靠主线程自己计数,就有可能在加一的过程中错过了某些 notify。所以需要由子线程完成的时候来给计数器加一。子线程之间是相互异步的,所以要用锁来同步这个动作。那么把计数器放在锁对象里面就是自然而然的事情了。
如何判断这些线程是否都执行完毕,这里有一点讲究。如果靠主线程自己计数,就有可能在加一的过程中错过了某些 notify。所以需要由子线程完成的时候来给计数器加一。子线程之间是相互异步的,所以要用锁来同步这个动作。那么把计数器放在锁对象里面就是自然而然的事情了。
/**
* 一个对象等待多个对象
*/
public class MultipleNotify {
/**
* 程序入口
*
* @param args 命令行参数
*/
public static void main(String[] args) {
new MultipleNotify().go();
}
/**
* 演示如何使用 wait() 和 notify() 实现一个对象等待多个对象
*/
public void go() {
// 锁对象中包含了一个数值
final Lock lock = new Lock();
// 运行 20 个线程
Notifier[] notifiers = new Notifier[20];
for (int i = 0; i < notifiers.length; i++) {
notifiers[i] = new Notifier(lock);
notifiers[i].start();
}
// 循环等待,通过 lock 的值来判断是否结束等待
boolean enough = false;
while (!enough) {
try {
synchronized (lock) {
lock.wait();
if (lock.value == notifiers.length) {
enough = true;
}
}
} catch (InterruptedException e) {
e.printStackTrace();
}
}
System.out.println("OK!");
}
/**
* 执行的线程。主线程要等待所有 Notifier 执行完毕才会继续
*/
private class Notifier extends Thread {
private final Lock lock;
private Notifier(Lock lock) {
this.lock = lock;
}
public void run() {
int sleep = 0;
try {
sleep = 500 + new Random().nextInt(2000);
Thread.sleep(sleep);
} catch (InterruptedException e) {
e.printStackTrace();
}
// 执行 lock.notify() 的同时将 lock 对象的值加 1
synchronized (lock) {
lock.setValue(lock.getValue() + 1);
System.out.println("lock = " + lock.getValue() + ", sleep:" + sleep);
lock.notify();
}
}
}
/**
* 包含一个数值的锁
*/
private class Lock {
private int value = 0;
public int getValue() {
return value;
}
public void setValue(int value) {
this.value = value;
}
}
}