C++原子性实现无锁队列

关于无锁队列的实现&＃xff0c;网上有很多文章&＃xff0c;虽然本文可能和那些文章有所重复&＃xff0c;但是我还是想以我自己的方式把这些文章中的重要的知识点串起来和大家讲一讲这个技术。下面开始正文。

关于CAS等原子操作

在开始说无锁队列之前&＃xff0c;我们需要知道一个很重要的技术就是CAS操作——Compare & Set或是 Compare & Swap&＃xff0c;现在几乎所有的CPU指令都支持CAS的原子操作&＃xff0c;X86下对应的是 CMPXCHG 汇编指令。有了这个原子操作&＃xff0c;我们就可以用其来实现各种无锁&＃xff08;lock free&＃xff09;的数据结构。

这个操作用C语言来描述就是下面这个样子&＃xff1a;&＃xff08;代码来自Wikipedia的Compare And Swap词条&＃xff09;意思就是说&＃xff0c;看一看内存*reg里的值是不是oldval&＃xff0c;如果是的话&＃xff0c;则对其赋值newval。

int compare_and_swap (int* reg, int oldval, int newval)
{int old_reg_val &＃61; *reg;if (old_reg_val &＃61;&＃61; oldval)*reg &＃61; newval;return old_reg_val;
}

bool compare_and_swap (int *accum, int *dest, int newval)
{if ( *accum &＃61;&＃61; *dest ) {*dest &＃61; newval;return true;}return false;
}

这个操作可以变种为返回bool值的形式&＃xff08;返回 bool值的好处在于&＃xff0c;可以调用者知道有没有更新成功&＃xff09;&＃xff1a;

CAS的语义是“我认为V的值应该为A&＃xff0c;如果是&＃xff0c;那么将V的值更新为B&＃xff0c;否则不修改并告诉V的值实际为多少”&＃xff0c;CAS是项乐观锁技术&＃xff0c;当多个线程尝试使用CAS同时更新同一个变量时&＃xff0c;只有其中一个线程能更新变量的值&＃xff0c;而其它线程都失败&＃xff0c;失败的线程并不会被挂起&＃xff0c;而是被告知这次竞争中失败&＃xff0c;并可以再次尝试。CAS有3个操作数&＃xff0c;内存值V&＃xff0c;旧的预期值A&＃xff0c;要修改的新值B。当且仅当预期值A和内存值V相同时&＃xff0c;将内存值V修改为B&＃xff0c;否则什么都不做。

另一种表述&＃xff1a;CAS操作需要输入两个数值&＃xff0c;一个旧值&＃xff08;期望操作前的值&＃xff09;和一个新值&＃xff0c;在操作期间先比较下旧值有没有发生变化&＃xff0c;如果没有发生变化&＃xff0c;才交换成新值&＃xff0c;发生了变化则不交换。

与CAS相似的还有下面的原子操作&＃xff1a;&＃xff08;这些东西大家自己看Wikipedia吧&＃xff09;

• Fetch And Add&＃xff0c;一般用来对变量做 &＃43;1 的原子操作
• Test-and-set&＃xff0c;写值到某个内存位置并传回其旧值。汇编指令BST
• Test and Test-and-set&＃xff0c;用来低低Test-and-Set的资源争夺情况

注&＃xff1a;在实际的C/C&＃43;&＃43;程序中&＃xff0c;CAS的各种实现版本如下&＃xff1a;

1&＃xff09;GCC的CAS

GCC4.1&＃43;版本中支持CAS的原子操作&＃xff08;完整的原子操作可参看 GCC Atomic Builtins&＃xff09;

bool __sync_bool_compare_and_swap (type *ptr, type oldval type newval, ...)
type __sync_val_compare_and_swap (type *ptr, type oldval type newval, ...)

2&＃xff09;Windows的CAS

在Windows下&＃xff0c;你可以使用下面的Windows API来完成CAS&＃xff1a;&＃xff08;完整的Windows原子操作可参看MSDN的InterLocked Functions&＃xff09;

InterlockedCompareExchange ( __inout LONG volatile  *Target,__in LONG Exchange,__in LONG Comperand);

3) C&＃43;&＃43;11中的CAS

C&＃43;&＃43;11中的STL中的atomic类的函数可以让你跨平台。&＃xff08;完整的C&＃43;&＃43;11的原子操作可参看 Atomic Operation Library&＃xff09;

template< class T >
bool atomic_compare_exchange_weak( std::atomic* obj,T* expected, T desired );
template< class T >
bool atomic_compare_exchange_weak( volatile std::atomic* obj,T* expected, T desired );

下面的东西主要来自John D. Valois 1994年10月在拉斯维加斯的并行和分布系统系统国际大会上的一篇论文——《Implementing Lock-Free Queues》。无锁队列的链表实现

我们先来看一下进队列用CAS实现的方式&＃xff1a;

EnQueue(x)//进队列
{//准备新加入的结点数据q &＃61; new record();q->value &＃61; x;q->next &＃61; NULL;do {p &＃61; tail;//取链表尾指针的快照} while( CAS(p->next, NULL, q) !&＃61; TRUE);//如果没有把结点链在尾指针上&＃xff0c;再试CAS(tail, p, q);//置尾结点
}

我们可以看到&＃xff0c;程序中的那个 do- while 的 Re-Try-Loop。就是说&＃xff0c;很有可能我在准备在队列尾加入结点时&＃xff0c;别的线程已经加成功了&＃xff0c;于是tail指针就变了&＃xff0c;于是我的CAS返回了false&＃xff0c;于是程序再试&＃xff0c;直到试成功为止。这个很像我们的抢电话热线的不停重播的情况。

你会看到&＃xff0c;为什么我们的“置尾结点”的操作&＃xff08;第12行&＃xff09;不判断是否成功&＃xff0c;因为&＃xff1a;

1. 如果有一个线程T1&＃xff0c;它的while中的CAS如果成功的话&＃xff0c;那么其它所有的 随后线程的CAS都会失败&＃xff0c;然后就会再循环&＃xff0c;
2. 此时&＃xff0c;如果T1 线程还没有更新tail指针&＃xff0c;其它的线程继续失败&＃xff0c;因为tail->next不是NULL了。
3. 直到T1线程更新完tail指针&＃xff0c;于是其它的线程中的某个线程就可以得到新的tail指针&＃xff0c;继续往下走了。

这里有一个潜在的问题——如果T1线程在用CAS更新tail指针的之前&＃xff0c;线程停掉或是挂掉了&＃xff0c;那么其它线程就进入死循环了。下面是改良版的EnQueue()

EnQueue(x)//进队列改良版
{q &＃61; new record();q->value &＃61; x;q->next &＃61; NULL;p &＃61; tail;oldp &＃61; pdo {while (p->next !&＃61; NULL)p &＃61; p->next;} while( CAS(p.next, NULL, q) !&＃61; TRUE);//如果没有把结点链在尾上&＃xff0c;再试CAS(tail, oldp, q);//置尾结点
}

我们让每个线程&＃xff0c;自己fetch 指针 p 到链表尾。但是这样的fetch会很影响性能。而通实际情况看下来&＃xff0c;99.9%的情况不会有线程停转的情况&＃xff0c;所以&＃xff0c;更好的做法是&＃xff0c;你可以接合上述的这两个版本&＃xff0c;如果retry的次数超了一个值的话&＃xff08;比如说3次&＃xff09;&＃xff0c;那么&＃xff0c;就自己fetch指针。

好了&＃xff0c;我们解决了EnQueue&＃xff0c;我们再来看看DeQueue的代码&＃xff1a;&＃xff08;很简单&＃xff0c;我就不解释了&＃xff09;

DeQueue()//出队列
{do{p &＃61; head;if (p->next &＃61;&＃61; NULL){return ERR_EMPTY_QUEUE;}while( CAS(head, p, p->next) !&＃61; TRUE );return p->next->value;
}

我们可以看到&＃xff0c;DeQueue的代码操作的是 head->next&＃xff0c;而不是head本身。这样考虑是因为一个边界条件&＃xff0c;我们需要一个dummy的头指针来解决链表中如果只有一个元素&＃xff0c;head和tail都指向同一个结点的问题&＃xff0c;这样EnQueue和DeQueue要互相排斥了。

注&＃xff1a;上图的tail正处于更新之前的装态。

CAS的ABA问题

所谓ABA&＃xff08;见维基百科的ABA词条&＃xff09;&＃xff0c;问题基本是这个样子&＃xff1a;

1. 进程P1在共享变量中读到值为A
2. P1被抢占了&＃xff0c;进程P2执行
3. P2把共享变量里的值从A改成了B&＃xff0c;再改回到A&＃xff0c;此时被P1抢占。
4. P1回来看到共享变量里的值没有被改变&＃xff0c;于是继续执行。

虽然P1以为变量值没有改变&＃xff0c;继续执行了&＃xff0c;但是这个会引发一些潜在的问题。ABA问题最容易发生在lock free 的算法中的&＃xff0c;CAS首当其冲&＃xff0c;因为CAS判断的是指针的地址。如果这个地址被重用了呢&＃xff0c;问题就很大了。&＃xff08;地址被重用是很经常发生的&＃xff0c;一个内存分配后释放了&＃xff0c;再分配&＃xff0c;很有可能还是原来的地址&＃xff09;

比如上述的DeQueue()函数&＃xff0c;因为我们要让head和tail分开&＃xff0c;所以我们引入了一个dummy指针给head&＃xff0c;当我们做CAS的之前&＃xff0c;如果head的那块内存被回收并被重用了&＃xff0c;而重用的内存又被EnQueue()进来了&＃xff0c;这会有很大的问题。&＃xff08;内存管理中重用内存基本上是一种很常见的行为&＃xff09;

这个例子你可能没有看懂&＃xff0c;维基百科上给了一个活生生的例子——

你拿着一个装满钱的手提箱在飞机场&＃xff0c;此时过来了一个火辣性感的美女&＃xff0c;然后她很暖昧地挑逗着你&＃xff0c;并趁你不注意的时候&＃xff0c;把用一个一模一样的手提箱和你那装满钱的箱子调了个包&＃xff0c;然后就离开了&＃xff0c;你看到你的手提箱还在那&＃xff0c;于是就提着手提箱去赶飞机去了。

这就是ABA的问题。

解决ABA的问题

维基百科上给了一个解——使用double-CAS&＃xff08;双保险的CAS&＃xff09;&＃xff0c;例如&＃xff0c;在32位系统上&＃xff0c;我们要检查64位的内容

1&＃xff09;一次用CAS检查双倍长度的值&＃xff0c;前半部是指针&＃xff0c;后半部分是一个计数器。

2&＃xff09;只有这两个都一样&＃xff0c;才算通过检查&＃xff0c;要吧赋新的值。并把计数器累加1。

这样一来&＃xff0c;ABA发生时&＃xff0c;虽然值一样&＃xff0c;但是计数器就不一样&＃xff08;但是在32位的系统上&＃xff0c;这个计数器会溢出回来又从1开始的&＃xff0c;这还是会有ABA的问题&＃xff09;

当然&＃xff0c;我们这个队列的问题就是不想让那个内存重用&＃xff0c;这样明确的业务问题比较好解决&＃xff0c;论文《Implementing Lock-Free Queues》给出一这么一个方法——使用结点内存引用计数refcnt&＃xff01;

SafeRead(q)
{loop:p &＃61; q->next;if (p &＃61;&＃61; NULL){return p;}Fetch&Add(p->refcnt, 1);if (p &＃61;&＃61; q->next){return p;}else{Release(p);}goto loop;
}

其中的 Fetch&Add和Release分是是加引用计数和减引用计数&＃xff0c;都是原子操作&＃xff0c;这样就可以阻止内存被回收了。

用数组实现无锁队列

本实现来自论文《Implementing Lock-Free Queues》

使用数组来实现队列是很常见的方法&＃xff0c;因为没有内存的分部和释放&＃xff0c;一切都会变得简单&＃xff0c;实现的思路如下&＃xff1a;

1&＃xff09;数组队列应该是一个ring buffer形式的数组&＃xff08;环形数组&＃xff09;

2&＃xff09;数组的元素应该有三个可能的值&＃xff1a;HEAD&＃xff0c;TAIL&＃xff0c;EMPTY&＃xff08;当然&＃xff0c;还有实际的数据&＃xff09;

3&＃xff09;数组一开始全部初始化成EMPTY&＃xff0c;有两个相邻的元素要初始化成HEAD和TAIL&＃xff0c;这代表空队列。

4&＃xff09;EnQueue操作。假设数据x要入队列&＃xff0c;定位TAIL的位置&＃xff0c;使用double-CAS方法把(TAIL, EMPTY) 更新成 (x, TAIL)。需要注意&＃xff0c;如果找不到(TAIL, EMPTY)&＃xff0c;则说明队列满了。

5&＃xff09;DeQueue操作。定位HEAD的位置&＃xff0c;把(HEAD, x)更新成(EMPTY, HEAD)&＃xff0c;并把x返回。同样需要注意&＃xff0c;如果x是TAIL&＃xff0c;则说明队列为空。

算法的一个关键是——如何定位HEAD或TAIL&＃xff1f;

1&＃xff09;我们可以声明两个计数器&＃xff0c;一个用来计数EnQueue的次数&＃xff0c;一个用来计数DeQueue的次数。

2&＃xff09;这两个计算器使用使用Fetch&ADD来进行原子累加&＃xff0c;在EnQueue或DeQueue完成的时候累加就好了。

3&＃xff09;累加后求个模什么的就可以知道TAIL和HEAD的位置了。

如下图所示&＃xff1a;

 小结

以上基本上就是所有的无锁队列的技术细节&＃xff0c;这些技术都可以用在其它的无锁数据结构上。

1&＃xff09;无锁队列主要是通过CAS、FAA这些原子操作&＃xff0c;和Retry-Loop实现。

2&＃xff09;对于Retry-Loop&＃xff0c;我个人感觉其实和锁什么什么两样。只是这种“锁”的粒度变小了&＃xff0c;主要是“锁”HEAD和TAIL这两个关键资源。而不是整个数据结构。

还有一些和Lock Free的文章你可以去看看&＃xff1a;

• Code Project 上的雄文 《Yet another implementation of a lock-free circular array queue》
• Herb Sutter的《Writing Lock-Free Code: A Corrected Queue》– 用C&＃43;&＃43;11的std::atomic模板。
• IBM developerWorks的《设计不使用互斥锁的并发数据结构》

【注&＃xff1a;一张look-free的自行车&＃xff0c;寓意为——如果不用专门的车锁&＃xff0c;那么自行得自己锁自己&＃xff01;】