一、背景
在实际项目中,我们经常会遇到一些资源访问限制在一定量的情况。Java中提供了Semaphore 通常用于限制可以访问某些资源(物理或逻辑的)的线程数目。
二、代码实战
Semaphore中的acquire()获取一个许可,释放给定数目的许可,将其返回到信号量。设置信号量为5,启动10个线程访问资源,从运行结果中可以看出。当前5个线程都获取到了许可后,后面的线程必须等待前5个中的一个释放信号量才能够进入执行业务逻辑。
import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Semaphore; public class SemaPhoreTest { public static void main(String[] args) { // 线程池 ExecutorService exec = Executors.newCachedThreadPool(); // 只能5个线程同时访问 final Semaphore semp = new Semaphore(5); // 模拟10个客户端访问 for (int index = 0; index < 10; index++) { final int NO = index; Runnable run = new Runnable() { public void run() { try { // 获取许可 semp.acquire(); System.out.println("Accessing: " + NO); Thread.sleep((long) (Math.random() * 6000)); System.out.println("执行完成"+ NO); // 访问完后,释放 semp.release(); //availablePermits()指的是当前信号灯库中有多少个可以被使用 System.out.println("-----------------" + semp.availablePermits()); } catch (InterruptedException e) { e.printStackTrace(); } } }; exec.execute(run); } // 退出线程池 exec.shutdown(); } } 执行结果:
Accessing: 1 Accessing: 4 Accessing: 3 Accessing: 2 Accessing: 0 执行完成3 -----------------1 Accessing: 5 执行完成5 -----------------1 Accessing: 9 执行完成0 Accessing: 7 -----------------0 执行完成1 -----------------1 Accessing: 8 执行完成4 -----------------1 Accessing: 6 执行完成9 -----------------1 执行完成2 -----------------2 执行完成6 -----------------3 执行完成8 -----------------4 执行完成7 -----------------5