Java 并发专题 ： Semaphore 实现 互斥 与 连接池

继续并发方面的知识。今天介绍Semaphore，同样在java.util.concurrent包下。

本来准备通过例子，从自己实现到最后使用并发工具实现，但是貌似效果并不是很好，有点太啰嗦的感觉，所有准备直入主题。

介绍：Semaphore中管理着一组虚拟的许可，许可的初始数量可通过构造函数来指定【new Semaphore(1);】，执行操作时可以首先获得许可【semaphore.acquire();】，并在使用后释放许可【semaphore.release();】。如果没有许可，那么acquire方法将会一直阻塞直到有许可（或者直到被终端或者操作超时）。

作用：可以用来控制同时访问某个特定资源的操作数量，或者某个操作的数量。

下面使用Semaphore实现两个例子：

1、互斥

大家都学过操作系统，都知道互斥的概念，比较简单的互斥实现，比如PV操作，判断资源，然后忙等实现互斥；上一篇博客也说过，忙等对CPU的消耗巨大，下面我们通过Semaphore来实现一个比较好的互斥操作：

假设我们公司只有一台打印机，我们需要对这台打印机的打印操作进行互斥控制：

package com.zhy.concurrency.semaphore;

import java.util.concurrent.Semaphore;

/**
* 使用信号量机制，实现互斥访问打印机
*
* @author zhy
*
*/
public class MutexPrint
{

/**
* 定义初始值为1的信号量
*/
private final Semaphore semaphore = new Semaphore(1);

/**
* 模拟打印操作
* @param str
* @throws InterruptedException
*/
public void print(String str) throws InterruptedException
{
//请求许可
semaphore.acquire();

System.out.println(Thread.currentThread().getName()+" enter ...");
Thread.sleep(1000);
System.out.println(Thread.currentThread().getName() + "正在打印 ..." + str);
System.out.println(Thread.currentThread().getName()+" out ...");
//释放许可
semaphore.release();
}

public static void main(String[] args)
{
final MutexPrint print = new MutexPrint();

/**
* 开启10个线程，抢占打印机
*/
for (int i = 0; i < 10; i++)
{
new Thread()
{
public void run()
{
try
{
print.print("helloworld");
} catch (InterruptedException e)
{
e.printStackTrace();
}
};
}.start();
}

}

}

输出结果：

Thread-1 enter ...
Thread-1正在打印 ...helloworld
Thread-1 out ...
Thread-2 enter ...
Thread-2正在打印 ...helloworld
Thread-2 out ...
Thread-0 enter ...
Thread-0正在打印 ...helloworld
Thread-0 out ...
Thread-3 enter ...
Thread-3正在打印 ...helloworld
Thread-3 out ...

通过初始值为1的Semaphore，很好的实现了资源的互斥访问。

2、连接池的模拟实现

在项目中处理高并发时，一般数据库都会使用数据库连接池，假设现在数据库连接池最大连接数为10，当10个连接都分配出去以后，现在有用户继续请求连接，可能的处理：

a、手动抛出异常，用户界面显示，服务器忙，稍后再试

b、阻塞，等待其他连接的释放

从用户体验上来说，更好的选择当然是阻塞，等待其他连接的释放，用户只会觉得稍微慢了一点，并不影响他的操作。下面使用Semaphore模拟实现一个数据库连接池：

package com.zhy.concurrency.semaphore;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Semaphore;
/**
* 使用Semaphore模拟数据库链接池的使用
* @author zhy
*
*/
public class ConnectPool
{
private final List<Conn> pool = new ArrayList<Conn>(3);
private final Semaphore semaphore = new Semaphore(3);

/**
* 初始化分配3个连接
*/
public ConnectPool()
{
pool.add(new Conn());
pool.add(new Conn());
pool.add(new Conn());
}

/**
* 请求分配连接
* @return
* @throws InterruptedException
*/
public Conn getConn() throws InterruptedException
{
semaphore.acquire();
Conn c = null  ;
synchronized (pool)
{
c = pool.remove(0);
}
System.out.println(Thread.currentThread().getName()+" get a conn " + c);
return c ;
}

/**
* 释放连接
* @param c
*/
public void release(Conn c)
{
pool.add(c);
System.out.println(Thread.currentThread().getName()+" release a conn " + c);
semaphore.release();
}

public static void main(String[] args)
{

final ConnectPool pool = new ConnectPool();

/**
* 第一个线程占用1个连接3秒
*/
new Thread()
{
public void run()
{
try
{
Conn c = pool.getConn();
Thread.sleep(3000);
pool.release(c);
} catch (InterruptedException e)
{
e.printStackTrace();
}
};
}.start();
/**
* 开启3个线程请求分配连接
*/
for (int i = 0; i < 3; i++)
{
new Thread()
{
public void run()
{
try
{
Conn c = pool.getConn();
} catch (InterruptedException e)
{
e.printStackTrace();
}
};
}.start();
}

}

private class Conn
{
}

}

Thread-0 get a conn com.zhy.concurrency.semaphore.ConnectPool$Conn@12b6651
Thread-2 get a conn com.zhy.concurrency.semaphore.ConnectPool$Conn@e53108
Thread-1 get a conn com.zhy.concurrency.semaphore.ConnectPool$Conn@1888759
Thread-0 release a conn com.zhy.concurrency.semaphore.ConnectPool$Conn@12b6651
Thread-3 get a conn com.zhy.concurrency.semaphore.ConnectPool$Conn@12b6651

我们测试时，让Thread-0持有一个连接3秒，然后瞬间让3个线程再去请求分配连接，造成Thread-3一直等到Thread-0对连接的释放，然后获得连接。

通过两个例子，基本已经了解了Semaphore的用法，这里的线程池例子只是为了说明Semaphore的用法，真实的实现代码比这复杂的多，而且可能也不会直接用Semaphore。

好了，之后会继续Java并发的博客。