JAVA多线程之Semaphore

来源：http://blog.youkuaiyun.com/huang_xw/article/details/7090185

package com.fei;
import java.util.ArrayList;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * Java 5.0里新加了4个协调线程间进程的同步装置，它们分别是：
 * Semaphore, CountDownLatch, CyclicBarrier和Exchanger.
 * 本例主要介绍Semaphore。
 * Semaphore是用来管理一个资源池的工具，可以看成是个通行证，
 * 线程要想从资源池拿到资源必须先拿到通行证，
 * 如果线程暂时拿不到通行证，线程就会被阻断进入等待状态。
 */
public class SemaphoreTest {
	/**
	 * 模拟资源池的类
	 * 只为池发放2个通行证，即同时只允许2个线程获得池中的资源。
	 */
	public static class Pool {
		// 保存资源池中的资源
		ArrayList<String> pool = null;
		// 通行证
		Semaphore pass = null;
		Lock lock = new ReentrantLock();
		public Pool(int size) {
			// 初始化资源池
			pool = new ArrayList<String>();
			for (int i = 0; i < size; i++) {
				pool.add("Resource " + i);
			}
			// 发放2个通行证
			pass = new Semaphore(2);
		}

		public String get() throws InterruptedException {
			// 获取通行证,只有得到通行证后才能得到资源
			System.out.println("Try to get a pass...");
			pass.acquire();
			System.out.println("Got a pass");
			return getResource();
		}

		public void put(String resource) {
			// 归还通行证，并归还资源
			System.out.println("Released a pass");
			pass.release();
			releaseResource(resource);
		}

		private String getResource() {
			lock.lock();
			String result = pool.remove(0);
			System.out.println("资源 " + result + " 被取走");
			lock.unlock();
			return result;
		}

		private void releaseResource(String resource) {
			lock.lock();
			System.out.println("资源 " + resource + " 被归还");
			pool.add(resource);
			lock.unlock();
		} 
	}
	
	public static void testPool() {
		// 准备10个资源的资源池
		final Pool aPool = new Pool(10);
		Runnable worker = new Runnable() {
			public void run() {
				String resource = null;
				try {
					//取得resource
					resource = aPool.get();
					//用resource做工作
					System.out.println("I am working on " + resource);
					Thread.sleep(500);
					System.out.println("I finished on " + resource);
				} catch (InterruptedException ex) {
				}
				//归还resource
				aPool.put(resource);
			}
		};
		// 启动5个任务
		ExecutorService service = Executors.newCachedThreadPool();
		for (int i = 0; i < 5; i++) {
			service.submit(worker);
		}
		service.shutdown();
	} 
	
	public static void main(String[] args) {
		SemaphoreTest.testPool();
	}
}

运行结果

Try to get a pass...
Got a pass
Try to get a pass...
Got a pass
Try to get a pass...
资源 Resource 0 被取走
I am working on Resource 0
资源 Resource 1 被取走
I am working on Resource 1
Try to get a pass...
Try to get a pass...
I finished on Resource 0
Released a pass
I finished on Resource 1
Released a pass
资源 Resource 1 被归还
资源 Resource 0 被归还
Got a pass
Got a pass
资源 Resource 2 被取走
I am working on Resource 2
资源 Resource 3 被取走
I am working on Resource 3
I finished on Resource 2
Released a pass
资源 Resource 2 被归还
Got a pass
资源 Resource 4 被取走
I am working on Resource 4
I finished on Resource 3
Released a pass
资源 Resource 3 被归还
I finished on Resource 4
Released a pass
资源 Resource 4 被归还