我想将computeIfPresent和putIfAbsent都放入一个原子函数中

Question

workObjectMap.computeIfPresent(key, (k,v) -> {
    v.memberIdSet.addAll(memberIdSet);
    v.memberPositionSet.addAll(memberPositionSet);
    return v;
});
// If it.remove() in run() is called at this point,
// there is a risk of the same work being done twice
workObjectMap.putIfAbsent(key, new WorkObjectValue(memberIdSet, memberPositionSet));

我有上面的代码。在一个单独的线程中，我可能会从workObjectMap. 我想要一个原子函数，如果键存在，它会修改值，如果不存在，它会放置键/值。

在我当前的代码中，删除有可能发生在computeIfPresent和之间putIfAbsent，以便已添加到地图中的相同项目将再次添加。

如何阻止这两个方法调用之间的任何删除？（computeIfPresent和putIfAbsent）

Answer 1

使用Map.compute()：

workObjectMap.compute(key, (k,v) -> {
    if (v == null) return new WorkObjectValue(memberIdSet, memberPositionSet);
    v.memberIdSet.addAll(memberIdSet);
    v.memberPositionSet.addAll(memberPositionSet);
    return v;
});

请注意，JavadocConcurrentHashMap明确指出这是原子的。