-
-
Notifications
You must be signed in to change notification settings - Fork 44
/
CHashMapReplaceClear.java
35 lines (29 loc) · 1 KB
/
CHashMapReplaceClear.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package nallar.collections;
import nallar.exception.ConcurrencyError;
import nallar.tickthreading.util.BooleanThreadLocalDefaultFalse;
import java.util.concurrent.*;
public class CHashMapReplaceClear<K, V> extends CHashMap<K, V> {
private final BooleanThreadLocalDefaultFalse inReplace = new BooleanThreadLocalDefaultFalse();
private ConcurrentHashMap<K, V> replaceMap;
@Override
public synchronized V put(K key, V value) {
if (inReplace.get() == Boolean.TRUE) {
return replaceMap.put(key, value);
} else {
return super.put(key, value);
}
}
public synchronized void start() {
if (inReplace.getAndSet(true) != Boolean.FALSE || replaceMap != null) {
throw new ConcurrencyError("Already replacing");
}
replaceMap = new ConcurrentHashMap<K, V>();
}
public synchronized void done() {
if (inReplace.getAndSet(false) != Boolean.TRUE || replaceMap == null) {
throw new ConcurrencyError("Not yet replacing");
}
hashMap = replaceMap;
replaceMap = null;
}
}