Add customization hook for EventLoopGroup creation #1273

redis · May 5, 2020 · 83d45aa · 83d45aa
1 parent 53c6268
commit 83d45aa
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/main/java/io/lettuce/core/resource/DefaultEventLoopGroupProvider.java b/src/main/java/io/lettuce/core/resource/DefaultEventLoopGroupProvider.java
@@ -130,12 +130,27 @@ private <T extends EventLoopGroup> T getOrCreate(Class<T> type) {
         }
 
         if (!eventLoopGroups.containsKey(type)) {
-            eventLoopGroups.put(type, createEventLoopGroup(type, numberOfThreads, threadFactoryProvider));
+            eventLoopGroups.put(type, doCreateEventLoopGroup(type, numberOfThreads, threadFactoryProvider));
         }
 
         return (T) eventLoopGroups.get(type);
     }
 
+    /**
+     * Customization hook for {@link EventLoopGroup} creation.
+     * 
+     * @param <T>
+     * @param type requested event loop group type.
+     * @param numberOfThreads number of threads to create.
+     * @param threadFactoryProvider provider for {@link ThreadFactory}.
+     * @return
+     * @since 6.0
+     */
+    protected <T extends EventLoopGroup> EventExecutorGroup doCreateEventLoopGroup(Class<T> type, int numberOfThreads,
+            ThreadFactoryProvider threadFactoryProvider) {
+        return createEventLoopGroup(type, numberOfThreads, threadFactoryProvider);
+    }
+
     /**
      * Create an instance of a {@link EventExecutorGroup} using the default {@link ThreadFactoryProvider}. Supported types are:
      * <ul>