操作系统的信号量是个很重要的概念,在进程控制方面都有应用。Java 并发库 的Semaphore 可以很轻松完成信号量控制,Semaphore可以控制某个资源可被同时访问的个数,acquire()获取一个许可,如果没有就等待,而release()释放一个许可。比如在Windows下可以设置共享文件的最大客户端访问个数。
Semaphore维护了当前访问的个数,提供同步机制,控制同时访问的个数。在数据结构中链表可以保存“无限”的节点,用Semaphore可以实现有限大小的链表。另外重入锁ReentrantLock也可以实现该功能,但实现上要负责些,代码也要复杂些。
下面是模拟一个连接池,控制同一时间最多只能有50个线程访问。
import java.util.UUID;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
public class TestSemaphore extends Thread {
public static void main(String[] args) {
int i = 0;
while (i < 500) {
i++;
new TestSemaphore().start();
try {
Thread.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
/**
* 控制某资源同时被访问的个数的类 控制同一时间最后只能有50个访问
*/
static Semaphore semaphore = new Semaphore(50);
static int timeout = 500;
public void run() {
try {
Object connec = getConnection();
System.out.println("获得一个连接" + connec);
Thread.sleep(300);
releaseConnection(connec);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public void releaseConnection(Object connec) {
/* 释放许可 */
semaphore.release();
System.out.println("释放一个连接" + connec);
}
public Object getConnection() {
try {/* 获取许可 */
boolean getAccquire = semaphore.tryAcquire(timeout, TimeUnit.MILLISECONDS);
if (getAccquire) {
return UUID.randomUUID().toString();
}
} catch (InterruptedException e) {
e.printStackTrace();
}
throw new IllegalArgumentException("timeout");
}
}