disruptor(二) MultiProducerSequencer
程序员文章站
2022-07-12 18:32:54
...
在《disruptor(一) 单一生产者和WorkPool消费者源码阅读》介绍了单一生产者
当多个生产者向RingBuffer中写入数据时,创建Disruptor时要修改对应的参数:
Disruptor disruptor = new Disruptor(eventFactory, BUFFER_SIZE, executor, ProducerType.MULTI, new YieldingWaitStrategy ());
ProducerType.MULTI 使用这个参数,单一生产者时,Sequencer接口的实现是SingleProducerSequencer;当是多个线程写入时,使用另一个实现MultiProducerSequencer
在写入获取RingBuffer的序列号的实现:
/** * @see Sequencer#next(int) */ @Override public long next(int n) { if (n < 1) { throw new IllegalArgumentException("n must be > 0"); } //生产者当前写入到的序列号 long current; //下一个序列号 long next; //while循环主要服务CAS算法,不成功就重来 do { //获取生产者最新的序列号 current = cursor.get(); //要获取的序列号 next = current + n; //wrapPoint是一个很关键的变量,这个变量决定生产者是否可以覆盖序列号nextSequence,wrapPoint是为什么是nextSequence - bufferSize;RingBuffer表现出来的是一个环形的数据结构,实际上是一个长度为bufferSize的数组, //nextSequence - bufferSize如果nextSequence小于bufferSize wrapPoint是负数,表示可以一直生产;如果nextSequence大于bufferSize wrapPoint是一个大于0的数,由于生产者和消费者的序列号差距不能超过bufferSize //(超过bufferSize会覆盖消费者未消费的数据),wrapPoint要小于等于多个消费者线程中消费的最小的序列号,即cachedGatingSequence的值,这就是下面if判断的根据 long wrapPoint = next - bufferSize; //cachedGatingSequence, gatingSequenceCache这两个变量记录着上一次获取消费者中最小的消费序列号 long cachedGatingSequence = gatingSequenceCache.get(); //生产者不能继续写入,否则会覆盖消费者未消费的数据 if (wrapPoint > cachedGatingSequence || cachedGatingSequence > current) { //获取最新的消费者最小的消费序号 long gatingSequence = Util.getMinimumSequence(gatingSequences, current); //依然不能满足写入条件(写入会覆盖为消费的数据) if (wrapPoint > gatingSequence) { //锁一会,结束本次循环,重来 LockSupport.parkNanos(1); // TODO, should we spin based on the wait strategy? continue; } //缓存一下消费者中最小的消费序列号 gatingSequenceCache.set(gatingSequence); } else if (cursor.compareAndSet(current, next)) //满足消费条件,有空余的空间让生产者写入,使用CAS算法,成功则跳出本次循环,不成功则重来 { break; } } while (true); return next; }
MultiProducerSequencer和SingleProducerSequencer next方法的区别就在于多了一个CAS算法,获取消费者最小序列号的while循环和放到外面和CAS的while循环合并
下一篇: Function 对代码封装带来的改变