複製代碼代碼如下:
package condition;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
/**
* 利用Lock、Condition實現生產者消費者模式
* @author will
*
*/
public class ProducerConsumerDemo {
public static void main(String[] args) {
int producerCount = 10;
int consumerCount = 15;
final ProducerConsumerDemo pcd = new ProducerConsumerDemo(5); // 緩衝區大小為5
Thread[] producerThreads = new Thread[producerCount];
for(int i = 0; i < producerCount; i++) {
producerThreads[i] = new Thread("producer" + (i+1)) {
@Override
public void run() {
pcd.produce();
}
};
}
Thread[] consumerThreads = new Thread[consumerCount];
for(int j = 0; j < consumerCount; j++) {
consumerThreads[j] = new Thread("consumer" + (j+1)) {
@Override
public void run() {
pcd.consume();
}
};
}
// 啟動生產者消費者線程
for(int i = 0; i < producerCount; i++) {
producerThreads[i].start();
}
for(int j = 0; j < consumerCount; j++) {
consumerThreads[j].start();
}
}
private static final int DEFAULT_BUFFER_SIZE = 10;
private int bufferSize; // 緩衝區大小
private List<Object> bufferList;
private final Lock lock = new ReentrantLock(true);
private final Condition condition = lock.newCondition();
public ProducerConsumerDemo(int bufferSize) {
this.bufferSize = bufferSize > 0 ? bufferSize : DEFAULT_BUFFER_SIZE;
bufferList = new ArrayList<Object>(bufferSize);
}
// 生產
public void produce() {
lock.lock(); // 加鎖
try {
while(bufferList.size() == bufferSize) { // 緩衝區滿了
System.out.println("Producer wait, thread: " + Thread.currentThread().getName());
condition.await();
}
// 生產
bufferList.add(new Object());
System.out.println("Producer produce one, now buffer size: "
+ bufferList.size() + ", and thread: " + Thread.currentThread().getName());
condition.signalAll(); // 通知消費者
} catch(InterruptedException e) {
e.printStackTrace();
} finally {
lock.unlock();
}
}
// 消費
public void consume() {
lock.lock(); // 加鎖
try {
while(bufferList.isEmpty()) { // 緩衝區空了
System.out.println("Consumer wait, thread: " + Thread.currentThread().getName());
condition.await();
}
// 消費
bufferList.remove(0); // 從鍊錶頭移除一個
System.out.println("Consumer consumer one, now buffer size: "
+ bufferList.size() + ", and thread: " + Thread.currentThread().getName());
condition.signalAll();
} catch(InterruptedException e) {
e.printStackTrace();
} finally {
lock.unlock();
}
}
}