手写一个简单的MQ消息队列

package com.bjsxt.base.conn009;

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

public class MyQueue1 {
    //创建一个装队列的容器
    private LinkedList<Object> list = new LinkedList<>();
    //创建一个计数器
    private AtomicInteger count = new AtomicInteger(0);
    //创建容器的最大值和最小值
    private final int minSize = 0;
    
    private final int maxSize ;
    
    public MyQueue1(int size){
        this.maxSize = size;
    }
    
    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();
                }
            }
            //把对象加入容器
            list.add(obj);
            //计算容器的数值
            count.incrementAndGet();
            //唤醒等待的线程
            lock.notify();
            System.out.println("新加入的元素为:" + obj);
        }
    }
    
    public Object take(){
        Object ret =null;
        synchronized (lock) {
            if (count.get() == this.minSize) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            ret = list.removeFirst();
            count.decrementAndGet();
            lock.notify();
        }
        
        return ret;
    }
    
    
    public int Size(){
        return this.count.get();
    }
    
    public static void main(String[] args) {
        final MyQueue1 mq = new MyQueue1(5);
        mq.put(1);
        mq.put(2);    
        mq.put(3);    
        mq.put(4);    
        mq.put(5);    
        
        System.out.println("当前容器的长度:" + mq.Size());
        
        Thread t1 = new Thread(new Runnable() {
            @Override
            public void run() {
                mq.put(6);
                mq.put(7);    
            }
        },"t1");
        
        t1.start();
        
        try {
            TimeUnit.SECONDS.sleep(2);
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        Thread t2 = new Thread(new Runnable() {
            @Override
            public void run() {
                Object o1 = mq.take();
                System.out.println("移除的元素为:" + o1);
                Object o2 = mq.take();
                System.out.println("移除的元素为:" + o2);
            }
        },"t2");
        t2.start();
        
    }
}
 

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值