信号量:Semaphore
信号量为多线程协作提供了更为强大的控制方法。也可以说,信号量是对锁的扩展。无论是内部锁 synchronized 还是重入锁 ReentrantLock,一次都只允许一个线程访问一个资源,而信号量却可以指定多个线程,同时访问某一个资源。
常用方法介绍:
// 尝试获得一个准入的许可。若无法获得,则线程会等待,直到有线程释放一个许可或者当前线程被中断。 public void acquire() // 与 acquire()基本相同,区别是其不响应中断。public void acquireUninterruptibly() // 尝试获得一个许可,如果成功返回true,失败则返回false,其不会进行等待,立即返回。public boolean tryAcquire() // 与 tryAcquire()基本相同,区别是其增加尝试的时间和时间单位。public boolean tryAcquire(long timeout, TimeUnit unit) // 用于在线程访问资源结束后,释放一个许可,以使其他等待许可的线程可以进行资源访问。public void release() |
使用场景:
Semaphore 可以用于做流量控制,特别是公用资源有限的应用场景,比如数据库连接。假如有一个需求,读取几万个文件的数据,因为都是 IO 密集型任务,我们可以启动几十个线程并发读取,但是如果读到内存后,还需要存储到数据库中,而此时数据库的连接数只有 10 个,此时就需要控制每次只有 10 个线程同时获取数据库连接保存数据,否则将无法获取数据库连接。这个时候,就可以用 Semaphore 来做流量的控制。
import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Semaphore; /** * 信号量测试 */ public class SemaphoreTest implements Runnable{ final Semaphore semp = new Semaphore(5); public static void main(String[] args) { final SemaphoreTest semapTest = new SemaphoreTest(); ExecutorService executorService = Executors.newFixedThreadPool(20); for (int i = 0; i < 20;i++) { executorService.execute(semapTest); } executorService.shutdown(); } public void run() { try { semp.acquire(); Thread.sleep(2000); System.out.println(Thread.currentThread().getName()+",执行完成!"); semp.release(); } catch (InterruptedException e) { e.printStackTrace(); } } } 上述代码中,申请拥有 20 个线程的线程池,和允许 5 个线程获取许可证的信号量。运行结果,每次只有 5 个线程同时获取许可证并运行,也就是控制最大的并发数为 5,使用完之后调用 release()释放许可,归还给信号量。 |