問題描述

有下面一段代碼：

package test;import java.util.concurrent.ConcurrentHashMap;import java.util.concurrent.ConcurrentMap;public class TestContain extends Thread{ private final String key = 'key'; private final static ConcurrentMap<String, Object> locks = new ConcurrentHashMap<>();private static Object getLock(String lockName) { if (!locks.containsKey(lockName)) {//這一句會存在并發問題locks.put(lockName, new String('我是值'));System.out.println('加了一次'); } return locks.get(lockName);}@Overridepublic void run() { getLock(this.key);};public static void main(String[] args) { for (int i = 0; i < 20; i++) {new TestContain().start();; }}}

輸出結果：

加了一次加了一次加了一次

表明了Map.containsKey() 在多線程的情況下會判斷不準確。

這是為什么呢？ 有什么方法改進呢？

問題解答

ConcurrentHashMap的doc上有一段

Retrieval operations (including <tt>get</tt>) generally do not block, so may overlap with update operations (including

<tt>put</tt> and <tt>remove</tt>). Retrievals reflect the results of the most recently completed update operations holding upon their onset.

里面的get方法并不加鎖，get方法只是拿到最新完成update的值。

所以題主方法中的locks.containsKey(lockName)沒有鎖來保證線程安全的。而且感覺ConcurrentHashMap的使用場景并不是用containsKey來保證更新操作只進行一次，而是用putIfAbsent來保證。

ConcurrentMap保證的是單次操作的原子性，而不是多次操作。

你的getLock函數中包含了多次操作，ConcurrentMap沒法擴大它的同步范圍，你需要自己實現getLock的鎖。

使用putIfAbsent方法。