上一篇介绍的一个生产者向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>{
@Override
public void onEvent(TradeTransaction event) throws Exception {
// TODO Auto-generated method stub
System.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>() {
@Override
public TradeTransaction newInstance() {
return new TradeTransaction();
}
};
// 创建bufferSize ,也就是RingBuffer大小,必须是2的N次方
int ringBufferSize = 1024 * 1024; //
WaitStrategy YIELDING_WAIT = new YieldingWaitStrategy();
// 创建ringBuffer
ringBuffer = 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个消费者用于消费生产者产生的消息,每一条消息只消费了一次