Merge remote-tracking branch 'github/develop' into feature-implement_ssl_network

Author: JavaSaBr

rlib-common/src/main/java/com/ss/rlib/common/util/array/ConcurrentArray.java

@@ -353,6 +353,31 @@ default void writeUnlock(long stamp) {
         }
     }
 
+    /**
+     * Execute a function to get some result under {@link #readLock()} block.
+     *
+     * @param first    the first argument for the function.
+     * @param second   the second argument for the function.
+     * @param function the function.
+     * @param <A>      the first argument's type.
+     * @param <T>      the second argument's type.
+     * @param <R>      the result's type.
+     * @return the result from the function.
+     * @since 9.9.0
+     */
+    default <A, T, R> @Nullable R getInReadLock(
+        @NotNull A first,
+        @NotNull T second,
+        @NotNull NotNullNullableTripleFunction<ConcurrentArray<E>, A, T, R> function
+    ) {
+        var stamp = readLock();
+        try {
+            return function.apply(this, first, second);
+        } finally {
+            readUnlock(stamp);
+        }
+    }
+
     /**
      * Execute a function to get some result under {@link #writeLock()} block.
      *
@@ -391,6 +416,31 @@ default void writeUnlock(long stamp) {
         }
     }
 
+    /**
+     * Execute a function to get some result under {@link #writeLock()} block.
+     *
+     * @param first    the first argument for the function.
+     * @param second   the second argument for the function.
+     * @param function the function.
+     * @param <A>      the first argument's type.
+     * @param <T>      the second argument's type.
+     * @param <R>      the result's type.
+     * @return the result from the function.
+     * @since 9.9.0
+     */
+    default <A, T, R> @Nullable R getInWriteLock(
+        @NotNull A first,
+        @NotNull T second,
+        @NotNull NotNullNullableTripleFunction<ConcurrentArray<E>, A, T, R> function
+    ) {
+        var stamp = writeLock();
+        try {
+            return function.apply(this, first, second);
+        } finally {
+            writeUnlock(stamp);
+        }
+    }
+
     /**
      * Execute a function under {@link #readLock()} block.
      *
@@ -433,6 +483,33 @@ default void writeUnlock(long stamp) {
         return this;
     }
 
+    /**
+     * Execute a function under {@link #readLock()} block.
+     *
+     * @param <A>      the first argument's type.
+     * @param <T>      the second argument's type.
+     * @param first    the first argument.
+     * @param second   the second argument.
+     * @param function the function.
+     * @return this array.
+     * @since 9.9.0
+     */
+    default <A, T> @NotNull ConcurrentArray<E> runInReadLock(
+        @NotNull A first,
+        @NotNull T second,
+        @NotNull NotNullTripleConsumer<ConcurrentArray<E>, A, T> function
+    ) {
+
+        var stamp = readLock();
+        try {
+            function.accept(this, first, second);
+        } finally {
+            readUnlock(stamp);
+        }
+
+        return this;
+    }
+
     /**
      * Execute a function under {@link #writeLock()} block.
      *
@@ -474,6 +551,31 @@ default void writeUnlock(long stamp) {
         return this;
     }
 
+    /**
+     * Execute a function under {@link #writeLock()} block.
+     *
+     * @param first    the first argument for the function.
+     * @param second   the second argument for the function.
+     * @param function the function.
+     * @param <A>      the first argument's type.
+     * @param <T>      the second argument's type.
+     * @return this array.
+     * @since 9.9.0
+     */
+    default <A, T> @NotNull ConcurrentArray<E> runInWriteLock(
+        @NotNull A first,
+        @NotNull T second,
+        @NotNull NotNullTripleConsumer<ConcurrentArray<E>, A, T> function
+    ) {
+        var stamp = writeLock();
+        try {
+            function.accept(this, first, second);
+        } finally {
+            writeUnlock(stamp);
+        }
+        return this;
+    }
+
     /**
      * Search an element using the condition under {@link #readLock()} block.
      *

rlib-common/src/test/java/com/ss/rlib/common/test/util/array/ConcurrentArrayTest.java

@@ -241,6 +241,8 @@ void runInWriteLockTest() {
 
         var array = ConcurrentArray.of("First", "Second", "Third", "  ");
 
+        array.runInWriteLock(1, "Second", (arr, first, second) -> assertEquals(second, arr.get(first)));
+
         array.runInWriteLock(object -> object.remove("Second"));
 
         assertEquals(3, array.size());
@@ -270,6 +272,13 @@ void runInReadLockTest() {
 
         array.runInReadLock("Third", (arr, arg) ->
             assertEquals(2, arr.count(arg, String::equals)));
+
+        array.runInReadLock(2, "Third",  (arr, first, second) -> {
+            assertType(arr, Array.class);
+            assertIntType(first);
+            assertType(second, String.class);
+            assertEquals(first, arr.count(second, String::equals));
+        });
     }
 
     @Test
@@ -279,6 +288,10 @@ void getInWriteLockTest() {
 
         assertEquals("Third", array.getInWriteLock(arr -> arr.get(2)));
         assertEquals("Second", array.getInWriteLock(1, Array::get));
+        assertEquals(
+            "FirstSecond",
+            array.getInWriteLock(0, 1, (arr, first, second) -> arr.get(first) + arr.get(second))
+        );
     }
 
     @Test
@@ -288,6 +301,10 @@ void getInReadLockTest() {
 
         assertEquals("Third", array.getInReadLock(arr -> arr.get(2)));
         assertEquals("Second", array.getInReadLock(1, Array::get));
+        assertEquals(
+            "FirstSecond",
+            array.getInReadLock(0, 1, (arr, first, second) -> arr.get(first) + arr.get(second))
+        );
     }
 
     @Test