消费者生产者队列BlockQueue

8月的笔记


import java.util.LinkedList;
import java.util.concurrent.atomic.AtomicInteger;

public class MyBlockQueue {
    //裝載元素的集合
    private final LinkedList<Object> list = new LinkedList<Object>();

    //計數器
    private final AtomicInteger count = new AtomicInteger(0);

    //隊列的大小,上下限制
    private final int minSize = 0;
    private final int maxSize;

    public MyBlockQueue(int maxSize){
        this.maxSize = maxSize;
    }

    private final Object lock = new Object();
    //put(Object)  入隊操作,阻塞加鎖;若隊列已滿,鎖一直等待,有空位才能入隊
    public void put(Object element){
        synchronized (lock) {
            if (count.get() == this.maxSize) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            //入隊
            list.add(element);
            //計數
            count.incrementAndGet();
            System.out.println("新加入元素: " + element);
            //通知消費者消費
            lock.notify();
        }
    }

    //take(Object) 出隊操作,阻塞加鎖;若隊列空,鎖等待,有元素才能出隊
    public Object take() {
        Object take = null;
        synchronized (lock) {
            if (count.get() == this.minSize) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

            //出隊
            take = list.removeFirst();
            //計數
            count.decrementAndGet();
            System.out.println("移除元素");
            //通知生產者放元素
            lock.notify();
        }
        return take;
    }

    public int getSize(){
        return this.count.get();
    }
}

同樣,我們可以使用ReentranLock,並配合try…finally語句塊來實現

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值