feat[concurrent]: add concurrent primitive type CopyOnWriteHashMapLongObject

protocol/src/main/java/com/zfoo/protocol/collection/concurrent/ConcurrentHashSet.java

@@ -23,15 +23,7 @@ import java.util.concurrent.ConcurrentHashMap;
  */
 public class ConcurrentHashSet<E> extends AbstractSet<E> {
 
-    private final Map<E, Boolean> map;
-
-    public ConcurrentHashSet() {
-        this.map = new ConcurrentHashMap<>();
-    }
-
-    public ConcurrentHashSet(int initialCapacity) {
-        this.map = new ConcurrentHashMap<>(initialCapacity);
-    }
+    private final Map<E, Boolean> map= new ConcurrentHashMap<>();
 
     @Override
     public Iterator<E> iterator() {

protocol/src/main/java/com/zfoo/protocol/collection/concurrent/CopyOnWriteHashMapLongObject.java

@@ -0,0 +1,133 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.collection.concurrent;
+
+import io.netty.util.collection.LongObjectHashMap;
+
+import java.util.Collection;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.locks.ReentrantLock;
+
+/**
+ * @author godotg
+ * @version 3.0
+ */
+public class CopyOnWriteHashMapLongObject<V> implements Map<Long, V> {
+
+    private final ReentrantLock lock = new ReentrantLock();
+    private volatile LongObjectHashMap<V> map = new LongObjectHashMap<>();
+
+    private LongObjectHashMap<V> newCopyMap() {
+        var newMap = new LongObjectHashMap<V>();
+        newMap.putAll(map);
+        return newMap;
+    }
+
+    private void setNewMap(LongObjectHashMap<V> newMap) {
+        map = newMap;
+    }
+
+    @Override
+    public int size() {
+        return map.size();
+    }
+
+    @Override
+    public boolean isEmpty() {
+        return map.isEmpty();
+    }
+
+    @Override
+    public boolean containsKey(Object key) {
+        return map.containsKey(key);
+    }
+
+    @Override
+    public boolean containsValue(Object value) {
+        return map.containsValue(value);
+    }
+
+    @Override
+    public V get(Object key) {
+        return map.get(key);
+    }
+
+    public V getPrimitive(long key) {
+        return map.get(key);
+    }
+
+    @Override
+    public V put(Long key, V value) {
+        lock.lock();
+        try {
+            var newMap = newCopyMap();
+            var oldValue = newMap.put(key, value);
+            setNewMap(newMap);
+            return oldValue;
+        } finally {
+            lock.unlock();
+        }
+    }
+
+    @Override
+    public V remove(Object key) {
+        lock.lock();
+        try {
+            var newMap = newCopyMap();
+            var oldValue = newMap.remove(key);
+            setNewMap(newMap);
+            return oldValue;
+        } finally {
+            lock.unlock();
+        }
+    }
+
+    @Override
+    public void putAll(Map<? extends Long, ? extends V> m) {
+        lock.lock();
+        try {
+            var newMap = newCopyMap();
+            newMap.putAll(m);
+            setNewMap(newMap);
+        } finally {
+            lock.unlock();
+        }
+    }
+
+    @Override
+    public void clear() {
+        lock.lock();
+        try {
+            var newMap = newCopyMap();
+            setNewMap(newMap);
+        } finally {
+            lock.unlock();
+        }
+    }
+
+    @Override
+    public Set<Long> keySet() {
+        return map.keySet();
+    }
+
+    @Override
+    public Collection<V> values() {
+        return map.values();
+    }
+
+    @Override
+    public Set<Entry<Long, V>> entrySet() {
+        return map.entrySet();
+    }
+}

protocol/src/test/java/com/zfoo/protocol/collection/ConcurrentTest.java

@@ -0,0 +1,64 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.collection;
+
+import com.zfoo.protocol.collection.concurrent.CopyOnWriteHashMapLongObject;
+import org.junit.Assert;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import java.util.concurrent.CountDownLatch;
+
+/**
+ * @author godotg
+ * @version 3.0
+ */
+@Ignore
+public class ConcurrentTest {
+
+    @Test
+    public void test() throws InterruptedException {
+        var map = new CopyOnWriteHashMapLongObject<Integer>();
+        var num = 1_0000;
+        var executorSize = Runtime.getRuntime().availableProcessors();
+        var countDownLatch = new CountDownLatch(executorSize);
+        for (var i = 0; i < executorSize; i++) {
+            new Thread(new Runnable() {
+                @Override
+                public void run() {
+                    for (int j = 0; j < num; j++) {
+                        map.put((long) j, j);
+                    }
+                    countDownLatch.countDown();
+                }
+            }).start();
+        }
+        countDownLatch.await();
+        Assert.assertEquals(map.size(), num);
+
+        var countDownLatch2 = new CountDownLatch(executorSize);
+        for (var i = 0; i < executorSize; i++) {
+            new Thread(new Runnable() {
+                @Override
+                public void run() {
+                    for (int j = 0; j < num; j++) {
+                        map.remove((long) j);
+                    }
+                    countDownLatch2.countDown();
+                }
+            }).start();
+        }
+        countDownLatch2.await();
+        Assert.assertTrue(map.isEmpty());
+    }
+}