在本文中，我們看到了不同的創建線程安全的Set的方式，也比較了他們之間的差異性。 所以大家以后使用的時候，可以考慮使用ConcurrentHashMap創建的Set。
大家好，我是指北君。
線程安全的問題，真的算是老生常談了。這幾天看到一個 HashSet 線程安全的騷操作，在這里分享給大家。 在本文中，我們將分享如何構造線程安全的HashSet的幾種方法。
使用ConcurrentHashMap工廠方法構造線程安全的HashSet
首先, 我們來看看_ConcurrentHashMap_暴露出來的靜態方法 -- newKeySet()。此方法返回一個Set的實例，等同于實現了 java.util.Set 接口，而且能夠使用Set的一些常用操作，比如 add(), contains() 等。
舉個例子：
Set<Integer> threadSafeUniqueNumbers = ConcurrentHashMap.newKeySet();
threadSafeUniqueNumbers.add(23);
threadSafeUniqueNumbers.add(45);
這里返回的Set，其實有點類似于 HashSet，因為兩者都是基于Hash算法實現的，另外線程同步邏輯帶來的額外開銷也很小，因為它最終還是 ConcurrentHashMap 的一部分。
不過，這個只能在 Java 8 以上版本才可以使用，我想大部分公司應該至少 Java 8 了吧。直接拿來用就行。
現在，我們已經了解了可以用 ConcurrentHashMap#newKeySet()構建類似于線程安全的HashSet，在 ConcurrentHashMap 其實被定義為 KeySetView。ConcurrentHashMap 其實還有兩個實例方法可以用于構建 KeySetView, 一個是 keySet() 另外一個就是keySet(defaultValue), 我這里就簡寫一下了， 大家可以在IDE中直接打出來看看。
這兩個方法都可以創建KeySetView的實例，KeySetView 與 Map 是一個連接的關系。 我們每次向Map中添加新的鍵值對的時候，Set中的數據也在相應的添加，我們通過幾個例子來看看這兩種方法有哪些區別。
KeySet() 方法
keySet() 方法 和 keySet(defaultValue) ，最大的區別就是不能直接往Set中添加數據。直接添加的話，會拋出 UnsupportedOperationException 異常，源碼中的定義如下。
public KeySetView<K,V> keySet(){
KeySetView<K,V> ks;
if ((ks = keySet) != null) return ks;
return keySet = new KeySetView<K,V>(this, null);
}
// add
public boolean add(K e){
V v;
if ((v = value) == null)
throw new UnsupportedOperationException();
return map.putVal(e, v, true) == null;
}所以我們只能通過如下的方式使用。
ConcurrentHashMap<Integer,String> numbersMap = new ConcurrentHashMap<>();
Set<Integer> numbersSet = numbersMap.keySet();
numbersMap.put(1, "One");
numbersMap.put(2, "Two");
numbersMap.put(3, "Three");
System.out.println("Map before remove: "+ numbersMap);
System.out.println("Set before remove: "+ numbersSet);
numbersSet.remove(2);
System.out.println("Set after remove: "+ numbersSet);
System.out.println("Map after remove: "+ numbersMap);
輸出結果如下。
Map before remove: {1=One, 2=Two, 3=Three}
Set before remove: [1, 2, 3]
Set after remove: [1, 3]
Map after remove: {1=One, 3=Three}KeySet(defaultValue) 方法
keySet(defaultValue) ,由于有設置默認的value，可以在添加的時候不會報錯，JDK 源碼縱定義如下：
public KeySetView<K,V> keySet(V mappedValue){
if (mappedValue == null)
throw new NullPointerException();
return new KeySetView<K,V>(this, mappedValue);
}所以我們可以通過如下的方式使用。
ConcurrentHashMap<Integer,String> numbersMap = new ConcurrentHashMap<>();
Set<Integer> numbersSet = numbersMap.keySet("SET-ENTRY");
numbersMap.put(1, "One");
numbersMap.put(2, "Two");
numbersMap.put(3, "Three");
System.out.println("Map before add: "+ numbersMap);
System.out.println("Set before add: "+ numbersSet);
numbersSet.addAll(asList(4,5));
System.out.println("Map after add: "+ numbersMap);
System.out.println("Set after add: "+ numbersSet);
輸出結果如下：
Map before add: {1=One, 2=Two, 3=Three}
Set before add: [1, 2, 3]
Map after add: {1=One, 2=Two, 3=Three, 4=SET-ENTRY, 5=SET-ENTRY}
Set after add: [1, 2, 3, 4, 5]使用Collections的來創建線程安全的 Set
java.util.Collections 中有一個線程同步的方法可以用于創建，示例代碼如下。
Set<Integer> syncNumbers = Collections.synchronizedSet(new HashSet<>());
syncNumbers.add(1);
這個方法的性能并沒有ConcurrentHashMap的那個效率高，由于使用了同步鎖，增加了一些額外的開銷。
使用CopyOnWriteArraySet構建線程安全的 Set
用CopyOnWriteArraySet 創建線程安全的 set 也是非常簡單的。示例代碼如下
Set<Integer> copyOnArraySet = new CopyOnWriteArraySet<>();
copyOnArraySet.add(1);
這個方法從性能的角度上來看，也不是很理想，CopyOnWriteArraySet 背后的實現是CopyOnWriteArrayList, 最終使用了數組來存儲數據，也就意味著 contains() 或者 remove() 操作，具有 O(n) 的復雜度，而使用HashMap 復雜度為 O(1) 。
建議使用此實現時，設置大小通常保持較小，只讀操作占大多數。
總結
在本文中，我們看到了不同的創建線程安全的Set的方式，也比較了他們之間的差異性。 所以大家以后使用的時候，可以考慮使用ConcurrentHashMap創建的Set。
