架构师养成记--16.disruptor并发框架中RingBuffer的使用

很多时候我们只需要消息中间件这样的功能，那么直需要RinBuffer就可以了。

入口：

 import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.Future;

 import com.lmax.disruptor.BatchEventProcessor;
 import com.lmax.disruptor.EventFactory;
 import com.lmax.disruptor.RingBuffer;
 import com.lmax.disruptor.SequenceBarrier;
 import com.lmax.disruptor.YieldingWaitStrategy;

 public class Main1 {  

     public static void main(String[] args) throws Exception {
         int BUFFER_SIZE=1024;
         int THREAD_NUMBERS=4;
         /*
          * createSingleProducer创建一个单生产者的RingBuffer，
          * 第一个参数叫EventFactory，从名字上理解就是"事件工厂"，其实它的职责就是产生数据填充RingBuffer的区块。
          * 第二个参数是RingBuffer的大小，它必须是2的指数倍 目的是为了将求模运算转为&运算提高效率
          * 第三个参数是RingBuffer的生产都在没有可用区块的时候(可能是消费者（或者说是事件处理器） 太慢了)的等待策略
          */
         final RingBuffer<Trade> ringBuffer = RingBuffer.createSingleProducer(new EventFactory<Trade>() {
             @Override
             public Trade newInstance() {
                 return new Trade();
             }
         }, BUFFER_SIZE, new YieldingWaitStrategy());  

         //创建线程池
         ExecutorService executors = Executors.newFixedThreadPool(THREAD_NUMBERS);  

         //创建SequenceBarrier
         SequenceBarrier sequenceBarrier = ringBuffer.newBarrier();  

         /****************** @beg 消费者消费数据 2017-1-11******************/
         //创建消息处理器
         BatchEventProcessor<Trade> transProcessor = new BatchEventProcessor<Trade>(
                 ringBuffer, sequenceBarrier, new TradeHandler());  

         //这一步的目的就是把消费者的位置信息引用注入到生产者    如果只有一个消费者的情况可以省略
         ringBuffer.addGatingSequences(transProcessor.getSequence());  

         //把消息处理器提交到线程池
         executors.submit(transProcessor);
         /****************** @end 消费者消费数据 2017-1-11******************/

         //如果存在多个消费者 那重复执行上面3行代码 把TradeHandler换成其它消费者类  

         /****************** @beg 生产者生产数据  2017-1-11******************/

         Future<?> future= executors.submit(new Callable<Void>() {
             @Override
             public Void call() throws Exception {
                 long seq;
                 for(int i=0;i<10;i++){
                     seq = ringBuffer.next();//占个坑 --ringBuffer一个可用区块
                     ringBuffer.get(seq).setPrice(Math.random()*9999);//给这个区块放入 数据
                     ringBuffer.publish(seq);//发布这个区块的数据使handler(consumer)可见
                 }
                 return null;
             }
         }); 

         /****************** @end 生产者生产数据 2017-1-11******************/

         future.get();//等待生产者结束
         Thread.sleep(1000);//等上1秒，等消费都处理完成
         transProcessor.halt();//通知事件(或者说消息)处理器 可以结束了（并不是马上结束!!!）
         executors.shutdown();//终止线程
     }
 }  

消费者：

 import java.util.UUID;

 import com.lmax.disruptor.EventHandler;
 import com.lmax.disruptor.WorkHandler;

 public class TradeHandler implements EventHandler<Trade>, WorkHandler<Trade> {  

     @Override
     public void onEvent(Trade event, long sequence, boolean endOfBatch) throws Exception {
         this.onEvent(event);
     }  

     @Override
     public void onEvent(Trade event) throws Exception {
         //这里做具体的消费逻辑
         event.setId(UUID.randomUUID().toString());//简单生成下ID
         System.out.println(event.getId());
     }
 }  

数据对象：

 import java.util.concurrent.atomic.AtomicInteger;

 public class Trade {  

     private String id;//ID
     private String name;
     private double price;//金额
     private AtomicInteger count = new AtomicInteger(0);

     public String getId() {
         return id;
     }
     public void setId(String id) {
         this.id = id;
     }
     public String getName() {
         return name;
     }
     public void setName(String name) {
         this.name = name;
     }
     public double getPrice() {
         return price;
     }
     public void setPrice(double price) {
         this.price = price;
     }
     public AtomicInteger getCount() {
         return count;
     }
     public void setCount(AtomicInteger count) {
         this.count = count;
     } 

 }