有这样一个经典场景:我们定义一个对象obj,该对象有一个int类型的属性i.有一组线程,他们将obj对象的i由0变为1(称为生产者),另外有一组线程,他们将obj对象的值由1变为0(称为消费者)。我们在内存中看到obj的i值应该是0-1-0-1…这样变化的,不允许出现i值变为其他的值,比如(-1,2)。
一般情况下,我们可以通过synchronized加Object的wait()和notify()方法实现,核心代码为:
synchronized (obj) {
while (i == 1) {
try {
obj.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
if (i == 0) {
1 = 1;
obj.notifyAll();
}
}
有两点需要注意:
1、如果只有一个生产者和一个消费者时,通知方法可以用obj.notify(),但如果有多个生产者或消费者时,就需要用notifyAll()。notify()方法为唤醒任意一个线程,notifyAll为唤醒所有线程。在多线程的情况下,一个消费者执行了notify之后如果唤醒的恰好又是一个消费者,那么后者会执行时进入到wait方法,这样的话所有的线程都进入了waited状态,永远不会被唤醒了。 2、判断要写成while,不能用if。因为不能确定被唤醒时当前是应该消费者执行还是生产者执行,所以需要再判断一次i的值。
以上实现有存在一个问题,我们在执行notifyAll时唤醒的是所有试图获取obj锁的线程,这样的话所有的生产者和消费者都会试图抢占锁,比如一个消费者执行完之后应该轮到一个生产者去执行,但是如果恰好是另外一个消费者抢占到了锁,那么它进去之后会先休眠退出,再由其他线程去抢占锁资源。这样的话就造成了一部分的资源浪费。我们有没有一种机制在生产者执行完之后通知消费者去执行,而不是通知所有的线程呢。 要实现上送的需求,需要使用到JUC(java.util.concurrent)的Lock和Condition ,关键代码如下:
ReentrantLock lock = new ReentrantLock();
Condition pCondition = lock.newCondition(); //生产者条件锁
Condition cCondition = lock.newCondition();//消费者条件锁
.... public void producer() {
try {
lock.lock();
while(i == 1) {
pCondition.await();
}
if(i == 0) {
i++;
}
cCondition.signalAll();
}catch (Exception e)
{
e.printStackTrace();
}finally {
lock.unlock();
}
}
从上面的代码可以看出,这段代码和之前的使用synchronized机制实现的大同小异,最大的区别在于对于一个lock我们可以定义多个唤醒条件。