併發框架Disruptor幾個Demo

掃盲:

要想了解Disruptor框架必需多花點時間研究下它的工做原理,否則代碼是無法擼的!!! 數據結構

關於Disruptor的詳細資料及原理請細看!!!    http://ifeve.com/disruptor/ 框架

 

Disruptor版本: 3.2.1

 

名詞解釋

消費者==事件處理器 dom

 

 

1、如今你必須明白如下問題:

一、你必須明白Ringbuffer是什麼,它的數據結構是怎麼樣的,有什麼約定,爲何高效,它的職責是什麼。 ide

二、ConsumerBarrier (ifeve網上的譯文版本比較早,這個類在2.0.0以後就一直被更名,3.2.1的版本中它是SequenceBarrier)它的職責是什麼。 函數

 

Disruptor框架在2.0版本以後再也不採用生產者、消費者模型來編寫API,而是使用事件模型,其實只是接口設計、類名和概念上的變化,內部原理其實仍是同樣的。 this

 

DEMO 1、使用原生API建立一個簡單的生產者和消費者

 

Java代碼   收藏代碼
  1. //DEMO中使用的 消息全假定是一條交易  
  2. public class TradeTransaction {  
  3.     private String id;//交易ID  
  4.     private double price;//交易金額  
  5.       
  6.     public TradeTransaction() {  
  7.     }  
  8.     public TradeTransaction(String id, double price) {  
  9.         super();  
  10.         this.id = id;  
  11.         this.price = price;  
  12.     }  
  13.     public String getId() {  
  14.         return id;  
  15.     }  
  16.     public void setId(String id) {  
  17.         this.id = id;  
  18.     }  
  19.     public double getPrice() {  
  20.         return price;  
  21.     }  
  22.     public void setPrice(double price) {  
  23.         this.price = price;  
  24.     }  
  25. }  
  26.   
  27. public class TradeTransactionInDBHandler implements EventHandler<TradeTransaction>,WorkHandler<TradeTransaction> {  
  28.   
  29.     @Override  
  30.     public void onEvent(TradeTransaction event, long sequence,  
  31.             boolean endOfBatch) throws Exception {  
  32.         this.onEvent(event);  
  33.     }  
  34.   
  35.     @Override  
  36.     public void onEvent(TradeTransaction event) throws Exception {  
  37.         //這裏作具體的消費邏輯  
  38.         event.setId(UUID.randomUUID().toString());//簡單生成下ID  
  39.         System.out.println(event.getId());  
  40.     }  
  41. }  
  42.   
  43.   
  44.   
  45. public class Demo1 {  
  46.     public static void main(String[] args) throws InterruptedException, ExecutionException {  
  47.         int BUFFER_SIZE=1024;  
  48.         int THREAD_NUMBERS=4;  
  49.         /* 
  50.          * createSingleProducer建立一個單生產者的RingBuffer, 
  51.          * 第一個參數叫EventFactory,從名字上理解就是「事件工廠」,其實它的職責就是產生數據填充RingBuffer的區塊。 
  52.          * 第二個參數是RingBuffer的大小,它必須是2的指數倍 目的是爲了將求模運算轉爲&運算提升效率 
  53.          * 第三個參數是RingBuffer的生產都在沒有可用區塊的時候(多是消費者(或者說是事件處理器) 太慢了)的等待策略 
  54.          */  
  55.         final RingBuffer<TradeTransaction> ringBuffer = RingBuffer.createSingleProducer(new EventFactory<TradeTransaction>() {  
  56.             @Override  
  57.             public TradeTransaction newInstance() {  
  58.                 return new TradeTransaction();  
  59.             }  
  60.         }, BUFFER_SIZE,new YieldingWaitStrategy());  
  61.         //建立線程池  
  62.         ExecutorService executors = Executors.newFixedThreadPool(THREAD_NUMBERS);  
  63.         //建立SequenceBarrier  
  64.         SequenceBarrier sequenceBarrier = ringBuffer.newBarrier();  
  65.           
  66.         //建立消息處理器  
  67.         BatchEventProcessor<TradeTransaction> transProcessor = new BatchEventProcessor<TradeTransaction>(  
  68.                 ringBuffer, sequenceBarrier, new TradeTransactionInDBHandler());  
  69.           
  70.         //這一部的目的是讓RingBuffer根據消費者的狀態    若是隻有一個消費者的狀況能夠省略  
  71.         ringBuffer.addGatingSequences(transProcessor.getSequence());  
  72.           
  73.         //把消息處理器提交到線程池  
  74.         executors.submit(transProcessor);  
  75.         //若是存大多個消費者 那重複執行上面3行代碼 把TradeTransactionInDBHandler換成其它消費者類  
  76.           
  77.         Future<?> future=executors.submit(new Callable<Void>() {  
  78.             @Override  
  79.             public Void call() throws Exception {  
  80.                 long seq;  
  81.                 for(int i=0;i<1000;i++){  
  82.                     seq=ringBuffer.next();//佔個坑 --ringBuffer一個可用區塊  
  83.                       
  84.                     ringBuffer.get(seq).setPrice(Math.random()*9999);//給這個區塊放入 數據  若是此處不理解,想一想RingBuffer的結構圖  
  85.                       
  86.                     ringBuffer.publish(seq);//發佈這個區塊的數據使handler(consumer)可見  
  87.                 }  
  88.                 return null;  
  89.             }  
  90.         });  
  91.         future.get();//等待生產者結束  
  92.         Thread.sleep(1000);//等上1秒,等消費都處理完成  
  93.         transProcessor.halt();//通知事件(或者說消息)處理器 能夠結束了(並非立刻結束!!!)  
  94.         executors.shutdown();//終止線程  
  95.     }  
  96. }  

 

 DEMO2、使用WorkerPool輔助建立消費者

Java代碼   收藏代碼
  1. public class Demo2 {  
  2.     public static void main(String[] args) throws InterruptedException {  
  3.         int BUFFER_SIZE=1024;  
  4.         int THREAD_NUMBERS=4;  
  5.         EventFactory<TradeTransaction> eventFactory=new EventFactory<TradeTransaction>() {  
  6.             public TradeTransaction newInstance() {  
  7.                 return new TradeTransaction();  
  8.             }  
  9.         };  
  10.         RingBuffer<TradeTransaction> ringBuffer=RingBuffer.createSingleProducer(eventFactory, BUFFER_SIZE);  
  11.           
  12.         SequenceBarrier sequenceBarrier = ringBuffer.newBarrier();  
  13.           
  14.         ExecutorService executor = Executors.newFixedThreadPool(THREAD_NUMBERS);  
  15.           
  16.         WorkHandler<TradeTransaction> workHandlers=new TradeTransactionInDBHandler();  
  17.         /* 
  18.          * 這個類代碼很簡單的,親本身看哈!~ 
  19.          */  
  20.         WorkerPool<TradeTransaction> workerPool=new WorkerPool<TradeTransaction>(ringBuffer, sequenceBarrier, new IgnoreExceptionHandler(), workHandlers);  
  21.           
  22.         workerPool.start(executor);  
  23.           
  24.         //下面這個生產8個數據,圖簡單就寫到主線程算了  
  25.         for(int i=0;i<8;i++){  
  26.             long seq=ringBuffer.next();  
  27.             ringBuffer.get(seq).setPrice(Math.random()*9999);  
  28.             ringBuffer.publish(seq);  
  29.         }  
  30.           
  31.         Thread.sleep(1000);  
  32.         workerPool.halt();  
  33.         executor.shutdown();  
  34.     }  
  35. }  

 DEMO3、demo3寫個流弊點的像下圖這樣。此次用Disruptor來完成整個構建工做.

從中圖能夠看出需求是介樣子的:生產者生產數據通過C1,C2處理完成後再到C3。 spa

假設以下場景: 線程

一、交易網關收到交易(P1)把交易數據發到RingBuffer中, 設計

二、負責處理增值業務的消費者C1和負責數據存儲的消費者C2負責處理交易 接口

三、負責發送JMS消息的消費者C3在C1和C2處理完成後再進行處理。

 

讓代碼說話:

Java代碼   收藏代碼
  1. public class TradeTransactionJMSNotifyHandler implements EventHandler<TradeTransaction> {  
  2.   
  3.     @Override  
  4.     public void onEvent(TradeTransaction event, long sequence,  
  5.             boolean endOfBatch) throws Exception {  
  6.         //do send jms message  
  7.     }  
  8. }  
  9.   
  10.   
  11.   
  12. public class TradeTransactionPublisher implements Runnable{  
  13.     Disruptor<TradeTransaction> disruptor;  
  14.     private CountDownLatch latch;  
  15.     private static int LOOP=10000000;//模擬一千萬次交易的發生  
  16.   
  17.     public TradeTransactionPublisher(CountDownLatch latch,Disruptor<TradeTransaction> disruptor) {  
  18.         this.disruptor=disruptor;  
  19.         this.latch=latch;  
  20.     }  
  21.   
  22.     @Override  
  23.     public void run() {  
  24.         TradeTransactionEventTranslator tradeTransloator=new TradeTransactionEventTranslator();  
  25.         for(int i=0;i<LOOP;i++){  
  26.             disruptor.publishEvent(tradeTransloator);  
  27.         }  
  28.         latch.countDown();  
  29.     }  
  30.       
  31. }  
  32.   
  33. class TradeTransactionEventTranslator implements EventTranslator<TradeTransaction>{  
  34.     private Random random=new Random();  
  35.     @Override  
  36.     public void translateTo(TradeTransaction event, long sequence) {  
  37.         this.generateTradeTransaction(event);  
  38.     }  
  39.     private TradeTransaction generateTradeTransaction(TradeTransaction trade){  
  40.         trade.setPrice(random.nextDouble()*9999);  
  41.         return trade;  
  42.     }  
  43. }  
  44.   
  45.   
  46. public class TradeTransactionVasConsumer implements EventHandler<TradeTransaction> {  
  47.   
  48.     @Override  
  49.     public void onEvent(TradeTransaction event, long sequence,  
  50.             boolean endOfBatch) throws Exception {  
  51.         //do something....  
  52.     }  
  53.       
  54. }  
  55.   
  56.   
  57.   
  58. public class Demo3 {  
  59.     public static void main(String[] args) throws InterruptedException {  
  60.         long beginTime=System.currentTimeMillis();  
  61.           
  62.         int bufferSize=1024;  
  63.         ExecutorService executor=Executors.newFixedThreadPool(4);  
  64.         //這個構造函數參數,相信你在瞭解上面2個demo以後就看下就明白了,不解釋了~  
  65.         Disruptor<TradeTransaction> disruptor=new Disruptor<TradeTransaction>(new EventFactory<TradeTransaction>() {  
  66.             @Override  
  67.             public TradeTransaction newInstance() {  
  68.                 return new TradeTransaction();  
  69.             }  
  70.         }, bufferSize, executor, ProducerType.SINGLE, new BusySpinWaitStrategy());  
  71.           
  72.         //使用disruptor建立消費者組C1,C2  
  73.         EventHandlerGroup<TradeTransaction> handlerGroup=disruptor.handleEventsWith(new TradeTransactionVasConsumer(),new TradeTransactionInDBHandler());  
  74.           
  75.         TradeTransactionJMSNotifyHandler jmsConsumer=new TradeTransactionJMSNotifyHandler();  
  76.         //聲明在C1,C2完事以後執行JMS消息發送操做 也就是流程走到C3  
  77.         handlerGroup.then(jmsConsumer);  
  78.           
  79.           
  80.         disruptor.start();//啓動  
  81.         CountDownLatch latch=new CountDownLatch(1);  
  82.         //生產者準備  
  83.         executor.submit(new TradeTransactionPublisher(latch, disruptor));  
  84.         latch.await();//等待生產者完事.  
  85.         disruptor.shutdown();  
  86.         executor.shutdown();  
  87.           
  88.         System.out.println("總耗時:"+(System.currentTimeMillis()-beginTime));  
  89.     }  
  90. }  
相關文章
相關標籤/搜索