1. 程式人生 > 程式設計 >Java基於Semaphore構建阻塞物件池

Java基於Semaphore構建阻塞物件池

java中使用Semaphore構建阻塞物件池

Semaphore是java 5中引入的概念,叫做計數訊號量。主要用來控制同時訪問某個特定資源的訪問數量或者執行某個操作的數量。

Semaphore中定義了一組虛擬的permits,通過獲取和釋放這些permits,Semaphore可以控制資源的個數。

Semaphore的這個特性可以用來構造資源池,比如資料庫連線池等。

Semaphore有兩個建構函式:

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

permits定義了許可資源的個數,而fair則表示是否支援FIFO的順序。

兩個比較常用的方法就是acquire和release了。

  public void acquire() throws InterruptedException {
    sync.acquireSharedInterruptibly(1);
  }
  public void release() {
    sync.releaseShared(1);
  }

其中acquire用來獲取資源,release用來釋放資源。

有了這兩個特性, 我們看一下怎麼使用Semaphore來定義一個一個有界容器。

我們可以將Semaphore初始化為容器池大小,並且在容器池獲取資源時呼叫acquire,將資源返回給容器池之後再呼叫release。

我們看下面的一個實現:

public class SemaphoreUsage<T> {

  private final Set<T> set;
  private final Semaphore sem;

  public SemaphoreUsage(int bound){
    this.set = Collections.synchronizedSet(new HashSet<T>());
    sem= new Semaphore(bound);
  }

  public boolean add (T o) throws InterruptedException{
    sem.acquire();
    boolean wasAdded = false;
    try{
      wasAdded=set.add(o);
      return wasAdded;
    }finally {
      if(!wasAdded){
        sem.release();
      }
    }
  }

  public boolean remove(Object o){
    boolean wasRemoved = set.remove(o);
    if(wasRemoved){
      sem.release();
    }
    return wasRemoved;
  }
}

上面的例子我們定義了一個有界的synchronizedSet。 要注意一點是在add方法中,只有add成功之後才會呼叫release方法。

本文的例子請參考https://github.com/ddean2009/learn-java-concurrency/tree/master/Semaphore

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。