采用ConcurrentLinkedQueue实现生产者,消费者模型

生产者,消费者模型在我们实际开发过程中经常用到。就用ConcurrentLinkedQueue来实现个简单的Demo。


一、介绍

ConcurrentLinkedQueue
是单向链表结构的无界并发队列。从JDK1.7开始加入到J.U.C的行列中。使用CAS实现并发安全,元素操作按照 FIFO(first-in-first-out 先入先出) 的顺序。适合“单生产,多消费”的场景。内存一致性遵循对ConcurrentLinkedQueue的插入操作先行发生于(happen-before)访问或移除操作。
拓展下另一个相关的并发队列:
ConcurrentLinkedDeque
是双向链表结构的无界并发队列。与 ConcurrentLinkedQueue 的区别是该阻塞队列同时支持FIFO和FILO两种操作方式,即可以从队列的头和尾同时操作(插入/删除)。适合“多生产,多消费”的场景。内存一致性遵循对 ConcurrentLinkedDeque 的插入操作先行发生于(happen-before)访问或移除操作。


二、代码实现

package com.advance.mistra.solution.producerconsumer;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @Author: Mistra
 * @Version: 1.0
 * @Time: 2020/2/2 16:54
 * @Description: 生产者消费者模型,采用1个生产者,多个消费者的形式
 * @Copyright (c) Mistra,All Rights Reserved.
 * @Github: https://github.com/MistraR
 * @CSDN: https://blog.csdn.net/axela30w
 */
@Slf4j
@Component
public class ProducerConsumerTest {

    /**
     * 待处理任务队列
     */
    public static ConcurrentLinkedQueue<String> queue = new ConcurrentLinkedQueue<String>();
    /**
     * 消费者线程个数
     */
    private static int threadCount = 5;
    /**
     * 线程池
     * LinkedBlockingQueue阻塞队列,默认为Integer.MAX_VALUE  -  无界队列
     */
    private static ExecutorService executorService = new ThreadPoolExecutor(threadCount + 1, threadCount + 1,
            0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>(1000), new NameTreadFactory());

    private static volatile AtomicInteger stopProduce = new AtomicInteger(0);

    @PostConstruct
    private void init() {
        executorService.submit(new Producer());
        for (int i = 1; i <= threadCount; i++) {
            executorService.submit(new Consumer());
        }
    }

    /**
     * 生产者
     */
    static class Producer implements Runnable {

        @SneakyThrows
        @Override
        public void run() {
            while (true) {
                try {
                    // 只生产15个
                    if (stopProduce.intValue() < 15) {
                        queue.add("队列元素" + System.currentTimeMillis());
                        Thread.sleep(1000);
                        stopProduce.incrementAndGet();
                    }
                } catch (Exception e) {
                    log.info("线程发生异常!---------------------------------------------------------------------");
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 消费者
     */
    static class Consumer implements Runnable {

        @Override
        public void run() {
            while (true) {
                try {
                    if (!queue.isEmpty()) {
                        String str = queue.poll();
                        if (str != null && !str.isEmpty()) {
                            log.info("线程:{},消费:{}", Thread.currentThread().getName(), str);
                        }
                    } else {
                        Thread.sleep(100);
                    }
                } catch (Exception e) {
                    log.info("线程发生异常!---------------------------------------------------------------------");
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 自定义的线程名称类
     * 定义线程池时使用
     */
    static class NameTreadFactory implements ThreadFactory {

        private final AtomicInteger threadNum = new AtomicInteger(1);

        @Override
        public Thread newThread(Runnable r) {
            Thread thread = new Thread(r, "my-thread-" + threadNum.getAndIncrement());
            log.info("{} has been created", thread.getName());
            return thread;
        }
    }

}

生产者生产15个消息放入队列,就停止了。启动多个线程来从队列中消费消息。

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值