同步工具类 Semaphore

 

1、Semaphore

Semaphore也是一个线程同步的辅助类，可以维护当前访问自身的线程个数，并提供了同步机制。使用Semaphore可以控制同时访问资源的线程个数，例如，实现一个文件允许的并发访问数。

方法

Semaphore的主要方法摘要：

       
       Semaphore myResources = new Semaphore(5, true);  // 一开始有5份共享资源。第二个参数表示是否是公平

　　void acquire():从此信号量获取一个许可，在提供一个许可前一直将线程阻塞，否则线程被中断。工作线程每获取一份资源，就在该对象上记下来

　　void release():释放一个许可，将其返回给信号量。

　　int availablePermits():返回此信号量中当前可用的许可数。

　　boolean hasQueuedThreads():查询是否有线程正在等待获取。

      释放指定数目的许可，并将它们归还给信标。
      可用许可数加上该指定数目。
      如果线程需要获取N个许可，在有N个许可可用之前，该线程阻塞。如果线程获取了N个许可，还有可用的许可，则依次将这些许可赋予等待获取许可的其他线程。

      semaphore.release(2);

     从信标获取指定数目的许可。如果可用许可数目不够，则线程阻塞，直到被中断。
     该方法效果与循环相同，
      for (int i = 0; i < permits; i++) acquire();
     只不过该方法是原子操作。
     如果可用许可数不够，则当前线程阻塞，直到：（二选一）
     1. 如果其他线程释放了许可，并且可用的许可数满足当前线程的请求数字；
     2. 其他线程中断了当前线程。
     permits – 要获取的许可数

    semaphore.acquire(3);

 

如下图所示，假设有n个线程来获取Semaphore里面的10份资源（n > 10），n个线程中只有10个线程能获取到，其他线程都会阻塞。直到有线程释放了资源，其他线程才能获取到。

 

例子：

public class SemaphoreTest {
 8     public static void main(String[] args) {
 9         ExecutorService service = Executors.newCachedThreadPool();
10         final  Semaphore sp = new Semaphore(3);//创建Semaphore信号量，初始化许可大小为3
11         for(int i=0;i<10;i++){
12             try {
13                 Thread.sleep(100);
14             } catch (InterruptedException e2) {
15                 e2.printStackTrace();
16             }
17             Runnable runnable = new Runnable(){
18                     public void run(){
19                     try {
20                         sp.acquire();//请求获得许可，如果有可获得的许可则继续往下执行，许可数减1。否则进入阻塞状态
21                     } catch (InterruptedException e1) {
22                         e1.printStackTrace();
23                     }
24                     System.out.println("线程" + Thread.currentThread().getName() + 
25                             "进入，当前已有" + (3-sp.availablePermits()) + "个并发");
26                     try {
27                         Thread.sleep((long)(Math.random()*10000));
28                     } catch (InterruptedException e) {
29                         e.printStackTrace();
30                     }
31                     System.out.println("线程" + Thread.currentThread().getName() + 
32                             "即将离开");                    
33                     sp.release();//释放许可，许可数加1
34                     //下面代码有时候执行不准确，因为其没有和上面的代码合成原子单元
35                     System.out.println("线程" + Thread.currentThread().getName() + 
36                             "已离开，当前已有" + (3-sp.availablePermits()) + "个并发");                    
37                 }
38             };
39             service.execute(runnable);            
40         }
41     }
42 
43 }

 

单个信号量的Semaphore对象可以实现互斥锁的功能，并且可以是由一个线程获得了“锁”，再由另一个线程释放“锁”，这可应用于死锁恢复的一些场合。

public class LockTest {
 9     public static void main(String[] args) {
10         final Business business = new Business();
11         ExecutorService executor =  Executors.newFixedThreadPool(3);
12         for(int i=0;i<3;i++)
13         {
14             executor.execute(
15                     new Runnable()
16                     {
17                         public void run()
18                         {
19                             business.service();
20                         }
21                     }
22             
23             );
24         }
25         executor.shutdown();
26     }
27     
28     private static class Business
29     {
30         private int count;
31         Lock lock = new ReentrantLock();
32         Semaphore sp = new Semaphore(1);
33         public void service() 
34         {
35             //lock.lock();
36             try {
37                 sp.acquire(); //当前线程使用count变量的时候将其锁住，不允许其他线程访问
38             } catch (InterruptedException e1) {
39                 e1.printStackTrace();
40             }
41             try {
42                 count++;
43                 try {
44                     Thread.sleep(1000);
45                 } catch (InterruptedException e) {
46                     e.printStackTrace();
47                 }
48                 System.out.println(count);
49             } catch (RuntimeException e) {
50                 e.printStackTrace();
51             }
52             finally
53             {
54                 //lock.unlock();
55                 sp.release();  //释放锁
56             }
57         }
58     }        
59     
60 }

 

当初始的资源个数为1的时候，Semaphore退化为排他锁。正因为如此，Semaphone的实现原理和锁十分类似，是基于AQS，有公平和非公平之分。Semaphore相关类的继承体系如下图所示：

Semaphore有内部类Sync

acquire、和release是阻塞的，因为抛出了InterruptedException 异常，因为sleep、await、join会阻塞并抛出中断异常

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

public abstract class AbstractQueuedSynchronizer
    extends AbstractOwnableSynchronizer
    implements java.io.Serializable {

  public final void acquireSharedInterruptibly(int arg)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        if (tryAcquireShared(arg) < 0)
            doAcquireSharedInterruptibly(arg);
    }

  public final boolean releaseShared(int arg) {
        if (tryReleaseShared(arg)) {
            doReleaseShared();
            return true;
        }
        return false;
    }

}

 

Semaphore和锁的实现原理基本相同。资源总数即state的初始值，在acquire里对state变量进行CAS减操作，减到0之后，线程阻塞；在release里对state变量进行CAS加操作。

 

以下是非阻塞的：

public class Semaphore {
protected final boolean tryReleaseShared(int releases) {
            for (;;) {
                int current = getState();
                int next = current + releases;
                if (next < current) // overflow
                    throw new Error("Maximum permit count exceeded");
                if (compareAndSetState(current, next))
                    return true;
            }
        }
static final class FairSync extends Sync {
  // ...
   FairSync(int permits) {
   super(permits);
 }
protected int tryAcquireShared(int acquires) {
            for (;;) {
                if (hasQueuedPredecessors())
                    return -1;
                int available = getState();
                int remaining = available - acquires;
                if (remaining < 0 ||
                    compareAndSetState(available, remaining))
                    return remaining;
            }
        }
    }
}