Set default connection pool size of RestClient to 80. (sofastack#129)

example/src/test/java/com/alipay/sofa/rpc/rest/start/RestClientMultipleMain.java

@@ -0,0 +1,100 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alipay.sofa.rpc.rest.start;
+
+import com.alipay.sofa.rpc.bolt.start.BoltClientMultipleMain;
+import com.alipay.sofa.rpc.common.struct.NamedThreadFactory;
+import com.alipay.sofa.rpc.config.ApplicationConfig;
+import com.alipay.sofa.rpc.config.ConsumerConfig;
+import com.alipay.sofa.rpc.context.RpcRuntimeContext;
+import com.alipay.sofa.rpc.log.Logger;
+import com.alipay.sofa.rpc.log.LoggerFactory;
+import com.alipay.sofa.rpc.rest.RestService;
+
+import java.util.concurrent.SynchronousQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicLong;
+
+/**
+ *
+ *
+ * @author <a href=ailto:[email protected]>GengZhang</a>
+ */
+public class RestClientMultipleMain {
+
+    /**
+     * slf4j Logger for this class
+     */
+    private final static Logger LOGGER = LoggerFactory.getLogger(BoltClientMultipleMain.class);
+
+    public static void main(String[] args) {
+
+        ApplicationConfig application = new ApplicationConfig().setAppName("test-client");
+
+        ConsumerConfig<RestService> consumerConfig = new ConsumerConfig<RestService>()
+            .setApplication(application)
+            .setInterfaceId(RestService.class.getName())
+            .setProtocol("rest")
+            .setBootstrap("rest")
+            .setDirectUrl("rest://127.0.0.1:8888")
+            .setTimeout(3000);
+        final RestService helloService = consumerConfig.refer();
+
+        LOGGER.warn("started at pid {}", RpcRuntimeContext.PID);
+
+        final int threads = 50;
+        final AtomicLong cnt = new AtomicLong(0);
+        final ThreadPoolExecutor service1 = new ThreadPoolExecutor(threads, threads, 0L, TimeUnit.MILLISECONDS,
+            new SynchronousQueue<Runnable>(), new NamedThreadFactory("client-"));// 无队列
+        for (int i = 0; i < threads; i++) {
+            service1.execute(new Runnable() {
+                @Override
+                public void run() {
+                    while (true) {
+                        try {
+                            String s = helloService.get("1234567890");
+                            cnt.incrementAndGet();
+                        } catch (Exception e) {
+                            LOGGER.error("", e);
+                        }
+                    }
+                }
+            });
+        }
+
+        Thread thread = new Thread(new Runnable() {
+            private long last = 0;
+
+            @Override
+            public void run() {
+                while (true) {
+                    long count = cnt.get();
+                    long tps = count - last;
+                    LOGGER.error("last 1s invoke: {}, queue: {}", tps, service1.getQueue().size());
+                    last = count;
+
+                    try {
+                        Thread.sleep(1000);
+                    } catch (InterruptedException e) {
+                    }
+                }
+            }
+        }, "Print-tps-THREAD");
+        thread.start();
+    }
+}

extension-impl/remoting-resteasy/src/main/java/com/alipay/sofa/rpc/transport/rest/RestClientTransport.java

@@ -40,6 +40,12 @@
  */
 @Extension("rest")
 public class RestClientTransport extends AbstractProxyClientTransport {
+
+    /**
+     * 默认至少的连接池大小
+     */
+    private static final int MIN_CONNECTION_POOL_SIZE = 80;
+
     public RestClientTransport(ClientTransportConfig transportConfig) {
         super(transportConfig);
     }
@@ -52,7 +58,7 @@ protected Object buildProxy(ClientTransportConfig transportConfig) throws SofaRp
             .registerProvider().logProviders()
             .establishConnectionTimeout(transportConfig.getConnectTimeout(), TimeUnit.MILLISECONDS)
             .socketTimeout(transportConfig.getInvokeTimeout(), TimeUnit.MILLISECONDS)
-            .connectionPoolSize(transportConfig.getConnectionNum())// 连接池？
+            .connectionPoolSize(Math.max(transportConfig.getConnectionNum(), MIN_CONNECTION_POOL_SIZE))
             .build();
 
         ProviderInfo provider = transportConfig.getProviderInfo();