Java：如何以原子方式替换Map中的所有值？

Question

我在多线程环境中有一个有状态的Bean，它将其状态保存在映射中。现在，我需要一种在一次原子操作中替换该映射的所有值的方法。

public final class StatefulBean {

    private final Map<String, String> state = new ConcurrentSkipListMap<>();

    public StatefulBean() {
        //Initial state
        this.state.put("a", "a1");
        this.state.put("b", "b1");
        this.state.put("c", "c1");
    }

    public void updateState() {
        //Fake computation of new state
        final Map<String, String> newState = new HashMap<>();
        newState.put("b", "b1");
        newState.put("c", "c2");
        newState.put("d", "d1");

        atomicallyUpdateState(newState);
        /*Expected result
         *  a: removed
         *  b: unchanged
         *  C: replaced
         *  d: added*/
    }

    private void atomicallyUpdateState(final Map<String, String> newState) {
        //???
    }
}

目前，我ConcurrentSkipListMap用作的实现ConcurrentMap，但这不是必需的。

我认为解决此问题的唯一方法是制作全局state
volatile地图并完全替换地图或使用AtomicReferenceFieldUpdater。有没有更好的办法？

我的更新非常频繁，每秒一次或两次，但是机会很少。同样，整个地图只会包含少于20个值。

Answer 1

使用CAS的方法AtomicReference是在每次批量更新时复制地图内容。

AtomicReference<Map<String, String>> workingMapRef = new AtomicReference<>(new HashMap<>());

该映射可以是并发的，但是对于“批量更新”它是只读的。然后updateState循环执行，doUpdateState()直到得到正确的结果，这意味着您的值已更新。

void updateState() {
    while (!doUpdateState());
}

boolean doUpdateState() {
    Map<String, String> workingMap = workingMapRef.get();
    //copy map content
    Map<String, String> newState = new HashMap<>(workingMap); //you can make it concurrent

    newState.put("b", "b1");
    newState.put("c", "c2");
    newState.put("d", "d1");

    return workingMapRef.compareAndSet(workingMap, newState);
}