生产者/消费者

 

1、使用Lock

package com.atguigu.thread.blockqueue.action;


import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;


/**
 * 传统生产者与消费者
 */
public class ProdConsumer_TraditionDemo
{
    public static void main(String[] args)
    {
        ShareData shareData = new ShareData();
        new Thread(() -> {
            for (int i = 0; i < 5; i++ )
            {
                try
                {
                    shareData.inc();
                }
                catch (Exception e)
                {
                    e.printStackTrace();
                }
            }

        }, "AAA").start();
        new Thread(() -> {
            for (int i = 0; i < 5; i++ )
            {
                try
                {
                    shareData.dec();
                }
                catch (Exception e)
                {
                    e.printStackTrace();
                }
            }

        }, "BBB").start();
    }
}


class ShareData
{
    private int num = 0;

    private Lock lock = new ReentrantLock();

    private Condition condition = lock.newCondition();

    public void inc()
        throws Exception
    {
        lock.lock();
        try
        {
            //多线程判断用while
            while (num != 0)
            {
                condition.await();
            }
            num++ ;
            System.out.println(Thread.currentThread().getName() + "\t" + num);
            condition.signalAll();
        }
        finally
        {
            lock.unlock();
        }
    }

    public void dec()
        throws Exception
    {
        lock.lock();
        try
        {

            while (num == 0)
            {
                condition.await();
            }
            num-- ;
            System.out.println(Thread.currentThread().getName() + "\t" + num);
            condition.signalAll();
        }
        finally
        {
            lock.unlock();
        }
    }
}

2、使用blockingqueue

package com.atguigu.thread.blockqueue.action;


import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;


/**
 * 阻塞队列的生产/消费模型
 */
public class ProdConsumer_BlockingQueueDemo
{
    public static void main(String[] args)
        throws Exception
    {
        Mysource mysource = new Mysource(new ArrayBlockingQueue<>(10));
        new Thread(() -> {
            System.out.println(Thread.currentThread().getName() + "\t 生产线程启动");
            try
            {
                mysource.myProd();
            }
            catch (Exception e)
            {}
        }, "prod").start();
        new Thread(() -> {
            System.out.println(Thread.currentThread().getName() + "\t 消费线程启动");
            try
            {
                mysource.myConsumer();
            }
            catch (Exception e)
            {}
        }, "consumer").start();
        TimeUnit.SECONDS.sleep(3);
        mysource.stop();
    }
}


class Mysource
{
    private volatile boolean flag = true;

    private AtomicInteger atomicInteger = new AtomicInteger();

    BlockingQueue<String> blockingQueue = null;

    public Mysource(BlockingQueue<String> blockingQueue)
    {
        this.blockingQueue = blockingQueue;
    }

    // 生产
    public void myProd()
        throws Exception
    {
        String data = null;
        boolean retValue = false;
        while (flag)
        {
            data = atomicInteger.incrementAndGet() + "";
            retValue = blockingQueue.offer(data, 2L, TimeUnit.SECONDS);
            if (retValue)
            {
                System.out.println(Thread.currentThread().getName() + "\t 插入队列" + data + "成功");
            }
            else
            {
                System.out.println(Thread.currentThread().getName() + "\t 插入队列" + data + "失败");
            }
            TimeUnit.SECONDS.sleep(1);
        }
        System.out.println(Thread.currentThread().getName() + "\t 停止生产");

    }

    public void myConsumer()
        throws Exception
    {
        String result = null;
        while (flag)
        {
            result = blockingQueue.poll(2L, TimeUnit.SECONDS);
            if (null == result || "".equals(result))
            {
                flag = false;
                System.out.println(Thread.currentThread().getName() + "\t 超过2s没取到值，推出 ");
                return;
            }
            System.out.println(Thread.currentThread().getName() + "\t 消费队列" + result + "成功");
        }
        System.out.println(Thread.currentThread().getName() + "\t 停止消费");

    }

    public void stop()
        throws Exception
    {
        this.flag = false;
    }

}