java同步工具類之--

package com.test;
/**
 * 今天聽同事講了一個需求,他需要啓動多個線程把接口數據分批導入目標,要求
 * 每次執行的時候必須保證前一次任務已結束,處理這個需求的方式有很多種,其實質即
 * 線程間同步問題,正好這兩天我也在關注線程同步相關的東東,jdk提供了不少的線程
 * 同步工具類,CountDownLatch:一個同步輔助類,在完成一組正在其他線程中執行的
 * 操作之前,它允許一個或多個線程一直等待。 
 * 用給定的計數 初始化 CountDownLatch。由於調用了 countDown() 方法,所以在當前計數到達零之前,
 * await 方法會一直受阻塞。之後,會釋放所有等待的線程,await 的所有後續調用都將立即返回。
 * 這種現象只出現一次——計數無法被重置(這點很重要哦)。如果需要重置計數,請考慮使用 CyclicBarrier。
 * 下面是一個簡單的例子來模擬該需求,當然可能因爲爲了模擬場景,會有一些不合理的地方,這裏主要闡述
 * CountDownLatch同步,關於CountDownLatch的源碼將在後面來分析,其主要涉及AbstractQueuedSynchronizer
 * 這個類,他的類容相對比較複雜
 * **/

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.concurrent.CountDownLatch;

public class Driver {
	static List<Integer> strList = null;
	int k = 0;
	static {
		//模擬數據
		strList = new ArrayList<Integer>();
		for (int i = 0; i < 50; i++) {
			strList.add(i);
		}
	}

	public static void main(String args[]) {
		boolean isEnd = true;
		//爲了驗證正確性,只執行20次
		int count=0;
		Driver d = new Driver();
		while (isEnd && strList.size() > 0&&count<20) {
			CountDownLatch startSignal = new CountDownLatch(1);
			final CountDownLatch doneSignal = new CountDownLatch(5);
			for (int i = 0; i < 5; ++i)
			{
				new Thread(d.new Worker(startSignal, doneSignal,i)).start();
			}
			//計數減1 子線程Worker可以執行
			startSignal.countDown(); 
			try {
				new Thread(new Runnable() {
					Random r = new Random();
					@Override
					public void run() {
						try {
							//主線程阻塞 知道所有子線程將doneSignal清零
							doneSignal.await();
						} catch (InterruptedException e) {
							e.printStackTrace();
						}
						while(strList.size()<=0){
							int pos = r.nextInt(1000);
							strList.clear();
							for (int i = pos; i < pos + 50; i++) {
								strList.add(i);
							}
						}
					}
				}).start();
				isEnd = true;
			} catch (Exception e) {
				e.printStackTrace();
			}
			count++;
		}
	}

	class Worker implements Runnable {
		private final CountDownLatch startSignal;
		private final CountDownLatch doneSignal;
		private int i;

		Worker(CountDownLatch startSignal, CountDownLatch doneSignal,int i) {
			this.startSignal = startSignal;
			this.doneSignal = doneSignal;
			this.i=i;
			
		}

		public void run() {
			try {
				// 等待主線程執行countDown
				startSignal.await();
				doWork();
				//計數減1
				doneSignal.countDown();
			} catch (InterruptedException ex) {
			} // return;
		}

		void doWork() {
			synchronized (strList) {
				int start=(i)*(50/5);
				int end=(i+1)*(50/5);
				for (int i = start; i < end; i++) {
					System.out.println(strList.get(i) + "---" + "已被刪除");
				}
			}
		}
	}
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章