java使用Semaphore实现限流器

Z时代
2024-01-10
分类：综合

美女程序员鼓励师

概念

1、Semaphore可以看作是已经被广泛地翻译成信号量，从概念上讲，信号量保持了一组凭证，获得凭证的线程可以访问资源，使用完成后释放，我们可以使用信号量来限制访问特定资源的并发线程。

2、可以简单概括为：一个计数器，一个等待队列，三种方法。在信号量模型中，计数器和等待队列是透明的，只能通过信号量模型提供的三种方式访问，即互联网、acquire和release。

实例

public class SemaphoreDemo {
 
    static class Link {
    }
 
    static class ObjPool<T, R> {
 
        final List<T> pool;
 
        final Semaphore semaphore;
 
        ObjPool(int size, T t) {
            pool = new Vector<>(size);
            for (int i = 0; i < size; i++) {
                pool.add(t);
            }
            semaphore = new Semaphore(size);
        }
 
        public R exec(Function<T, R> func) throws Exception {
            T t = null;
            semaphore.acquire();
            try {
                System.out.println(Thread.currentThread().getName() + "---------争夺锁--------");
 
                t = pool.remove(0);
                System.out.println(Thread.currentThread().getName() + " 拿到锁执行");
                return func.apply(t);
            } finally {
                pool.add(t);
                semaphore.release();
            }
        }
    }
 
    public static void main(String[] args) {
        ObjPool objPool = new ObjPool(5, new Link());
        for (int i = 0; i < 30; i++) {
            new Thread(() -> {
                try {
                    objPool.exec(t -> t.toString());
                } catch (Exception e) {
                }
            }).start();
        }
    }
}