add runInWriteLock with two args

Author: crazyrokr

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

@@ -551,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());