perf[map]: get put remove by primitive way

zfoo-project · Sep 23, 2024 · 628463d · 628463d
1 parent 082c2fa
commit 628463d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/...l/src/main/java/com/zfoo/protocol/collection/concurrent/CopyOnWriteHashMapLongObject.java b/...l/src/main/java/com/zfoo/protocol/collection/concurrent/CopyOnWriteHashMapLongObject.java
@@ -70,12 +70,16 @@ public V get(Object key) {
         return map.get(key);
     }
 
-    public V getPrimitive(long key) {
+    public V get(long key) {
         return map.get(key);
     }
 
     @Override
     public V put(Long key, V value) {
+        return put(key.longValue(), value);
+    }
+
+    public V put(long key, V value) {
         lock.lock();
         try {
             var newMap = newCopyMap();
@@ -89,6 +93,10 @@ public V put(Long key, V value) {
 
     @Override
     public V remove(Object key) {
+        return remove(((Long) key).longValue());
+    }
+
+    public V remove(long key) {
         lock.lock();
         try {
             var newMap = newCopyMap();

diff --git a/protocol/src/main/java/com/zfoo/protocol/util/ThreadUtils.java b/protocol/src/main/java/com/zfoo/protocol/util/ThreadUtils.java
@@ -117,15 +117,15 @@ public static void registerSingleThreadExecutor(Thread thread, Executor executor
     }
 
     public static Executor executorByThreadId(long threadId) {
-        var threadExecutor = threadExecutorMap.getPrimitive(threadId);
+        var threadExecutor = threadExecutorMap.get(threadId);
         return threadExecutor == null ? null : threadExecutor.getValue();
     }
 
     /**
      * search for the corresponding thread by the thread id
      */
     public static Thread findThread(long threadId) {
-        var threadExecutor = threadExecutorMap.getPrimitive(threadId);
+        var threadExecutor = threadExecutorMap.get(threadId);
         if (threadExecutor != null) {
             return threadExecutor.getKey();
         }