Java中的Semaphore信號量詳解

更新時間：2023年12月11日 09:12:13 作者：warybee

這篇文章主要介紹了Java中的Semaphore信號量詳解,Semaphore（信號量）是用來控制同時訪問特定資源的線程數(shù)量,通過協(xié)調(diào)各個線程以保證合理地使用公共資源,需要的朋友可以參考下

1、介紹

Semaphore（信號量）是用來控制同時訪問特定資源的線程數(shù)量，通過協(xié)調(diào)各個線程以保證合理地使用公共資源。

Semaphore通過使用計數(shù)器來控制對共享資源的訪問。如果計數(shù)器大于0，則允許訪問。

如果為0，則拒絕訪問。計數(shù)器所計數(shù)的是允許訪問共享資源的許可。因此，要訪問資源，必須從信號量中授予線程許可。

2、主要方法

void acquire() ：從信號量獲取一個許可，如果無可用許可前將一直阻塞等待，
void acquire(int permits) ：獲取指定數(shù)目的許可，如果無可用許可前也將會一直阻塞等待
boolean tryAcquire()：從信號量嘗試獲取一個許可，如果無可用許可，直接返回false，不會阻塞
boolean tryAcquire(int permits)：嘗試獲取指定數(shù)目的許可，如果無可用許可直接返回false
boolean tryAcquire(int permits, long timeout, TimeUnit unit)：在指定的時間內(nèi)嘗試從信號量中獲取許可，如果在指定的時間內(nèi)獲取成功，返回true，否則返回false
void release()：釋放一個許可，別忘了在finally中使用，注意：多次調(diào)用該方法，會使信號量的許可數(shù)增加，達到動態(tài)擴展的效果，如：初始permits為1，調(diào)用了兩次release，最大許可會改變?yōu)?
int availablePermits()：獲取當前信號量可用的許可

Semaphore構(gòu)造函數(shù)

 public Semaphore(int permits) {
        sync = new NonfairSync(permits);
    }

public Semaphore(int permits, boolean fair) {
        sync = fair ? new FairSync(permits) : new NonfairSync(permits);
    }

permits 初始許可數(shù)，也就是最大訪問線程數(shù)
fair 當設(shè)置為false時，創(chuàng)建的信號量為非公平鎖；當設(shè)置為true時，信號量是公平鎖

3、Semaphore登錄限流示例

在以下示例中，實現(xiàn)一個簡單的登錄隊列，通過Semaphore來限制系統(tǒng)中的用戶數(shù)：

public static void main(String[] args) {
        //允許最大的登錄數(shù)
        int slots=10;
        ExecutorService executorService = Executors.newFixedThreadPool(slots);
        LoginQueueUsingSemaphore loginQueue = new LoginQueueUsingSemaphore(slots);
        //線程池模擬登錄
        for (int i = 1; i <= slots; i++) {
            final int num=i;
            executorService.execute(()->{
                 if (loginQueue.tryLogin()){
                     System.out.println("用戶:"+num+"登錄成功！");
                 }else {
                     System.out.println("用戶:"+num+"登錄失?。?);
                 }
            });
        }
        executorService.shutdown();
        System.out.println("當前可用許可證數(shù)："+loginQueue.availableSlots());
        //此時已經(jīng)登錄了10個用戶，再次登錄的時候會返回false
        if (loginQueue.tryLogin()){
            System.out.println("登錄成功！");
        }else {
            System.out.println("系統(tǒng)登錄用戶已滿，登錄失??！");
        }
        //有用戶退出登錄
        loginQueue.logout();
        //再次登錄
        if (loginQueue.tryLogin()){
            System.out.println("登錄成功！");
        }else {
            System.out.println("系統(tǒng)登錄用戶已滿，登錄失?。?);
        }
    }

class LoginQueueUsingSemaphore{

    private Semaphore semaphore;

    /**
     *
     * @param slotLimit
     */
    public LoginQueueUsingSemaphore(int slotLimit){
        semaphore=new Semaphore(slotLimit);
    }

    boolean tryLogin() {
        //獲取一個憑證
        return semaphore.tryAcquire();
    }

    void logout() {
        semaphore.release();
    }

    int availableSlots() {
        return semaphore.availablePermits();
    }
}

運行結(jié)果：

在這里插入圖片描述