怎么在Java中使用wait和notifyAll實現阻塞隊列

本篇文章給大家分享的是有關怎么在Java中使用wait和notifyAll實現阻塞隊列，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

import java.util.concurrent.atomic.AtomicInteger;
/**
 * @author lhd
 */
public class BlockQueue {
	/**
  * 生產者鎖對象
  */
	private final Object addLock = new Object();
	/**
  * 消費者鎖對象
  */
	private final Object deleteLock = new Object();
	/**
  * 隊列總大小
  */
	private final Integer size = 30;
	/**
  * 數據存放
  */
	private Object[] queue = new Object[size];
	/**
  * 存放的數量,使用AtomicInteger是因為普通的int遞增遞減操作會存在非原子性的問題，會使數量異常
  */
	private AtomicInteger count = new AtomicInteger(0);
	/**
  * 生產
  * @param o 對象
  */
	public void add(Object o) {
		//獲取生產鎖,wait方法必須獲取到對象鎖后才可以調用，否則拋出異常
		synchronized (addLock){
			//判斷是否超過隊列大小,超過則進入等待
			while (count.get() >= size){
				try {
					addLock.wait();
				}
				catch (InterruptedException e) {
					e.printStackTrace();
				}
			}
			//存放一個
			queue[count.get()] = o;
			//遞增
			int i = count.incrementAndGet();
			//打印一下日志
			String name = Thread.currentThread().getName();
			System.out.println(name + "生產了一個，現有數量" + i);
		}
		//如果隊列有數據，則調用notifyAll喚醒消費者
		if (count.get() >= 1){
			//notifyAll、notify都需要先獲取對象鎖,否則會拋出異常
			synchronized (deleteLock){
				deleteLock.notifyAll();
			}
		}
	}
	/**
  * 消費
  * @return
  */
	public Object poll(){
		Object o;
		//先獲取對象鎖，和生產者類似
		synchronized (deleteLock){
			//隊列里沒有數據則等待
			while (count.get() <= 0){
				try {
					deleteLock.wait();
				}
				catch (InterruptedException e) {
					e.printStackTrace();
				}
			}
			//獲取數據
			o = queue[count.get()];
			//遞減
			int i = count.decrementAndGet();
			String name = Thread.currentThread().getName();
			System.out.println(name + "消費了一個，現有數量" + i);
		}
		//如果隊列沒有滿，則可以喚醒生產者
		if (count.get() < size){
			//需要先獲取到鎖
			synchronized (addLock){
				addLock.notifyAll();
			}
		}
		return o;
	}
	/**
  * 簡單的測試
  * @param args
  */
	public static void main(String[] args) {
		BlockQueue blockQueue = new BlockQueue();
		Thread t1 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t2 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t3 = new Thread(()-> {
			while (true){
				blockQueue.add(new Object());
			}
		}
		);
		Thread t4 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		Thread t5 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		Thread t6 = new Thread(()-> {
			while (true){
				blockQueue.poll();
			}
		}
		);
		t1.start();
		t2.start();
		t3.start();
		t4.start();
		t5.start();
		t6.start();
	}
}

效果：其實這個遞增遞減操作和打印操作也不是原子操作

依次打印線程1,2,3

/**
 * @author lhd
 */
public class JoinTest {


 public static void main(String[] args) throws InterruptedException {
  Thread t1 = new Thread(() -> System.out.println(1));
  Thread t2 = new Thread(()-> System.out.println(2));
  Thread t3 = new Thread(()-> System.out.println(3));

  t1.start();
  t1.join();

  t2.start();
  t2.join();

  t3.start();
  t3.join();
 }
}

以上就是怎么在Java中使用wait和notifyAll實現阻塞隊列，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。