Disruptor多个消费者不重复处理生产者发送的消息的demo

上一篇介绍的一个生产者向ringbuffer发送了10条消息，每个消费者都会把这个10个消息消费一遍，但是我们还有一种需求就是要让多个消费者不重复消费消息，下面这个简单demo实现了此功能：

创建一个消息的生产者类：

package Disruptor2;import com.lmax.disruptor.RingBuffer;import Disruptor.TradeTransaction;public class Producer {private RingBuffer<TradeTransaction> ringBuffer;public Producer(RingBuffer<TradeTransaction> ringBuffer) {this.ringBuffer = ringBuffer;}public void onData() {// 可以把ringBuffer看做一个事件队列，那么next就是得到下面一个事件槽long sequence = ringBuffer.next();try {} finally {System.out.println("生产者发送了一条消息");ringBuffer.publish(sequence);}}}

创建一个消息的消费者类：

package Disruptor2;import com.lmax.disruptor.WorkHandler;import Disruptor.TradeTransaction;public class Consumer implements WorkHandler<TradeTransaction>{@Overridepublic void onEvent(TradeTransaction event) throws Exception {// TODO Auto-generated method stubSystem.out.println("消费者C1消费了一条消息");}}

创建一个IntEventExceptionHandler类：

package Disruptor2;import org.apache.log4j.Logger;import com.lmax.disruptor.ExceptionHandler;public class IntEventExceptionHandler implements ExceptionHandler {private static final Logger logger = Logger.getLogger(IntEventExceptionHandler.class);public void handleEventException(Throwable ex, long sequence, Object event) {logger.error("handleEventException", ex);}public void handleOnStartException(Throwable ex) {logger.error("handleOnStartException", ex);}public void handleOnShutdownException(Throwable ex) {logger.error("handleOnShutdownException", ex);}}

编写一个测试类：

package Disruptor2;import java.util.concurrent.ExecutorService;import java.util.concurrent.Executors;import com.lmax.disruptor.EventFactory;import com.lmax.disruptor.RingBuffer;import com.lmax.disruptor.SequenceBarrier;import com.lmax.disruptor.WaitStrategy;import com.lmax.disruptor.WorkerPool;import com.lmax.disruptor.YieldingWaitStrategy;import com.lmax.disruptor.dsl.ProducerType;import Disruptor.TradeTransaction;public class Disruptor2 {public static void main(String[] args) {Long time = System.currentTimeMillis();RingBuffer<TradeTransaction> ringBuffer;Producer producer = null;// 创建缓冲池ExecutorService executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());// 创建工厂EventFactory<TradeTransaction> factory = new EventFactory<TradeTransaction>() {@Overridepublic TradeTransaction newInstance() {return new TradeTransaction();}};// 创建bufferSize ,也就是RingBuffer大小，必须是2的N次方int ringBufferSize = 1024 * 1024; //WaitStrategy YIELDING_WAIT = new YieldingWaitStrategy();// 创建ringBufferringBuffer = RingBuffer.create(ProducerType.MULTI, factory, ringBufferSize, YIELDING_WAIT);SequenceBarrier barriers = ringBuffer.newBarrier();// 创建10个消费者来处理同一个生产者发的消息(这10个消费者不重复消费消息)Consumer[] consumers = new Consumer[10];for (int i = 0; i < consumers.length; i++) {consumers[i] = new Consumer();}WorkerPool<TradeTransaction> workerPool = new WorkerPool<TradeTransaction>(ringBuffer, barriers,new IntEventExceptionHandler(), consumers);ringBuffer.addGatingSequences(workerPool.getWorkerSequences());workerPool.start(executor);producer = new Producer(ringBuffer);for (int i = 0; i < 10; i++) {producer.onData();}//executor.shutdown();System.out.println("花费时间 :" + (System.currentTimeMillis() - time));}}

测试结果如下：

生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
生产者发送了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
消费者C1消费了一条消息
花费时间 :146
消费者C1消费了一条消息

由结果可以看出来，一共生成了10个消费者用于消费生产者产生的消息，每一条消息只消费了一次