在java中是否存在BlockingMap作为BlockingQueue？

Question

我想要一个与BlockingQueue非常相似的BlockingMap数据结构.BlockingQueue的take方法将在那里等待,直到元素可用.我希望BlockingMap的get方法在那里等待,直到相应的密钥可用？我可以使用这种数据结构吗？

Answer 1

我过去只是用过BlockingQueue<Map.Entry<K,V>>.但最近,我遇到了这个阻止Java的地图.但是,我自己没有用过它.

Answer 2

这是一个使用 BlockingQueue 和 ConcurrentHashMap 的极其简单的实现：

public class BlockingMap<K, V> {
    private Map<K, ArrayBlockingQueue<V>> map = new ConcurrentHashMap<>();

    private BlockingQueue<V> getQueue(K key, boolean replace) {
        return map.compute(key, (k, v) -> replace || v == null ? new ArrayBlockingQueue<>(1) : v);
    }

    public void put(K key, V value) {
        getQueue(key, true).add(value);
    }

    public V get(K key) throws InterruptedException {
        return getQueue(key, false).take();
    }

    public V get(K key, long timeout, TimeUnit unit) throws InterruptedException {
        return getQueue(key, false).poll(timeout, unit);
    }
}