c++链表实现的阻塞队列

最近从java源码里发现了阻塞队列的实现,觉得非常有趣。

首先,介绍下什么是阻塞队列。阻塞队列代表着一个队列可以线程安全的往该队列中写数据和从该队列中读数据。也就是说,我们可以在多个线程之间并发的进行写数据和读数据,而不会引发任何并发问题。

下面我们就说说如何实现一个阻塞队列。

而实现一个阻塞队列的前提:

  1. 需要能够使用链表实现一个队列
  2. 能够使用c++的锁机制,去给队列的写和读操作加锁。

为了性能,这里的读和写的锁不能是同一把锁。而对于一个链表队列来说,读取操作肯定需要涉及头指针,写操作肯定涉及尾指针。既然要实现读操作一把锁和写操作一把锁。那么就要求读操作只能更改头指针而不能更改尾指针,写操作只能更改尾指针而不能更改头指针。不满足这个要求,那么读写操作就不可能实现用两把锁分别对读写进行加锁。

基本队列的实现

现在我们先说说如何实现这个队列。

要求:入队操作(enqueue)只能操作尾指针(last), 出队操作(dequeue)只能操作头指针(head)。

对于队列的初始化,这里不能单纯的设置为空指针,需要将头尾指针同一节点。

下面我们来看入队操作如何实现

从这个入队操作来看,该操作只更改了尾指针last, 而没有更改头指针head。

其代码实现为:

    void enqueue(int item) {
Node *node = new Node(item);
last = last->next = node;
}

接下来我们来看出队操作如何实现

从出队操作来看就有趣的多,它抛弃了head所指向的节点,而这个节点有可能是那些节点呢?

  1. 初始化时所赋值的那个节点
  2. 出队后的节点

也就是说,head所指向的节点中的val值没有任何实际含义。当需要出队时,出队head指向的下一个节点first中val的值,然后抛弃head本身指向的值,让head指向head的下一个节点first,此时head原来所指向的节点将被删除。现在我们可以看出出队操作也只改变了头指针head的值。

其代码实现为:

    int dequeue() {
Node *h = head;
Node *first = head->next;
delete h;
head = first;
int x = first->item;
return x;
}

现在队列已经实现,下面就看看阻塞队列如何实现。

阻塞队列的实现

既然是阻塞队列,那就意味这加锁和等待。那就需要对c++的一些锁知识和条件变量有了解。

先来看看我们需要实现阻塞队列的那些方法:

Special Value Blocks Times out
Insert offer(o) put(o) offer(o,timeout)
Remove poll() take() poll(timeout)

入队

让我们先来实现put这个方法。

先看其实现流程图:

由于该队列实现有最大值限制,故在插入数据之前需要先判断该队列是否已满,已满则需等待该队列有可用空间。在该线程入队操作完成后,可能有别的线程也在等待入队,需要唤醒其他写数据的线程,使其继续执行后续操作。如果入队前队列为空,可能有出队操作正在阻塞等待读数,也需要去唤醒读数据的线程。

在看代码实现之前,我们需要定义一些变量用于代码实现环节:

    /* The capacity bound*/
int64_t capacity;
/*Current number of elements */
std::atomic<int64_t> count;
/** Lock held by take, pool, etc */
std::mutex takeLock;
/** Wait queue for waiting takes */
std::condition_variable notEmpty;
/** Lock held by put, offer, etc */
std::mutex putLock;
/** Wait queue for waiting puts */
std::condition_variable notFull;

put函数的代码实现:

void LinkedBlockingQueue::put(const int item){
int c;
{
std::unique_lock<std::mutex> lck{putLock};
if( count == capacity) {
notFull.wait(lck, [this](){return count < capacity;}); //(1)
}
enqueue(item); //不应该把申请空间放在锁里面,耗时有点大
c = count.fetch_add(1);
}
if(c + 1 < capacity) {
notFull.notify_one();
} if(0 == c) {
notEmpty.notify_one();
}
}

对于offer(o)的实现,主要更改是对上述代码(1)中的等待改为直接返回false, 表示,当前没有可用空间插入数据。正常插入就返回true.

对于offer(o,timeout)的实现,就需要在上述代码(1)中的wait函数添加上时间参数,使其可以在timeout时间内返回,如果是正常唤醒,正常入队,则返回ture,否则返回false.

该更改为:

            if(!notFull.wait_for(lck, rel_time, [this](){return count < capacity;})){
return false;
}

出队

对于出队,其实现和入队基本相同,基本上只需要更改其中的关键判断和通知。

take函数的代码实现为:

void LinkedBlockingQueue::take(int & returnVal) {
int c;
{
std::unique_lock<std::mutex> lck{takeLock};
if( 0 == count) {
notEmpty.wait(lck, [this](){return count > 0 ;});
}
returnVal = dequeue();
c = count.fetch_sub(1);
} if( c > 1 ) {
notEmpty.notify_one();
} if(c == capacity) {
notFull.notify_one();
}
}

剩下的实现细节可以参考我的实现

LinkBlockedQueue的c++实现的更多相关文章

  1. OO 第四单元总结

    一.总结本单元两次作业的框架设计 1.1. 需求分析 通过分析mdj文件可知,两次作业如果对于时间复杂度没有要求,可以不涉及任何数据结构,直接根据读入的UML_ELEMENT逐个分析得到各个函数的结果 ...

随机推荐

  1. ERP与EHR系统的恩怨纠葛--开源软件诞生13

    ERP中需要EHR的存在吗--第13篇 用日志记录"开源软件"的诞生 [点亮星标]----祈盼着一个鼓励 博主开源地址: 码云:https://gitee.com/redragon ...

  2. 趣图:后端工程师做 UI 的活

      扩展阅读 趣图:苦逼的后端工程师 趣图:前端 VS 后端 [趣图]开发人员腹黑的一面

  3. vulnhub-Os-hackNos-3

    vulnhub-Os-hackNos-3 nmap 192.168.114.0/24 查看存活主机,发现192.168.114.142. 具体探究: 发现80和22端口是开放的. dirb没扫出来啥, ...

  4. Java8——Stream流式操作的一点小总结

    我发现,自从我学了Stream流式操作之后,工作中使用到的频率还是挺高的,因为stream配合着lambda表达式或者双冒号(::)使用真的是优雅到了极致!今天就简单分(搬)享(运)一下我对strea ...

  5. Spring学习(九)--Spring的AOP

    1.配置ProxyFactoryBean Spring IOC容器中创建Spring AOP的方法. (1)配置ProxyFactoryBean的Advisor通知器 通知器实现定义了对目标对象进行增 ...

  6. IPV6介绍已经IPV6改造基本步骤

    IPV6介绍 地址资源无限多 通常见到的124.33.24.116这种形式的是ipv4版本的地址,这种地址由32位二进制数表示. ipv6是一种新的ip地址的表示方式形如fc80::2367:7cff ...

  7. Java知识系统回顾整理01基础05控制流程03 while

    while和do-while循环语句 一.while:条件为true时 重复执行 只要while中的表达式成立,就会不断地循环执行 public class HelloWorld { public s ...

  8. 学习Jmeter,这一篇就够了

    Jmeter在软件测试中的常用功能,看完你应该就会了 1.Jmeter的下载安装 1.1  Jmeter的官方网址是:http://jmeter.apache.org/:下载最新的包到本地进行解压,路 ...

  9. 如何快速构建React组件库

    前言 俗话说:"麻雀虽小,五脏俱全",搭建一个组件库,知之非难,行之不易,涉及到的技术方方面面,犹如海面风平浪静,实则暗礁险滩,处处惊险- 目前团队内已经有较为成熟的 Vue 技术 ...

  10. thinkphp5 chunk 分块处理数据小坑

    场景: 使用chunk方法进行分块查询写入数据,执行发现chunk分几条一次处理 数据库就插入几条,并没有return false; 代码如下 解决方法: 增加排序字段