如有一個需求，要讀取幾萬個文件的數(shù)據(jù)，因為都是IO密集型任務(wù)，我們可以啟動幾十個線程并發(fā)地讀取，但是如果讀到內(nèi)存后，還需要存儲到數(shù)據(jù)庫中，而數(shù)據(jù)庫的連接數(shù)只有10個，這時我們必須控制只有10個線程同時獲取數(shù)據(jù)庫連接保存數(shù)據(jù)，否則會報錯無法獲取數(shù)據(jù)庫連接。這個時候，就可以使用Semaphore來做流量控制

import java.util.concurrent.ExecutorService;import java.util.concurrent.Executors;import java.util.concurrent.Semaphore;public class SemaphoreTest {PRivate static final int THREAD_COUNT = 30;private static ExecutorService threadPool = Executors.newFixedThreadPool(THREAD_COUNT);private static Semaphore s = new Semaphore(10);public static void main(String[] args) {for (int i = 0; i < THREAD_COUNT; i++) {threadPool.execute(new Runnable() {@Overridepublic void run() {try {s.acquire(); //獲取一個許可證System.out.println("save data");s.release(); //歸還許可證} catch (InterruptedException e) {}}});}threadPool.shutdown();}}

雖然有30個線程在執(zhí)行，但是只允許10個并發(fā)執(zhí)行。Semaphore的構(gòu)造方法Semaphore（int permits）接受一個整型的數(shù)字，表示可用的許可證數(shù)量。Semaphore（10）表示允許10個線程獲取許可證，也就是最大并發(fā)數(shù)是10。Semaphore的用法也很簡單，首先線程使用Semaphore的acquire()方法獲取一個許可證，使用完之后調(diào)用release()方法歸還許可證。還可以用tryAcquire()方法嘗試獲取許可證。