通過佇列實現批量處理

壹頁書發表於2016-06-14
佇列
經常有這樣一種需求
批量非同步寫資料庫.
比如滿50個物件,批量寫入資料庫.
或者每5秒,寫一次.防止業務低峰由於數量不夠,導致不能及時入庫的問題.

原來做的比較呆板.
生產者每5秒,會向佇列中put一種特殊型別的物件.(假如這種物件叫做 pBlock,普通物件叫做oBlock)
消費者從佇列中獲取資訊,滿50個oBlock寫入.如果收到的是pBlock,則無論現在收到多少oBlock,都寫入資料庫.
這種方式呆板,是因為生產者和消費者的程式有耦合.

今天發現有一個新方式,感覺挺新穎.
  1. import java.util.ArrayList;  
  2. import java.util.List;  
  3. import java.util.concurrent.BlockingQueue;  
  4. import java.util.concurrent.LinkedBlockingQueue;  
  5. import java.util.concurrent.TimeUnit;  
  6.   
  7. import com.google.common.collect.Queues;  
  8.   
  9. public class T {  
  10.     public static void main(String[] args) {  
  11.         final BlockingQueue<Long> q = new LinkedBlockingQueue<Long>();  
  12.           
  13.         new Thread(new Runnable(){  
  14.   
  15.             public void run() {  
  16.                 long num=1L;  
  17.                 while(true)  
  18.                 {  
  19.                     try {  
  20.                         System.out.println(num);  
  21.                         q.put(num);  
  22.                         num++;  
  23.                         Thread.sleep(1000);  
  24.                     } catch (InterruptedException e) {  
  25.                         e.printStackTrace();  
  26.                     }  
  27.                 }  
  28.             }}).start();  
  29.           
  30.         while (true) {  
  31.             try {  
  32.                 List<Long> l = new ArrayList<Long>(10);  
  33.                 Queues.drain(q, l, 10, 5, TimeUnit.SECONDS);  
  34.                 System.out.println(l);  
  35.             } catch (InterruptedException e) {  
  36.                 e.printStackTrace();  
  37.             }  
  38.         }  
  39.     }  
  40. }  


來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/29254281/viewspace-2120130/，如需轉載，請註明出處，否則將追究法律責任。

相關文章