new_init_method

americanexpress · Jul 20, 2023 · 0b9b42a · 0b9b42a
1 parent 9120715
commit 0b9b42a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 7 deletions.
diff --git a/pom.xml b/pom.xml
@@ -19,7 +19,7 @@
 
     <groupId>com.americanexpress.unify.flowret</groupId>
     <artifactId>unify-flowret</artifactId>
-    <version>1.4.0</version>
+    <version>1.4.1</version>
     <packaging>jar</packaging>
 
     <name>unify-flowret</name>

diff --git a/readme.md b/readme.md
@@ -20,7 +20,7 @@ Flowret is available as a jar file in Maven central with the following Maven coo
 ````pom
 <groupId>com.americanexpress.unify.flowret</groupId>
 <artifactId>unify-flowret</artifactId>
-<version>1.4.0</version>
+<version>1.4.1</version>
 ````
 
 ---

diff --git a/src/main/java/com/americanexpress/unify/flowret/Flowret.java b/src/main/java/com/americanexpress/unify/flowret/Flowret.java
@@ -21,10 +21,7 @@
 import com.americanexpress.unify.flowret.CONSTS_FLOWRET.DAO;
 import com.americanexpress.unify.jdocs.JDocument;
 
-import java.util.concurrent.ArrayBlockingQueue;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.ThreadPoolExecutor;
-import java.util.concurrent.TimeUnit;
+import java.util.concurrent.*;
 
 /*
  * @author Deepak Arora
@@ -86,11 +83,22 @@ public static void init(int maxThreads, int idleTimeout, String typeIdSep) {
   }
 
   public static void init(int maxThreads, int idleTimeout, String typeIdSep, String errorWorkbasket) {
+    init(maxThreads, idleTimeout, typeIdSep, errorWorkbasket, null);
+  }
+
+  public static void init(int maxThreads, int idleTimeout, String typeIdSep, String errorWorkbasket, ThreadFactory threadFactory) {
     Flowret am = instance();
     am.maxThreads = maxThreads;
     am.idleTimeout = idleTimeout;
     BlockOnOfferQueue<Runnable> q = new BlockOnOfferQueue(new ArrayBlockingQueue<>(am.maxThreads * 2));
-    am.es = new ThreadPoolExecutor(am.maxThreads, am.maxThreads, am.idleTimeout, TimeUnit.MILLISECONDS, q, new RejectedItemHandler());
+
+    if (threadFactory == null) {
+      am.es = new ThreadPoolExecutor(am.maxThreads, am.maxThreads, am.idleTimeout, TimeUnit.MILLISECONDS, q, new RejectedItemHandler());
+    }
+    else {
+      am.es = new ThreadPoolExecutor(am.maxThreads, am.maxThreads, am.idleTimeout, TimeUnit.MILLISECONDS, q, threadFactory, new RejectedItemHandler());
+    }
+
     DAO.SEP = typeIdSep;
     am.errorWorkbasket = errorWorkbasket;
     ERRORS_FLOWRET.load();