update dictionary API

Author: JavaSaBr

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

@@ -43,6 +43,29 @@ public interface ConcurrentObjectDictionary<K, V> extends ObjectDictionary<K, V>
         return DictionaryFactory.newConcurrentStampedLockObjectDictionary();
     }
 
+    /**
+     * Execute a function for this dictionary under block {@link ConcurrentObjectDictionary#readLock()}.
+     *
+     * @param argument the argument.
+     * @param consumer the function.
+     * @param <A>      the argument's type.
+     * @return this dictionary.
+     */
+    default <A> @NotNull ConcurrentObjectDictionary<K, V> runInReadLock(
+        @NotNull A argument,
+        @NotNull NotNullBiConsumer<ConcurrentObjectDictionary<K, V>, A> consumer
+    ) {
+
+        var stamp = readLock();
+        try {
+            consumer.accept(this, argument);
+        } finally {
+            readUnlock(stamp);
+        }
+
+        return this;
+    }
+
     /**
      * Execute a function for this dictionary under block {@link ConcurrentObjectDictionary#writeLock()}.
      *
@@ -118,11 +141,12 @@ public interface ConcurrentObjectDictionary<K, V> extends ObjectDictionary<K, V>
      * @param argument the argument.
      * @param function the function.
      * @param <A>      the argument's type.
+     * @param <R>      the result's type.
      * @return the result of the function.
      */
-    default <A> @Nullable V getInReadLock(
+    default <A, R> @Nullable R getInReadLock(
         @NotNull A argument,
-        @NotNull NotNullNullableBiFunction<ConcurrentObjectDictionary<K, V>, A, V> function
+        @NotNull NotNullNullableBiFunction<ConcurrentObjectDictionary<K, V>, A, R> function
     ) {
         var stamp = readLock();
         try {
@@ -138,11 +162,12 @@ public interface ConcurrentObjectDictionary<K, V> extends ObjectDictionary<K, V>
      * @param argument the argument.
      * @param function the function.
      * @param <A>      the argument's type.
+     * @param <R>      the result's type.
      * @return the result of the function.
      */
-    default <A> @Nullable V getInWriteLock(
+    default <A, R> @Nullable R getInWriteLock(
         @NotNull A argument,
-        @NotNull NotNullNullableBiFunction<ConcurrentObjectDictionary<K, V>, A, V> function
+        @NotNull NotNullNullableBiFunction<ConcurrentObjectDictionary<K, V>, A, R> function
     ) {
         var stamp = writeLock();
         try {

rlib-common/src/test/java/com/ss/rlib/common/test/util/dictionary/ConcurrentObjectDictionaryTest.java

@@ -44,4 +44,19 @@ void runInWriteLockTest() {
             assertType(arg2, Integer.class);
         });
     }
+
+    @Test
+    void runInReadLockTest() {
+
+        var dictionary = ConcurrentObjectDictionary.ofType(
+            String.class,
+            Integer.class
+        );
+
+        dictionary.runInReadLock("4", (dic, arg) -> {
+            assertType(dic, ConcurrentObjectDictionary.class);
+            assertType(arg, String.class);
+            dic.get(arg);
+        });
+    }
 }