feat: optimize the thread pool size (#83)

* feat: optimize the thread pool size

Author: wardseptember

trpc-core/src/main/java/com/tencent/trpc/core/common/Constants.java

@@ -74,9 +74,13 @@ public class Constants {
      */
     public static final String DEFAULT_BOSS_THREADS = "1";
     /**
-     * Default business thread count cpus * 2
+     * Default business core thread count 50
      */
-    public static final int DEFAULT_BIZ_THREADS = CPUS * 2;
+    public static final int DEFAULT_CORE_THREADS = 50;
+    /**
+     * Default business max thread count 200
+     */
+    public static final int DEFAULT_MAX_THREADS = 200;
     /**
      * Default coroutine pool core count
      */

trpc-core/src/main/java/com/tencent/trpc/core/worker/WorkerPoolManager.java

@@ -63,11 +63,11 @@ public class WorkerPoolManager {
 
     static {
         DEF_PROVIDER_WORKER_POOL_CONFIG = newThreadWorkerPoolConfig(DEF_PROVIDER_WORKER_POOL_NAME,
-                Constants.DEFAULT_BIZ_THREADS, Boolean.FALSE);
+                Constants.DEFAULT_CORE_THREADS, Constants.DEFAULT_MAX_THREADS, Boolean.FALSE);
         DEF_CONSUMER_WORKER_POOL_CONFIG = newThreadWorkerPoolConfig(DEF_CONSUMER_WORKER_POOL_NAME,
-                Constants.DEFAULT_BIZ_THREADS, Boolean.FALSE);
+                Constants.DEFAULT_CORE_THREADS, Constants.DEFAULT_MAX_THREADS, Boolean.FALSE);
         DEF_NAMING_WORKER_POOL_CONFIG = newThreadWorkerPoolConfig(DEF_NAMING_WORKER_POOL_NAME,
-                Constants.DEFAULT_BIZ_THREADS, Boolean.FALSE);
+                Constants.DEFAULT_CORE_THREADS, Constants.DEFAULT_MAX_THREADS, Boolean.FALSE);
         shareScheduler = new ScheduledThreadPoolExecutor(Math.min(Constants.CPUS, 4),
                 // Parameters can be considered for startup parameter configuration
                 new NamedThreadFactory("trpc_share_scheduler"));
@@ -122,6 +122,11 @@ public static PluginConfig newThreadWorkerPoolConfig(String name, int thread, bo
         return ThreadWorkerPool.newThreadWorkerPoolConfig(name, thread, useFiber);
     }
 
+    public static PluginConfig newThreadWorkerPoolConfig(String name, int corePoolSize,
+            int maxPoolSize, boolean useFiber) {
+        return ThreadWorkerPool.newThreadWorkerPoolConfig(name, corePoolSize, maxPoolSize, useFiber);
+    }
+
     /**
      * Close shared thread pool, graceful shutdown.
      */

trpc-core/src/main/java/com/tencent/trpc/core/worker/support/thread/ThreadWorkerPool.java

@@ -74,10 +74,15 @@ public class ThreadWorkerPool extends AbstractWorkerPool
     private UncaughtExceptionHandler uncaughtExceptionHandler;
 
     public static PluginConfig newThreadWorkerPoolConfig(String name, int corePoolSize, boolean useFiber) {
+        return newThreadWorkerPoolConfig(name, corePoolSize, corePoolSize, useFiber);
+    }
+
+    public static PluginConfig newThreadWorkerPoolConfig(String name,
+            int corePoolSize, int maxPoolSize, boolean useFiber) {
         ThreadPoolConfig poolConfig = new ThreadPoolConfig();
         poolConfig.setUseFiber(useFiber);
         poolConfig.setCorePoolSize(corePoolSize);
-        poolConfig.setMaximumPoolSize(corePoolSize);
+        poolConfig.setMaximumPoolSize(maxPoolSize);
         poolConfig.setShareSchedule(Boolean.TRUE);
         return new PluginConfig(name, WorkerPool.class, ThreadWorkerPool.class, poolConfig.toMap());
     }

trpc-core/src/test/java/com/tencent/trpc/core/common/config/BackendConfigTest.java

@@ -239,7 +239,8 @@ public void testIp() {
     public void test() {
         ExtensionLoader
                 .registerPlugin(new PluginConfig("attalog", Filter.class, RemoteLoggerTest.class));
-        ExtensionLoader.registerPlugin(ThreadWorkerPool.newThreadWorkerPoolConfig("thread", 10, Boolean.FALSE));
+        ExtensionLoader.registerPlugin(ThreadWorkerPool.newThreadWorkerPoolConfig("thread", 10,
+                10, Boolean.FALSE));
         BackendConfig config = new BackendConfig();
         config.setCallee("trpc.calleeapp.calleeserver.calleeservice.calleemethod");
         config.setNamingUrl("ip://127.0.0.1:8888");

trpc-core/src/test/java/com/tencent/trpc/core/worker/WorkerPoolManagerTest.java

@@ -100,7 +100,17 @@ public void testRefresh() {
         WorkerPoolManager.refresh(WorkerPoolManager.DEF_NAMING_WORKER_POOL_NAME, pluginConfig);
         WorkerPool workerPool = WorkerPoolManager.get(WorkerPoolManager.DEF_CONSUMER_WORKER_POOL_NAME);
         Assert.assertNotNull(workerPool);
+    }
 
+    @Test
+    public void testWorkPool() {
+        this.testRegistDefaultPluginConfig();
+        PluginConfig pluginConfig = WorkerPoolManager
+                .newThreadWorkerPoolConfig(WorkerPoolManager.DEF_CONSUMER_WORKER_POOL_NAME,
+                        20,20, Boolean.FALSE);
+        WorkerPoolManager.refresh(WorkerPoolManager.DEF_NAMING_WORKER_POOL_NAME, pluginConfig);
+        WorkerPool workerPool = WorkerPoolManager.get(WorkerPoolManager.DEF_CONSUMER_WORKER_POOL_NAME);
+        Assert.assertNotNull(workerPool);
     }
 
     /**

trpc-transport/trpc-transport-http/src/main/java/com/tencent/trpc/transport/http/support/jetty/JettyHttpServer.java

@@ -136,8 +136,8 @@ private QueuedThreadPool getServerThreadPool(ProtocolConfig config) {
         int threads = config.getIoThreads();
         QueuedThreadPool threadPool = new QueuedThreadPool();
         threadPool.setDaemon(true);
-        threadPool.setMaxThreads(Math.max(threads, 4));
-        threadPool.setMinThreads(Math.max(threads, 4));
+        threadPool.setMaxThreads(Math.max(threads, Constants.DEFAULT_MAX_THREADS));
+        threadPool.setMinThreads(Math.max(threads, Constants.DEFAULT_CORE_THREADS));
         threadPool.setIdleTimeout(Math.max(config.getIdleTimeout(), Integer.parseInt(Constants.DEFAULT_IDLE_TIMEOUT)));
         return threadPool;
     }