refactor: delegate resetDefaultExecutor() to CffuFactoryBuilder, …

…and avoiding re-wrapping
foldright · Aug 3, 2024 · 639a703 · 639a703
1 parent 0a5d113
commit 639a703
Show file tree

Hide file tree

Showing 6 changed files with 80 additions and 14 deletions.
diff --git a/cffu-core/src/main/java/io/foldright/cffu/CffuFactory.java b/cffu-core/src/main/java/io/foldright/cffu/CffuFactory.java
@@ -63,15 +63,15 @@ public final class CffuFactory {
      */
     @Contract(pure = true)
     public static CffuFactoryBuilder builder(Executor defaultExecutor) {
-        return new CffuFactoryBuilder(CompletableFutureUtils.screenExecutor(defaultExecutor));
+        return new CffuFactoryBuilder(defaultExecutor);
     }
 
     /**
      * Returns a new CffuFactory from this CffuFactory that reset the defaultExecutor.
      */
     @Contract(pure = true)
-    CffuFactory resetDefaultExecutor(Executor defaultExecutor) {
-        return new CffuFactory(defaultExecutor, forbidObtrudeMethods);
+    public CffuFactory resetDefaultExecutor(Executor defaultExecutor) {
+        return CffuFactoryBuilder.resetDefaultExecutor(this, defaultExecutor);
     }
 
     @Contract(pure = true)

diff --git a/cffu-core/src/main/java/io/foldright/cffu/CffuFactoryBuilder.java b/cffu-core/src/main/java/io/foldright/cffu/CffuFactoryBuilder.java
@@ -2,6 +2,7 @@
 
 import io.foldright.cffu.spi.ExecutorWrapperProvider;
 import org.jetbrains.annotations.Contract;
+import org.jetbrains.annotations.VisibleForTesting;
 
 import javax.annotation.concurrent.ThreadSafe;
 import java.util.List;
@@ -28,7 +29,43 @@ public final class CffuFactoryBuilder {
     private volatile boolean forbidObtrudeMethods = false;
 
     CffuFactoryBuilder(Executor defaultExecutor) {
-        this.defaultExecutor = wrapExecutor(defaultExecutor);
+        this.defaultExecutor = makeExecutor(defaultExecutor);
+    }
+
+    private static Executor makeExecutor(Executor executor) {
+        // check CffuMadeExecutor interface to avoid re-wrapping.
+        if (executor instanceof CffuMadeExecutor) return executor;
+
+        Executor wrapByProviders = wrapExecutorByProviders(CompletableFutureUtils.screenExecutor(executor));
+        return CffuMadeExecutor.wrapMadeInterface(wrapByProviders);
+    }
+
+    /**
+     * An interface for avoiding re-wrapping.
+     */
+    @VisibleForTesting
+    interface CffuMadeExecutor extends Executor {
+        static CffuMadeExecutor wrapMadeInterface(Executor executor) {
+            return new CffuMadeExecutor() {
+                @Override
+                public Executor unwrap() {
+                    return executor;
+                }
+
+                @Override
+                public void execute(Runnable command) {
+                    executor.execute(command);
+                }
+
+                @Override
+                public String toString() {
+                    return "CffuMadeExecutor of executor(" + executor + ")";
+                }
+            };
+        }
+
+        @VisibleForTesting
+        Executor unwrap();
     }
 
     ////////////////////////////////////////////////////////////////////////////////
@@ -57,7 +94,12 @@ public CffuFactory build() {
         return new CffuFactory(defaultExecutor, forbidObtrudeMethods);
     }
 
-    private static Executor wrapExecutor(Executor executor) {
+    @Contract(pure = true)
+    static CffuFactory resetDefaultExecutor(CffuFactory fac, Executor defaultExecutor) {
+        return new CffuFactory(makeExecutor(defaultExecutor), fac.forbidObtrudeMethods());
+    }
+
+    private static Executor wrapExecutorByProviders(Executor executor) {
         for (ExecutorWrapperProvider provider : EXECUTOR_WRAPPER_PROVIDERS) {
             Supplier<String> msg = () -> provider + "(class: " + provider.getClass().getName() + ") return null";
             executor = requireNonNull(provider.wrap(executor), msg);

diff --git a/cffu-core/src/test/java/io/foldright/cffu/CffuFactoryTest.java b/cffu-core/src/test/java/io/foldright/cffu/CffuFactoryTest.java
@@ -19,6 +19,7 @@
 
 import static io.foldright.cffu.CompletableFutureUtils.failedFuture;
 import static io.foldright.cffu.CompletableFutureUtils.toCompletableFutureArray;
+import static io.foldright.cffu.DefaultExecutorTestUtils.unwrapMadeExecutor;
 import static io.foldright.test_utils.TestUtils.*;
 import static java.util.concurrent.CompletableFuture.completedFuture;
 import static java.util.concurrent.ForkJoinPool.commonPool;
@@ -811,7 +812,7 @@ void test_toCffu() throws Exception {
         CffuFactory fac = CffuFactory.builder(anotherExecutorService).forbidObtrudeMethods(true).build();
         Cffu<Integer> cffu = fac.toCffu(cffu_in);
         assertNotSame(cffu_in, cffu);
-        assertSame(anotherExecutorService, cffu.defaultExecutor());
+        assertSame(anotherExecutorService, unwrapMadeExecutor(cffu));
         assertSame(fac, cffu.cffuFactory());
         assertEquals("obtrude methods is forbidden by cffu", assertThrowsExactly(UnsupportedOperationException.class, () ->
                 cffu.obtrudeValue(44)
@@ -913,16 +914,20 @@ void test_cffuListToArray() {
 
     @Test
     void test_getter() {
-        assertSame(executorService, cffuFactory.defaultExecutor());
+        assertSame(executorService, unwrapMadeExecutor(cffuFactory));
         assertFalse(cffuFactory.forbidObtrudeMethods());
 
         CffuFactory fac = CffuFactory.builder(anotherExecutorService).forbidObtrudeMethods(true).build();
-        assertSame(anotherExecutorService, fac.defaultExecutor());
+        assertSame(anotherExecutorService, unwrapMadeExecutor(fac));
         assertTrue(fac.forbidObtrudeMethods());
 
         final CffuFactory fac2 = cffuFactory.resetDefaultExecutor(anotherExecutorService);
-        assertSame(anotherExecutorService, fac2.defaultExecutor());
+        assertSame(anotherExecutorService, unwrapMadeExecutor(fac2));
         assertEquals(cffuFactory.forbidObtrudeMethods(), fac2.forbidObtrudeMethods());
+
+        final CffuFactory fac3 = cffuFactory.resetDefaultExecutor(fac2.defaultExecutor());
+        assertSame(fac2.defaultExecutor(), fac3.defaultExecutor());
+        assertEquals(fac2.forbidObtrudeMethods(), fac3.forbidObtrudeMethods());
     }
 
     @Test
@@ -945,15 +950,16 @@ void test_executorSetting_MayBe_ThreadPerTaskExecutor() throws Exception {
 
         CffuFactory fac = CffuFactory.builder(commonPool()).build();
         if (USE_COMMON_POOL) {
-            assertSame(commonPool(), fac.defaultExecutor());
+            assertSame(commonPool(), unwrapMadeExecutor(fac));
         } else {
-            String executorClassName = fac.defaultExecutor().getClass().getName();
+            String executorClassName = unwrapMadeExecutor(fac).getClass().getName();
             assertTrue(executorClassName.endsWith("$ThreadPerTaskExecutor"));
         }
 
         assertEquals(42, fac.supplyAsync(() -> 42).get());
     }
 
+
     // endregion
     ////////////////////////////////////////////////////////////////////////////////
     // region# Test helper methods/fields

diff --git a/cffu-core/src/test/java/io/foldright/cffu/CffuTest.java b/cffu-core/src/test/java/io/foldright/cffu/CffuTest.java
@@ -18,6 +18,7 @@
 import java.util.function.Consumer;
 import java.util.function.Function;
 
+import static io.foldright.cffu.DefaultExecutorTestUtils.unwrapMadeExecutor;
 import static io.foldright.test_utils.TestUtils.*;
 import static java.util.concurrent.CompletableFuture.completedFuture;
 import static java.util.function.Function.identity;
@@ -486,7 +487,7 @@ void test_resetCffuFactory() {
 
         Executor executor = Runnable::run;
         final Cffu<Integer> f2 = cf.resetDefaultExecutor(executor);
-        assertSame(executor, f2.defaultExecutor());
+        assertSame(executor, unwrapMadeExecutor(f2));
         assertEquals(cffuFactory.forbidObtrudeMethods(), f2.cffuFactory().forbidObtrudeMethods());
     }
 

diff --git a/cffu-core/src/test/java/io/foldright/cffu/DefaultExecutorTestUtils.java b/cffu-core/src/test/java/io/foldright/cffu/DefaultExecutorTestUtils.java
@@ -0,0 +1,16 @@
+package io.foldright.cffu;
+
+import java.util.concurrent.Executor;
+
+
+public class DefaultExecutorTestUtils {
+    public static Executor unwrapMadeExecutor(CffuFactory factory) {
+        final CffuFactoryBuilder.CffuMadeExecutor executor = (CffuFactoryBuilder.CffuMadeExecutor) factory.defaultExecutor();
+        return executor.unwrap();
+    }
+
+    public static Executor unwrapMadeExecutor(Cffu<?> cffu) {
+        final CffuFactoryBuilder.CffuMadeExecutor executor = (CffuFactoryBuilder.CffuMadeExecutor) cffu.defaultExecutor();
+        return executor.unwrap();
+    }
+}
diff --git a/cffu-core/src/test/java/io/foldright/cffu/spi/ExecutorWrapperProviderTest.kt b/cffu-core/src/test/java/io/foldright/cffu/spi/ExecutorWrapperProviderTest.kt
@@ -1,6 +1,7 @@
 package io.foldright.cffu.spi
 
 import io.foldright.cffu.CffuFactory
+import io.foldright.cffu.DefaultExecutorTestUtils.unwrapMadeExecutor
 import io.foldright.test_utils.testThreadPoolExecutor
 import io.kotest.core.spec.style.FunSpec
 import io.kotest.matchers.types.shouldBeSameInstanceAs
@@ -10,15 +11,15 @@ class ExecutorWrapperProviderTest : FunSpec({
     test("disable TestExecutorWrapper") {
         val factory = CffuFactory.builder(testThreadPoolExecutor).build()
         val cffu = factory.runAsync {}
-        cffu.defaultExecutor().shouldBeSameInstanceAs(testThreadPoolExecutor)
+        unwrapMadeExecutor(cffu).shouldBeSameInstanceAs(testThreadPoolExecutor)
     }
 
     test("enable TestExecutorWrapper") {
         enableTestExecutorWrapper()
 
         val factory = CffuFactory.builder(testThreadPoolExecutor).build()
         val cffu = factory.runAsync {}
-        cffu.defaultExecutor().shouldNotBeSameInstanceAs(testThreadPoolExecutor)
+        unwrapMadeExecutor(cffu).shouldNotBeSameInstanceAs(testThreadPoolExecutor)
     }
 
     beforeTest {