控制併發線程數的Semaphore

Semaphore被用於控制特定資源在同一個時間被訪問的線程數量，它經過協調各個線程，以保證資源能夠被合理的使用。

作個比喻，把Semaphore比做是控制流量的紅綠燈，好比xx馬路要如今流量，只容許同時有一百輛車在馬路上行駛，其餘的都必須在路口等待，因此前一百輛會看到綠燈，能夠開進馬路，後面的車會看到紅燈，不能開進馬路，可是若是前面一百輛車中有5輛已經離開了馬路，那後面就容許有5輛車駛入馬路，這裏例子裏說的車就是線程，駛入馬路就表明線程正在執行，離開馬路就表示線程執行完成，看到紅燈就表明線程被阻塞，不能執行。

應用場景

Semaph能夠用來作流量限制，特別是公共資源有限的應用場景，好比說數據庫鏈接。

假若有一個需求，要讀取幾萬個文件的數據，由於都是IO密集型人物，咱們能夠啓動幾十個線程併發的讀取，可是若是讀取到內存後，還須要儲存到數據庫，而數據庫的鏈接數只有10個，這時候咱們就必須控制只有10個線程同時獲取到數據庫鏈接，不然會拋出異常提示沒法鏈接數據庫。針對這種狀況，咱們就可使用Semaphore來作流量控制。代碼以下：

package com.dreyer.javadoc.thread;



import java.util.concurrent.*;



/**

 * @description

 * @author: Dreyer

 * @date: 16/5/15 上午11:59

 */

public class SemaphoreDemo {

    /**

     * 線程數量

     */

    private static final int THREAD_COUNT = 30;



    /**

     * 線程池

     */

    private static ExecutorService executor = Executors.newFixedThreadPool(THREAD_COUNT);



    private static Semaphore semaphore = new Semaphore(10);





    public static void main(String[] args) {

        for (int i = 0; i < THREAD_COUNT; i++) {

            executor.execute(new Runnable() {

                public void run() {

                    try {

                        // 獲取一個"許可證"

                        semaphore.acquire();



                        // 模擬數據保存

                        TimeUnit.SECONDS.sleep(2);

                        System.out.println("save date...");



                        // 執行完後,歸還"許可證"

                        semaphore.release();

                    } catch (InterruptedException e) {

                        e.printStackTrace();

                    }

                }

            });

        }

        executor.shutdown();

    }

}

在代碼中，雖然有30個線程在執行，可是隻運行10個併發的執行。因此咱們能夠看到在執行的過程當中

save data...是每10輸出的。

Semaphore的構造方法Semaphore(int permits)接受一個整形的數字，表示可用的許可證數量。

Semaphore(10)表示運行10個線程獲取許可證，也就是最大的併發數是10。

Semaphore的用法也很簡單，首先使用Semaphore.acquire()方法獲取一個許可證，使用完以後調用

release()方法歸還許可證