JUC之Semaphore的使用

Semaphore：信号量，用来控制同时访问某个特定资源的操作数量，可以用来实现资源池以及也可以为容器添加边界。

Semaphore中管理着一组虚拟的许可，许可的初始数量可以通过构造方法来指定，在执行操作时首先获取许可(只要还有剩余许可)，使用完毕后释放许可。

acquire()：用来获取许可的，如果许可为0，都被使用了，将阻塞到直到有许可（或中断或超时异常）

release()：释放许可，并把许可放回到信号量里。（如果此时acquire()阻塞，就会立马释放阻塞继续执行）

咱们接下来就用Semaphore实现下有界阻塞队列，代码如下：

/**
 * @Author df
 * @Date 2022/5/25 11:07
 * @Version 1.0
 * 使用Semaphore为容器设置边界，有界阻塞容器
 * Semaphore还可实现资源池，
 * 1.初始化构造时指定信号量数量
 * 2.使用时通过acquire()获取一个许可，如果还有剩余的许可，则可继续，如果没有则阻塞直到有许可可用
 * 3.通过release()释放许可，release()一次则信号量里就有一个许可可用
 */
public class BoundedHashSet {
    private final Set<String> set;
    private final Semaphore sem;

    // bound：形成...边界或界限，这里指几个边界数量
    public BoundedHashSet(int bound) {
        this.set = Collections.synchronizedSet(new HashSet<String>());
        this.sem = new Semaphore(bound);
    }

   
    public boolean add(String str) throws InterruptedException {
        // 获取一个许可
        sem.acquire();
        System.out.println("add:获得许可finished");
        boolean wasAdded = false;
        try {
            wasAdded = set.add(str);
            System.out.println("add:添加容器数据：" + str);
            return wasAdded;
        } finally {
            // 没添加成功需释放许可
            if (wasAdded == false) {
                // 释放许可
                sem.release();
                System.out.println("add:添加失败释放许可：" + str);
            }
        }

    }

    public boolean remove(String str) {
        boolean wasRemoved = set.remove(str);
        if (wasRemoved) sem.release();
        return wasRemoved;
    }

    public static void main(String[] args) {
        // 此处设置一个许可，只可添加一次数据，如在调用add则方法被阻塞，直到从容器中remove并释放锁
        BoundedHashSet boundedHashSet = new BoundedHashSet(1);
        try {
            boundedHashSet.add("xdf");
            //boundedHashSet.remove("xdf");
            boundedHashSet.add("xxh");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

}

1.首先构造方法里初始化信号量和存储数据的Set容器

2.定义add方法实现阻塞队列的添加，进入方法首先获取一个许可，然后存储元素到容器里，如果存储失败则释放许可，最后返回是否存储成功标识

3.定义remove方法实现阻塞队列的元素移除的工作，当移除成功时释放一个许可

4.添加main方法测试下，定义一个信号量测试存储第二个元素阻塞的情况

结果展示：

一直在阻塞中，除非存储两个就设置两个信号量，或者移除一个

 设置两个信号量就不会阻塞了