java - 对不可变 Map 的无锁原子更新

Question

给定一个 Javaslang / Vavr不可变映射，以及一个更新该映射的函数：

private Map<Foo, Bar> myMap = HashMap.empty();

public void setBar(Foo foo, Bar bar) {
  myMap = myMap.put(foo, bar);
}

如何确保setBar()对不同Foo密钥的两个并发调用都会记录其更新？

// thread A
setBar(fooA, barA)

// thread B
setBar(fooB, barB)

似乎存在调用交错的风险，这样：

1. 线程 A 得到{}
2. 线程 B 得到{}
3. 线程 B 计算{}+ fooB -> barB={(fooB -> barB)}
4. 线程 B 设置myMap为{(fooB -> barB)}
5. 线程 A 计算{}+ fooA -> barA={(fooA -> barA)}
6. 线程 A 设置myMap为{(fooA -> barA)}
7. 线程 B 的更新丢失

使用AtomicReference，我或多或少地基于Java Concurrency in PracticeConcurrentStack的“非阻塞算法”部分中的方法提出了以下内容。

private AtomicReference<Map<Foo, Bar>> myMap = 
  new AtomicReference<>(HashMap.empty());

public void setBar(Foo foo, Bar bar) {
  Map<Foo, Bar> myMap0;
  Map<Foo, Bar> myMap1;
  do {
    myMap0 = myMap.get();
    myMap1 = myMap0.put(foo, bar);
  } while (!myMap.compareAndSet(myMap0, myMap1));
}

这个对吗？如果是这样，它是我可能得到的那样好的实现，还是有一些更简单的东西（例如AtomicReference，我缺少一些实现这种模式的 Java 8 API）？

Answer 1

在这种情况下使用AtomicReference很好。您可以使用快捷方式

public void setBar(Foo foo, Bar bar) {
    myMap.updateAndGet(map -> map.put(foo, bar)));
}

反而。请参阅.java 文档AtomicReference.updateAndGet。默认的 java 实现与您在 Java 8 中的实现完全相同。