在检查ConcurrentHashMap中是否存在密钥时,是否需要客户端锁定?

时间:2017-06-25 01:08:32

标签: java multithreading concurrency concurrenthashmap

我知道我可以使用ConcurrentHashMap的putIfAbsent。但是,我需要进行webservice调用以获取给定键的值(如果它不存在)然后存储它(缓存类型),这样我就不需要在下次使用相同的键时。以下哪项是正确的?我认为同步需要第二个版本。

更新1:我无法使用任何功能界面。

更新2:根据Costi Ciudatu的回复更新代码段

private static final Map<String, String> KEY_VALUE_MAP = new ConcurrentHashMap<>();
public String getValueVersion1(String key) {
    String value  = KEY_VALUE_MAP.get(key);
    if (value != null) {
        return value;
    }

    // Else fetch the value for the key from the webservice.
    value = doRestCall(key);
    KEY_VALUE_MAP.putIfAbsent(key, value);

    return value;
} // Version 1 Ends here.

public synchronized String getValueVersion2(String key) {
    String value  = KEY_VALUE_MAP.get(key);
    if (value != null) {
        return value;
    }

    // Else fetch the value for the key from the webservice.
    value = doRestCall(key);
    KEY_VALUE_MAP.put(key, value);
    return value;
} // Version 2 ends here.

1 个答案:

答案 0 :(得分:7)

你应该看看ConcurrentMap#computeIfAbsent,它原子地为你做这件事:

return KEY_VALUE_MAP.computeIfAbsent(key, this::doRestCall);

编辑(解决“无功能接口”约束):

如果您想确保只为任何给定的密钥调用doRestCall一次,则只需要“客户端锁定”。否则,这段代码可以正常工作:

final String value = KEY_VALUE_MAP.get(key);
if (value == null) {
    // multiple threads may call this in parallel
    final String candidate = doRestCall(key);
    // but only the first result will end up in the map
    final String winner = KEY_VALUE_MAP.putIfAbsent(key, candidate);
    // local computation result gets lost if another thread made it there first
    // otherwise, our "candidate" is the "winner"
    return winner != null ? winner : candidate;
}
return value;

但是,如果您确实希望强制执行doRestCall仅针对任何给定密钥调用 (我的猜测 确实需要此),你需要某种同步。但试着比你的例子中的“全有或全无”方法更有创意:

final String value = KEY_VALUE_MAP.get(key);
if (value != null) {
    return value;
}
synchronized(KEY_VALUE_MAP) {
    final String existing = KEY_VALUE_MAP.get(key);
    if (existing != null) {  // double-check
        return existing;
    }
    final String result = doRestCall(key);
    KEY_VALUE_MAP.put(key, result);  // no need for putIfAbsent
    return result;
}

如果您想使用第二种(偏执)方法,您还可以考虑使用密钥本身进行锁定(将范围缩小到最小)。但这可能需要您管理自己的密钥池,因为syncrhonized (key.intern())不是好的做法。

这完全取决于您的doRestCall()方法永远不会返回null这一事实。否则,您必须将地图值包装在Optional(或某些自制的pre-java8替代版本)中。

作为(最后)附注,在您的代码示例中,您反转了put()putIfAbsent()的使用(后者是没有外部同步时使用的那个)并且您读取了两次值用于空检查。