Java Semaphore用法

﹏ヽ暗。殇╰゛Y 2022-09-17 11:29 166阅读 0赞

操作系统的信号量是个很重要的概念，在进程控制方面都有应用。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");
    	}
    }