Question

我的应用程序中需要一个基于密钥的Semaphore机制，并且由Guava偶然发现Striped.semaphore(int, int)。但是，它的行为并不像预期的那样。

使用以下代码，fetch有时会返回null。两种方法都由不同的线程访问。我希望调用fetch的线程等到地图中的Blubb可用。

private final Striped<Semaphore> semaphores = Striped.semaphore(64, 0);

private final Map<String, Blubb> blubbs = Collections.synchronizedMap(new HashMap<String, Blubb>());

private Semaphore getSemaphore(final String key) {
    return semaphores.get(key);
}

@Override
public void put(String key, Blubb blubb)  {
    blubb.put(key, blubb);
    final Semaphore semaphore = getSemaphore(toUser);
    semaphore.release();
}

@Override
public blubb fetch(final String key) {
    try {
        final Semaphore semaphore = getSemaphore(key);
        final boolean acquired = semaphore.tryAcquire(30, TimeUnit.SECONDS);
        return blubbs.get(key);
    } catch (final InterruptedException e) {
        e.printStackTrace();
    }

    return null;
}

如果我使用以下代码切换回基本Java，一切都按预期工作。

private final Map<String, Semaphore> semaphoresMap = new ConcurrentHashMap<String, Semaphore>();

private Semaphore getSemaphore(final String key) {
    Semaphore semaphore = semaphoresMap.get(key);
    if (semaphore == null) {
        semaphore = new Semaphore(0);
        semaphoresMap.put(key, semaphore);
    }
    return semaphore;
}

我在这里缺少什么？感谢

Answer 1

Guava的Striped指定多个键可能映射到同一个信号量。来自Javadoc：

此类提供的保证是相等的键导致相同的锁（或信号量），即if（key1.equals（key2））then striped.get（key1）== striped.get（key2）（假设正确实现了对象的Object.hashCode（））。请注意，如果key1不等于key2，则无法保证striped.get（key1）！= striped.get（key2）;但是，元素可能会映射到同一个锁。条纹数越少，发生这种情况的概率就越高。

代码中的基本假设似乎是，如果与特定对象关联的信号量具有许可，那么该对象在地图中有一个条目，但事实并非如此 - 如果地图中有条目对于恰好与同一Semaphore相关联的另一个对象，那么fetch可能会在完全不同的对象上执行该许可，该对象实际上在地图中没有条目。

Answer 2

'basic java' 示例有一个潜在的竞争条件，computeIfAbsent 是一个原子操作并解决了这个问题：

private final Map<String, Semaphore> semaphoresMap = new ConcurrentHashMap<String, Semaphore>();

private Semaphore getSemaphore(final String key) {
    return semaphoresMap.computeIfAbsent(key, (String absentKey) -> new Semaphore(0));
}

在ConcurrentHashMap中使用Guava与Semaphores的基于密钥的信号量

2 个答案: