update concurrent dictionary API

Author: JavaSaBr

rlib-common/src/main/java/com/ss/rlib/common/function/NotNullNullableTripleFunction.java

@@ -0,0 +1,11 @@
+package com.ss.rlib.common.function;
+
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+@FunctionalInterface
+public interface NotNullNullableTripleFunction<F, S, T, R> extends TripleFunction<F, S, T, R> {
+
+    @Override
+    @Nullable R apply(@NotNull F first, @NotNull S second, @NotNull T third);
+}

rlib-common/src/main/java/com/ss/rlib/common/util/dictionary/ConcurrentObjectDictionary.java

@@ -1,9 +1,6 @@
 package com.ss.rlib.common.util.dictionary;
 
-import com.ss.rlib.common.function.NotNullBiConsumer;
-import com.ss.rlib.common.function.NotNullConsumer;
-import com.ss.rlib.common.function.NotNullNullableBiFunction;
-import com.ss.rlib.common.function.NotNullTripleConsumer;
+import com.ss.rlib.common.function.*;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -156,6 +153,30 @@ public interface ConcurrentObjectDictionary<K, V> extends ObjectDictionary<K, V>
         }
     }
 
+    /**
+     * Get the value from a function for this dictionary under block {@link ConcurrentObjectDictionary#readLock()}.
+     *
+     * @param first    the first argument.
+     * @param second   the second argument.
+     * @param function the function.
+     * @param <F>      the first argument's type.
+     * @param <S>      the second argument's type.
+     * @param <R>      the result's type.
+     * @return the result of the function.
+     */
+    default <F, S, R> @Nullable R getInReadLock(
+        @NotNull F first,
+        @NotNull S second,
+        @NotNull NotNullNullableTripleFunction<ConcurrentObjectDictionary<K, V>, F, S, R> function
+    ) {
+        var stamp = readLock();
+        try {
+            return function.apply(this, first, second);
+        } finally {
+            readUnlock(stamp);
+        }
+    }
+
     /**
      * Get the value from a function for this dictionary under block {@link ConcurrentObjectDictionary#writeLock()}.
      *
@@ -177,6 +198,30 @@ public interface ConcurrentObjectDictionary<K, V> extends ObjectDictionary<K, V>
         }
     }
 
+    /**
+     * Get the value from a function for this dictionary under block {@link ConcurrentObjectDictionary#writeLock()}.
+     *
+     * @param first    the first argument.
+     * @param second   the second argument.
+     * @param function the function.
+     * @param <F>      the first argument's type.
+     * @param <S>      the second argument's type.
+     * @param <R>      the result's type.
+     * @return the result of the function.
+     */
+    default <F, S, R> @Nullable R getInWriteLock(
+        @NotNull F first,
+        @NotNull S second,
+        @NotNull NotNullNullableTripleFunction<ConcurrentObjectDictionary<K, V>, F, S, R> function
+    ) {
+        var stamp = writeLock();
+        try {
+            return function.apply(this, first, second);
+        } finally {
+            writeUnlock(stamp);
+        }
+    }
+
     /**
      * Performs the given action for each key-value pair of this dictionary.
      *