线程间的通信与生产者消费者模型
在前面我们讲了很多关于同步的问题,然而在现实中,需要线程之间的协作。比如说最经典的生产者-消费者模型:当队列满时,生产者需要等待队列有空间才能继续往里面放入商品,而在等待的期间内,生产者必须释放对临界资源(即队列)的占用权。因为生产者如果不释放对临界资源的占用权,那么消费者就无法消费队列中的商品,就不会让队列有空间,那么生产者就会一直无限等待下去。因此,一般情况下,当队列满时,会让生产者交出对临界资源的占用权,并进入挂起状态。然后等待消费者消费了商品,然后消费者通知生产者队列有空间了。同样地,当队列空时,消费者也必须等待,等待生产者通知它队列中有商品了。这种互相通信的过程就是线程间的协作。
在Java中线程协作有最常见的两种方式:1)利用Object.wait()、Object.notify();2)使用Condition
wait()、notify()和notifyAll()
wait()、notify()和notifyAll()是Object类中的方法:
/**
* Wakes up a single thread that is waiting on this object's
* monitor. If any threads are waiting on this object, one of them
* is chosen to be awakened. The choice is arbitrary and occurs at
* the discretion of the implementation. A thread waits on an object's
* monitor by calling one of the wait methods
*/
public final native void notify();
/**
* Wakes up all threads that are waiting on this object's monitor. A
* thread waits on an object's monitor by calling one of the
* wait methods.
*/
public final native void notifyAll();
/**
* Causes the current thread to wait until either another thread invokes the
* {@link java.lang.Object#notify()} method or the
* {@link java.lang.Object#notifyAll()} method for this object, or a
* specified amount of time has elapsed.
* <p>
* The current thread must own this object's monitor.
*/
public final native void wait(long timeout) throws InterruptedException;
从这三个方法的文字描述可以知道以下几点信息:
1)wait()、notify()和notifyAll()方法是本地方法,并且为final方法,无法被重写。
2)调用某个对象的wait()方法能让当前线程阻塞,并且当前线程必须拥有此对象的monitor(即锁)。如果在调用wait()时,没有持有适当的锁,则抛出IllegalMonitorStateException,它是RuntimeException的一个子类。
3)调用某个对象的notify()方法能够唤醒一个正在等待这个对象的monitor的线程,如果有多个线程都在等待这个对象的monitor,则只能唤醒其中一个线程;如果在调用notify()时没有持有适当的锁,也会抛出IllegalMonitorStateException。
4)调用notifyAll()方法能够唤醒所有正在等待这个对象的monitor的线程;
上面已经提到,如果调用某个对象的wait()方法,当前线程必须拥有这个对象的monitor(即锁),因此调用wait()方法必须在同步块或者同步方法中进行(synchronized块或者synchronized方法)。
调用某个对象的wait()方法,相当于让当前线程交出此对象的monitor,然后进入等待状态,等待后续再次获得此对象的锁(Thread类中的sleep方法使当前线程暂停执行一段时间,从而让其他线程有机会继续执行,但它并不释放对象锁);
notify()方法能够唤醒一个正在等待该对象的monitor的线程,当有多个线程都在等待该对象的monitor的话,则只能唤醒其中一个线程,具体唤醒哪个线程则不得而知。
同样地,调用某个对象的notify()方法,当前线程也必须拥有这个对象的monitor,因此调用notify()方法必须在同步块或者同步方法中进行(synchronized块或者synchronized方法)。
nofityAll()方法能够唤醒所有正在等待该对象的monitor的线程,这一点与notify()方法是不同的。
这里要注意一点:notify()和notifyAll()方法只是唤醒等待该对象的monitor的线程,并不决定哪个线程能够获取到monitor。
举个简单的例子:假如有三个线程Thread1、Thread2和Thread3都在等待对象objectA的monitor,此时Thread4拥有对象objectA的monitor,当在Thread4中调用objectA.notify()方法之后,Thread1、Thread2和Thread3只有一个能被唤醒。注意,被唤醒不等于立刻就获取了objectA的monitor。假若在Thread4中调用objectA.notifyAll()方法,则Thread1、Thread2和Thread3三个线程都会被唤醒,至于哪个线程接下来能够获取到objectA的monitor就具体依赖于操作系统的调度了。
上面尤其要注意一点:一个线程被唤醒不代表立即获取了对象的monitor,只有等调用完notify()或者notifyAll()并退出synchronized块,释放对象锁后,其余线程才可获得锁执行。
举个例子:
public class WaitNotifyTest {
private static Object obj = new Object();
public static void main(String[] args) throws Exception {
Thread t1 = new Thread() {
public void run() {
synchronized(obj) {
try {
obj.wait();
} catch (InterruptedException ex) {}
System.out.println("t1 成功获得了锁");
}
}
};
Thread t2 = new Thread() {
public void run() {
synchronized(obj) {
System.out.println("t2 调用notify()");
obj.notify();
}
System.out.println("t2 释放了锁");
}
};
t1.start();
Thread.sleep(2000);
t2.start();
}
}
无论如何运行,执行结果如下:
t2 调用notify()
t2 释放了锁
t1 成功获得了锁
显式的Condition对象
Condition是在java 1.5中才出现的,它用来替代传统的Object的wait()、notify()实现线程间的协作,相比使用Object的wait()、notify(),使用Condition的await()、signal()这种方式实现线程间协作更加安全和高效。因此通常来说比较推荐使用Condition。
Condition是个接口,基本的方法如下:
public interface Condition {
void await() throws InterruptedException;
void awaitUninterruptibly();
long awaitNanos(long nanosTimeout) throws InterruptedException;
boolean await(long time, TimeUnit unit) throws InterruptedException;
boolean awaitUntil(Date deadline) throws InterruptedException;
void signal();
void signalAll();
}
Condition依赖于Lock接口,生成一个Condition的基本代码是:lock.newCondition();
调用Condition的await()和signal()方法,都必须在lock保护之内,就是说必须在lock.lock()和lock.unlock()之间才可以使用。
Conditon中的await()对应Object的wait();Condition中的signal()对应Object的notify();Condition中的signalAll()对应Object的notifyAll()。
生产者与消费者模型
1)使用内置锁、Object的wait()和notify()实现:
public class ProducerAndConsumer {
public static void main(String[] args) {
Producer producer = new Producer();
Consumer consumer = new Consumer();
for (int i=0; i<5; i++) {
new Thread(producer).start();
new Thread(consumer).start();
}
}
private static final int CAPACITY = 10;
private static final LinkedList<Integer> list = new LinkedList<Integer>();
private static final Object obj = new Object();
static class Producer implements Runnable {
private static final AtomicInteger count = new AtomicInteger(0);
public void run() {
for (int i=0; i<10; i++)
produce();
}
private void produce() {
synchronized (obj) {
while (list.size() == CAPACITY) {
try {
obj.wait();
} catch (InterruptedException ex) {
}
}
Integer num = count.incrementAndGet();
list.add(num);
System.out.println(Thread.currentThread().getName() + "生产了 " + num);
obj.notifyAll();
}
}
}
static class Consumer implements Runnable {
public void run() {
for (int i=0; i<10; i++)
consume();
}
private void consume() {
synchronized (obj) {
while (list.size() == 0) {
try {
obj.wait();
} catch (InterruptedException ex) {
}
}
System.out.println(Thread.currentThread().getName() + "消费了 " + list.remove());
obj.notifyAll();
}
}
}
}
注意:在多生产者和多消费者的模式下,必须要使用Object.notifyAll()方法,并且当线程唤醒之后要循环判断条件是否满足,否则很容易陷入死锁的状态。
2)使用显式锁和Condition实现
public class ConditionTest {
public static void main(String[] args) {
Producer producer = new Producer();
Consumer consumer = new Consumer();
for (int i=0; i<5; i++) {
new Thread(producer).start();
new Thread(consumer).start();
}
}
private static final int CAPACITY = 1;
private static final LinkedList<Integer> list = new LinkedList<Integer>();
private static final Lock lock = new ReentrantLock();
private static final Condition notFull = lock.newCondition();
private static final Condition notEmpty = lock.newCondition();
static class Producer implements Runnable {
private static final AtomicInteger count = new AtomicInteger(0);
public void run() {
for (int i=0; i<10; i++)
produce();
}
private void produce() {
lock.lock();
try {
while (list.size() == CAPACITY) {
try {
notFull.await();
} catch (InterruptedException ex) {
}
}
Integer num = count.incrementAndGet();
list.add(num);
System.out.println(Thread.currentThread().getName() + "生产了 " + num);
notEmpty.signal();
} finally {
lock.unlock();
}
}
}
static class Consumer implements Runnable {
public void run() {
for (int i=0; i<10; i++)
consume();
}
private void consume() {
lock.lock();
try {
while (list.size() == 0) {
try {
notEmpty.await();
} catch (InterruptedException ex) {
}
}
System.out.println(Thread.currentThread().getName() + "消费了 " + list.remove());
notFull.signal();
} finally {
lock.unlock();
}
}
}
}
两种实现方式的区别
1)使用内置锁时,每个锁都只能有一个相关联的条件队列,因而在像上面的例子中,多个线程可能在同一个条件队列上等待不同的条件谓词,如:生产者线程在等待队列notFull,消费者线程在等待队列notEmpty。当条件满足时,必须使用notifyAll()方法唤醒所有等待的线程,否则程序很容易产生活跃性故障:死锁。但是,当使用notifyAll()方法唤醒所有线程后,会使得所有线程在锁上发生竞争,然后,大多数线程又都回到等待状态,因此将出现大量的上下文切换操作以及发生竞争的锁获取操作。
2)使用内置锁时,最大的问题就是:无法根据满足的条件去唤醒对应的线程,如:notEmpty时唤醒消费者线程,notFull时唤醒生产者线程。而使用Lock时,可以产生多个与之关联的条件队列,不同类型的线程将在不同的条件队列上等待,当条件满足时,可以使用signal()方法唤醒相应的线程,从而极大地减少上下文切换与锁请求的次数。
另外,Condition比Object提供了更丰富的功能:在每个锁上可存在多个等待、条件等待可以是可中断的或不可中断的、基于时限的等待,以及公平的或非公平的队列操作。Condition对象继承了相关的Lock对象的公平性,对于公平的锁,线程会依照FIFO顺序从Condition.await中释放。