在并发HashMap中，如何同步更新？

Question

假设，我有如下 concurrentHashMap 代码：

ConcurrentHashMap<Integer,Integer> balances = new ConcurrentHashMap<>();

public void add(int accountId, int amountToAdd){
     int currentBalance = balances.get(accountId);
     balances.put(accountId, currentBalance + amountToAdd);
}

此 add 方法是从多个线程调用的，可以尝试同时更新同一 accountId 的数量。

如何确保currentBalance在< code>get和< code>put之间不发生变化？因为根据我的理解，如果线程在执行< code>get后抢占，同时其他线程更新余额，< code>put将使用旧余额执行更新。

Answer 1

您可以使用 AtomicIntegers 的哈希图：

ConcurrentHashMap<Integer,AtomicInteger> balances = new ConcurrentHashMap<>();

原子整数实现比较和交换操作。

然后:

public void add(int accountId, int amountToAdd){
     balances.get(accountId).addAndGet( amountToAdd );
}

要创建一个新的帐户，并且不要覆盖已经创建的帐户，并且由另一个线程初始化一些金额，使用这个:

public void addNewAccount(int accountId){
     balances.putIfAbsent( accountId, new AtomicInteger(0) );
}

您还可以组合使用两个 Metod 并仅使用一个：

public void add(int accountId, int amountToAdd){
     balances.putIfAbsent( accountId, new AtomicInteger(0) );
     balances.get(accountId).addAndGet( amountToAdd );
}

Answer 2

Java8向Map接口添加了一系列更新方法，并为ConcurrentHashMap提供了原子性保证。在你的情况下，你可以这样做：

public void add(int accountId, int amountToAdd){
    balances.computeIfPresent(accountId, (key, currentBalance) -> currentBalance + amountToAdd);
}

整个方法调用以原子方式执行。在计算过程中，其他线程在此映射上尝试的某些更新操作可能会被阻止，因此计算应简短，并且不得尝试更新此映射的任何其他映射。