如何在Java中找到兼具唯一性与大小限制的线程安全集合？

核心结论 #

Java标准库中没有原生同时满足「线程安全、元素唯一性、固定大小限制」的集合类，你需要通过自定义逻辑或包装现有集合来实现需求。

对现有方案的补充分析 #

• ConcurrentSkipListSet：线程安全且保证元素唯一，但没有内置的大小限制机制，无法直接实现"达到上限后拒绝插入"的逻辑。
• BlockingQueue实现类（如ArrayBlockingQueue）：线程安全且支持大小限制，但不保证元素唯一性，重复元素会被正常插入。
• 基于ConcurrentHashMap.newKeySet()的自定义实现：这是最贴合需求的可行方案，通过手动结合大小检查与原子性插入操作，能实现线程安全的有界唯一集合。

简单自定义实现示例 #

可以基于ConcurrentHashMap的键集来封装，利用其线程安全特性，同时添加大小检查逻辑：

import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

public class BoundedConcurrentSet<E> {
    private final Set<E> internalSet;
    private final int maxCapacity;

    public BoundedConcurrentSet(int maxCapacity) {
        this.internalSet = ConcurrentHashMap.newKeySet();
        this.maxCapacity = maxCapacity;
    }

    // 线程安全的添加方法：先检查大小，再尝试原子性添加
    public boolean add(E element) {
        if (element == null) {
            throw new NullPointerException("Element cannot be null");
        }
        // 注意：ConcurrentHashMap的size()是近似值，高并发下可能存在短暂的大小偏差
        if (internalSet.size() >= maxCapacity) {
            return false;
        }
        // add方法是原子性的，只有元素不存在时才会返回true
        return internalSet.add(element);
    }

    // 委托其他常用方法
    public boolean contains(E element) {
        return internalSet.contains(element);
    }

    public boolean remove(E element) {
        return internalSet.remove(element);
    }

    public int size() {
        return internalSet.size();
    }
}

注意事项 #

• 上述实现中size()是近似值，在高并发场景下可能出现短暂的超量插入。如果需要严格的大小限制，可以在add方法中引入ReentrantLock来保证检查与插入的原子性，但会带来一定的性能损耗。
• 如果你的场景可以接受"阻塞等待"而非"直接拒绝"，也可以考虑结合Semaphore与ConcurrentSkipListSet，用信号量控制并发插入的数量，同时保证元素唯一性。

内容的提问来源于stack exchange，提问作者Meriem