使用Object的wait,notify,notifyAll做线程调度
我们知道java中的所有类的祖先都是Object,Object类有四个个方法wait(),wait(long timeout),notify(),notifyAll(),这四个方法可以用来做线程的调度或者说是线程的同步控制。
- wait() 方法用来控制当前线程停止执行,等待其他线程对此Object实例调用notify或者notifyAll方法之后再继续执行
- wait(long timeout) 此方法的作用和wait()类似,但是增加了一个超时的设置,如果等待时间超过了timeout设定的毫秒数,那么当前线程会继续执行
- notify()方法从所有wait线程中选择一个线程,让它开始执行
- notifyAll()方法通知所有等待此对象的线程,开始执行
上面的解释字面意思上很容易理解,但是实际使用起来,却并不是那么简单,我们以一个实际的例子来看下如何使用这些方法。
假定我们有两个线程要打印1到9这9个数字,要求第一个线程打印1,2,3然后停止打印,由线程2打印4,5,6,然后线程2停止打印,通知线程1继续打印7,8,9.
需求很简单,我们可以建两个Runnable类,假定为PrinterA和PrinterB,先由PrinterB等待,由PrinterA打印1,2,3;PrinterA打印完之后通知PrinterB,然后自己进入等待状态;PrintB获得PrinterA的通知之后开始打印4、5、6,打印完毕之后需要通知PrinterA;然后PrinterA得到通知之后开始打印剩下的7、8、9。任务就完成了。
package cn.outofmemory.threading; public class WaitNotifyDemo { private volatile int val = 1; private synchronized void printAndIncrease() { System.out.println(Thread.currentThread().getName() + " prints " + val); val++; } // print 1,2,3 7,8,9 public class PrinterA implements Runnable { @Override public void run() { while (val <= 3) { printAndIncrease(); } // print 1,2,3 then notify printerB synchronized (WaitNotifyDemo.this) { System.out.println("PrinterA printed 1,2,3; notify PrinterB"); WaitNotifyDemo.this.notify(); } try { while (val <= 6) { synchronized (WaitNotifyDemo.this) { System.out.println("wait in printerA"); WaitNotifyDemo.this.wait(); } } System.out.println("wait end printerA"); } catch (InterruptedException e) { e.printStackTrace(); } while (val <= 9) { printAndIncrease(); } System.out.println("PrinterA exits"); } } // print 4,5,6 after printA print 1,2,3 public class PrinterB implements Runnable { @Override public void run() { while (val < 3) { synchronized (WaitNotifyDemo.this) { try { System.out .println("printerB wait for printerA printed 1,2,3"); WaitNotifyDemo.this.wait(); System.out .println("printerB waited for printerA printed 1,2,3"); } catch (InterruptedException e) { e.printStackTrace(); } } } while (val <= 6) { printAndIncrease(); } System.out.println("notify in printerB"); synchronized (WaitNotifyDemo.this) { WaitNotifyDemo.this.notify(); } System.out.println("notify end printerB"); System.out.println("PrinterB exits."); } } public static void main(String[] args) { WaitNotifyDemo demo = new WaitNotifyDemo(); demo.doPrint(); } private void doPrint() { PrinterA pa = new PrinterA(); PrinterB pb = new PrinterB(); Thread a = new Thread(pa); a.setName("printerA"); Thread b = new Thread(pb); b.setName("printerB"); // 必须让b线程先执行,否则b线程有可能得不到锁,执行不了wait,而a线程一直持有锁,会先notify了 b.start(); a.start(); } }
我们先把所有代码奉上了,你可以自己调试代码,在实际执行中来了解代码的实现机制。下面我们逐步分析下我们是如何控制两个线程调度的。
首先看main方法,在main方法中我们初始化了一个WaitNotifyDemo实例,然后调用了这个实例的doPrint方法。
在doPrint方法中我们使用PrinterA和PrinterB的实例初始化了两个线程,然后启动他们。
private void doPrint() {
PrinterA pa = new PrinterA();
PrinterB pb = new PrinterB();
Thread a = new Thread(pa);
a.setName("printerA");
Thread b = new Thread(pb);
b.setName("printerB");
// 必须让b线程先执行,否则b线程有可能得不到锁,执行不了wait,而a线程一直持有锁,会先notify了
b.start();
a.start();
}
这里需要注意必须让b线程先执行,这样b线程才能先获得WaitNotifyDemo实例上的锁,并开始等待。在PrinterB的run方法中开始等待的代码片段如下:
while (val < 3) {
synchronized (WaitNotifyDemo.this) {
try {
System.out
.println("printerB wait for printerA printed 1,2,3");
WaitNotifyDemo.this.wait();
System.out
.println("printerB waited for printerA printed 1,2,3");
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
这里有一个while循环,如果val的值小于3,那么在WaitNotifyDemo的实例的同步块中调用WaitNotifyDemo.this.wait()方法,这里要注意无论是wait,还是notify,notifyAll方法都需要在其实例对象的同步块中执行,这样当前线程才能获得同步实例的同步控制权,如果不在同步块中执行wait或者notify方法会出现java.lang.IllegalMonitorStateException异常。另外还要注意在wait方法两边的同步块会在wait执行完毕之后释放对象锁。
这样PrinterB就进入了等待状态,我们再看下PrinterA的run方法:
while (val <= 3) {
printAndIncrease();
}
// print 1,2,3 then notify printerB
synchronized (WaitNotifyDemo.this) {
System.out.println("PrinterA printed 1,2,3; notify PrinterB");
WaitNotifyDemo.this.notify();
}
try {
while (val <= 6) {
synchronized (WaitNotifyDemo.this) {
System.out.println("wait in printerA");
WaitNotifyDemo.this.wait();
}
}
System.out.println("wait end printerA");
} catch (InterruptedException e) {
e.printStackTrace();
}
这里首先打印了1、2、3,然后在同步块中调用了WaitNotifyDemo实例的notify方法,这样PrinterB就得到了继续执行的通知,然后PrinterA进入等待状态,等待PrinterB通知。
我们再看下PrinterB run方法剩下的代码:
while (val <= 6) {
printAndIncrease();
}
System.out.println("notify in printerB");
synchronized (WaitNotifyDemo.this) {
WaitNotifyDemo.this.notify();
}
System.out.println("notify end printerB");
System.out.println("PrinterB exits.");
PrinterB首先打印了4、5、6,然后在同步块中调用了notify方法,通知PrinterA开始执行。
PrinterA得到通知后,停止等待,打印剩下的7、8、9三个数字,如下是PrinterA run方法中剩下的代码:
while (val <= 9) {
printAndIncrease();
}
整个程序就分析完了,run下程序,控制台输出如下:
printerB wait for printerA printed 1,2,3
printerA prints 1
printerA prints 2
printerA prints 3
PrinterA printed 1,2,3; notify PrinterB
printerB waited for printerA printed 1,2,3
printerB prints 4
printerB prints 5
printerB prints 6
notify in printerB
notify end printerB
PrinterB exits.
wait end printerA
printerA prints 7
printerA prints 8
printerA prints 9
PrinterA exits
从输出内容上也可以看到wait,notify的执行过程。
在用wait,notify做线程同步是要特别注意下面两点:
- 不要选择字符串,Integer,Long,Type之类的对象做同步对象,因为这些类型在jvm中都有一些特殊的处理,有可能会有意想不到的情况。比如Integer,JVM对小于128的数字做了cache,如果你用Integer做同步对象的话,可能不同的逻辑锁定了相同的同步块。这类问题调试起来也不好调试,所以最好避免这样使用。
- 在调用obj.notify(),obj.wait方法时要在synchronized(obj)块中进行调用,否则会出现java.lang.IllegalMonitorStateException异常