直接列代码
package com.wq.队列;
import com.sun.corba.se.spi.ior.ObjectKey;
import java.util.LinkedList;
import java.util.concurrent.atomic.AtomicInteger;
/**
* @author wangqiang
* @date 2018/12/17 19:29
*/
public class MyQueue {
//1.需要一个承装元素的集合
private final LinkedList<Object> list = new LinkedList<Object>();
//2.有界的queue, 需要计数器 统计加入元素的个数
private AtomicInteger count = new AtomicInteger(0);
//3.需要指定容器的上限和下限
private final int minSize=0;
private final int maxSize;
//4.构造方法
MyQueue(int maxSize){
this.maxSize=maxSize;
}
//5.初始化一个对象,用于加锁
private final Object lock = new Object();
//往里面加元素,如果容器满了,则阻塞
public void put(Object obj){
synchronized (lock){
//已经满了
if(count.get()==this.maxSize){
try {
lock.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
//1.加入元素
list.add(obj);
//2.计数器累加
count.incrementAndGet();
System.out.println("新加入的元素为:"+obj);
//3.通知另一个线程
lock.notify();
}
}
public Object take(){
Object ret=null;
synchronized (lock){
if (count.get()==this.minSize){
try {
lock.wait();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
//1.删除元素
ret= list.removeFirst();
//2.计数器减一
count.decrementAndGet();
// System.out.println("删除的元素为:"+ret);
//3.通知另一个线程
lock.notify();
}
return ret;
}
public int getSize(){
return this.count.get();
}
public static void main(String[] args) {
final MyQueue myQueue = new MyQueue(5);
myQueue.put("a");
myQueue.put("b");
myQueue.put("c");
myQueue.put("d");
myQueue.put("e");
System.out.println("当前容器长度:"+myQueue.getSize());
Thread t1 = new Thread(new Runnable() {
@Override
public void run() {
myQueue.put("f");
myQueue.put("g");
myQueue.put("h");
myQueue.put("i");
/*for (int i = 0; i < 20; i++) {
myQueue.put(i);
}*/
}
});
Thread t2 = new Thread(new Runnable() {
@Override
public void run() {
Object take1 = myQueue.take();
System.out.println("删除的元素为:"+take1);
Object take2 = myQueue.take();
System.out.println("删除的元素为:"+take2);
}
});
t1.start();
t2.start();
}
}