java使用BlockingQueue实现生产者、消费者业务

生产者代码：

/**
 * 生产者
 */
public class Producer implements Runnable{
    private BlockingQueue<String> queue;
    private List<String> list;
    public volatile boolean isRunning = true;//生产者的运行状态

    /**
     * 重写构造方法
     */
    public Producer(BlockingQueue<String> queue,List<String> list){
            this.queue=queue;
            this.list=list;
    }

    @Override
    public void run() {
            for(int i=0;i<this.list.size();i++){
                this.queue.add(this.list.get(i));
            }
            this.isRunning=false;//当把数据全部放到队列后，修改生产者的运行状态
    }
}

消费者代码：

**
 * 消费者
 */
public class Consumer implements Runnable{
    private BlockingQueue<String> queue;
    private Producer producer;//把生产者弄进来 防止消费者比生产者提前关闭

    /**
     * 重写构造方法
     */
    public Consumer(BlockingQueue<String> queue,Producer producer){
        this.queue=queue;
        this.producer=producer;
    }



    @Override
    public void run() {
        try {
            while (!queue.isEmpty() || producer.isRunning){
                String value = queue.take();
                /**
                 * 执行你需要的业务操作
                 */
                Thread.sleep(1000);
            }
            System.out.println("消费完毕");
        }catch (Exception e){
            System.out.println("出错了");
        }
    }
}

调用生产者和消费者：

import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;

public class MyQueue {

    public void test(List<String> list){
        if(list!=null && !list.isEmpty()){
            //创建一个队列 线程安全的
            BlockingQueue<String> queue = new LinkedBlockingQueue<String>(list.size());

            //创建生产者
            Producer producer = new Producer(queue,list);
            //创建消费者
            Consumer consumer = new Consumer(queue,producer);

            // 创建线程池 启动任务
            ExecutorService service = Executors.newCachedThreadPool();
            service.execute(producer);
            service.execute(consumer);

            service.shutdown();//关闭线程池 不再接受新任务
        }
    }
}

示例中用到的数据载体为List<String>类型的，可以根据实际需求更改数据类型

java使用BlockingQueue实现生产者、消费者业务

猜你喜欢