You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

213 lines
13 KiB
Markdown

2 years ago
# 55 | 理解Disruptor不需要换挡和踩刹车的CPU有多快
上一讲我们学习了一个精妙的想法Disruptor通过缓存行填充来利用好CPU的高速缓存。不知道你做完课后思考题之后有没有体会到高速缓存在实践中带来的速度提升呢
不过利用CPU高速缓存只是Disruptor“快”的一个因素那今天我们就来看一看Disruptor快的另一个因素也就是“无锁”而尽可能发挥CPU本身的高速处理性能。
## 缓慢的锁
Disruptor作为一个高性能的生产者-消费者队列系统一个核心的设计就是通过RingBuffer实现一个无锁队列。
上一讲里我们讲过Java里面的基础库里就有像LinkedBlockingQueue这样的队列库。但是这个队列库比起Disruptor里用的RingBuffer要慢上很多。慢的第一个原因我们说过因为链表的数据在内存里面的布局对于高速缓存并不友好而RingBuffer所使用的数组则不然。
![](https://static001.geekbang.org/resource/image/9c/69/9ce732cb22c49a8a26e870dddde66b69.jpeg)
LinkedBlockingQueue慢有另外一个重要的因素那就是它对于锁的依赖。在生产者-消费者模式里,我们可能有多个消费者,同样也可能有多个生产者。多个生产者都要往队列的尾指针里面添加新的任务,就会产生多个线程的竞争。于是,在做这个事情的时候,生产者就需要拿到对于队列尾部的锁。同样地,在多个消费者去消费队列头的时候,也就产生竞争。同样消费者也要拿到锁。
那只有一个生产者,或者一个消费者,我们是不是就没有这个锁竞争的问题了呢?很遗憾,答案还是否定的。一般来说,在生产者-消费者模式下,消费者要比生产者快。不然的话,队列会产生积压,队列里面的任务会越堆越多。
一方面,你会发现越来越多的任务没有能够及时完成;另一方面,我们的内存也会放不下。虽然生产者-消费者模型下,我们都有一个队列来作为缓冲区,但是大部分情况下,这个缓冲区里面是空的。也就是说,即使只有一个生产者和一个消费者者,这个生产者指向的队列尾和消费者指向的队列头是同一个节点。于是,这两个生产者和消费者之间一样会产生锁竞争。
在LinkedBlockingQueue上这个锁机制是通过ReentrantLock这个Java 基础库来实现的。这个锁是一个用Java在JVM上直接实现的加锁机制这个锁机制需要由JVM来进行裁决。这个锁的争夺会把没有拿到锁的线程挂起等待也就需要经过一次上下文切换Context Switch
不知道你还记不记得,我们在[第28讲](https://time.geekbang.org/column/article/103717)讲过的异常和中断,这里的上下文切换要做的和异常和中断里的是一样的。上下文切换的过程,需要把当前执行线程的寄存器等等的信息,保存到线程栈里面。而这个过程也必然意味着,已经加载到高速缓存里面的指令或者数据,又回到了主内存里面,会进一步拖慢我们的性能。
我们可以按照Disruptor介绍资料里提到的Benchmark写一段代码来看看是不是真是这样的。这里我放了一段Java代码代码的逻辑很简单就是把一个long类型的counter从0自增到5亿。一种方式是没有任何锁另外一个方式是每次自增的时候都要去取一个锁。
你可以在自己的电脑上试试跑一下这个程序。在我这里两个方式执行所需要的时间分别是207毫秒和9603毫秒性能差出了将近50倍。
```
package com.xuwenhao.perf.jmm;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
public class LockBenchmark{
public static void runIncrement()
{
long counter = 0;
long max = 500000000L;
long start = System.currentTimeMillis();
while (counter < max) {
counter++;
}
long end = System.currentTimeMillis();
System.out.println("Time spent is " + (end-start) + "ms without lock");
}
public static void runIncrementWithLock()
{
Lock lock = new ReentrantLock();
long counter = 0;
long max = 500000000L;
long start = System.currentTimeMillis();
while (counter < max) {
if (lock.tryLock()){
counter++;
lock.unlock();
}
}
long end = System.currentTimeMillis();
System.out.println("Time spent is " + (end-start) + "ms with lock");
}
public static void main(String[] args) {
runIncrement();
runIncrementWithLock();
```
加锁和不加锁自增counter
```
Time spent is 207ms without lock
Time spent is 9603ms with lock
```
性能差出将近10倍
## 无锁的RingBuffer
加锁很慢所以Disruptor的解决方案就是“无锁”。这个“无锁”指的是没有操作系统层面的锁。实际上Disruptor还是利用了一个CPU硬件支持的指令称之为CASCompare And Swap比较和交换。在Intel CPU里面这个对应的指令就是 cmpxchg。那么下面我们就一起从Disruptor的源码到具体的硬件指令来看看这是怎么一回事儿。
Disruptor的RingBuffer是这么设计的它和直接在链表的头和尾加锁不同。Disruptor的RingBuffer创建了一个Sequence对象用来指向当前的RingBuffer的头和尾。这个头和尾的标识呢不是通过一个指针来实现的而是通过一个**序号**。这也是为什么对应源码里面的类名叫Sequence。
![](https://static001.geekbang.org/resource/image/b6/ec/b64487a7b6b45393fdfa7e2d63e176ec.jpeg)
在这个RingBuffer当中进行生产者和消费者之间的资源协调采用的是对比序号的方式。当生产者想要往队列里加入新数据的时候它会把当前的生产者的Sequence的序号加上需要加入的新数据的数量然后和实际的消费者所在的位置进行对比看看队列里是不是有足够的空间加入这些数据而不会覆盖掉消费者还没有处理完的数据。
在Sequence的代码里面就是通过compareAndSet这个方法并且最终调用到了UNSAFE.compareAndSwapLong也就是直接使用了CAS指令。
```
public boolean compareAndSet(final long expectedValue, final long newValue)
{
return UNSAFE.compareAndSwapLong(this, VALUE_OFFSET, expectedValue, newValue);
}
public long addAndGet(final long increment)
{
long currentValue;
long newValue;
do
{
currentValue = get();
newValue = currentValue + increment;
}
while (!compareAndSet(currentValue, newValue));
return newValue;
```
Sequence源码中的addAndGet如果CAS的操作没有成功它会不断忙等待地重试
这个CAS指令也就是比较和交换的操作并不是基础库里的一个函数。它也不是操作系统里面实现的一个系统调用而是**一个CPU硬件支持的机器指令**。在我们服务器所使用的Intel CPU上就是cmpxchg这个指令。
```
compxchg [ax] (隐式参数EAX累加器), [bx] (源操作数地址), [cx] (目标操作数地址)
```
cmpxchg指令一共有三个操作数第一个操作数不在指令里面出现是一个隐式的操作数也就是EAX累加寄存器里面的值。第二个操作数就是源操作数并且指令会对比这个操作数和上面的累加寄存器里面的值。
如果值是相同的那一方面CPU会把ZF也就是条件码寄存器里面零标志位的值设置为1然后再把第三个操作数也就是目标操作数设置到源操作数的地址上。如果不相等的话就会把源操作数里面的值设置到累加器寄存器里面。
我在这里放了这个逻辑对应的伪代码,你可以看一下。如果你对汇编指令、条件码寄存器这些知识点有点儿模糊了,可以回头去看看[第5](https://time.geekbang.org/column/article/93359)[讲](https://time.geekbang.org/column/article/93359)、[第6讲](https://time.geekbang.org/column/article/94075)关于汇编指令的部分。
```
IF [ax]< == [bx] THEN [ZF] = 1, [bx] = [cx]
ELSE [ZF] = 0, [ax] = [bx]
```
单个指令是原子的这也就意味着在使用CAS操作的时候我们不再需要单独进行加锁直接调用就可以了。
没有了锁CPU这部高速跑车就像在赛道上行驶不会遇到需要上下文切换这样的红灯而停下来。虽然会遇到像CAS这样复杂的机器指令就好像赛道上会有U型弯一样不过不用完全停下来等待我们CPU运行起来仍然会快很多。
那么CAS操作到底会有多快呢我们还是用一段Java代码来看一下。
```
package com.xuwenhao.perf.jmm;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
public class LockBenchmark {
public static void runIncrementAtomic()
{
AtomicLong counter = new AtomicLong(0);
long max = 500000000L;
long start = System.currentTimeMillis();
while (counter.incrementAndGet() < max) {
}
long end = System.currentTimeMillis();
System.out.println("Time spent is " + (end-start) + "ms with cas");
}
public static void main(String[] args) {
runIncrementAtomic();
}
```
```
Time spent is 3867ms with cas
```
和上面的counter自增一样只不过这一次自增我们采用了AtomicLong这个Java类。里面的incrementAndGet最终到了CPU指令层面在实现的时候用的就是CAS操作。可以看到它所花费的时间虽然要比没有任何锁的操作慢上一个数量级但是比起使用ReentrantLock这样的操作系统锁的机制还是减少了一半以上的时间。
## 总结延伸
好了咱们专栏的正文内容到今天就要结束了。今天最后一讲我带着你一起看了Disruptor代码的一个核心设计也就是它的RingBuffer是怎么做到无锁的。
Java基础库里面的BlockingQueue都需要通过显示地加锁来保障生产者之间、消费者之间乃至生产者和消费者之间不会发生锁冲突的问题。
但是加锁会大大拖慢我们的性能。在获取锁过程中CPU没有去执行计算的相关指令而要等待操作系统或者JVM来进行锁竞争的裁决。而那些没有拿到锁而被挂起等待的线程则需要进行上下文切换。这个上下文切换会把挂起线程的寄存器里的数据放到线程的程序栈里面去。这也意味着加载到高速缓存里面的数据也失效了程序就变得更慢了。
Disruptor里的RingBuffer采用了一个无锁的解决方案通过CAS这样的操作去进行序号的自增和对比使得CPU不需要获取操作系统的锁。而是能够继续顺序地执行CPU指令。没有上下文切换、没有操作系统锁自然程序就跑得快了。不过因为采用了CAS这样的忙等待Busy-Wait的方式会使得我们的CPU始终满负荷运转消耗更多的电算是一个小小的缺点。
程序里面的CAS调用映射到我们的CPU硬件层面就是一个机器指令这个指令就是cmpxchg。可以看到当想要追求最极致的性能的时候我们会从应用层、贯穿到操作系统乃至最后的CPU硬件搞清楚从高级语言到系统调用乃至最后的汇编指令这整个过程是怎么执行代码的。而这个也是学习组成原理这门专栏的意义所在。
## 推荐阅读
不知道上一讲说的Disruptor相关材料你有没有读完呢如果没有读完的话我建议你还是先去研读一下。
如果你已经读完了,这里再给你推荐一些额外的阅读材料,那就是著名的[Implement Lock-Free Queues](http://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.53.8674&rep=rep1&type=pdf)这篇论文。你可以更深入地学习一下,怎么实现一个无锁队列。
## 课后思考
最后,给你留一道思考题。这道题目有点儿难,不过也很有意思。
请你阅读一下Disruptor开源库里面的Sequence这个类的代码看看它和一个普通的AtomicLong到底有什么区别以及为什么它要这样实现。
欢迎在留言区写下你的思考和答案,和大家一起探讨应用层和硬件层之间的关联性。如果有收获,你也可以把这篇文章分享给你的朋友。