sdk初始化增加dispatcher配置

Author: FuDongHai

dependency-reduced-pom.xml

@@ -4,7 +4,7 @@
   <groupId>com.easemob.im</groupId>
   <artifactId>im-sdk-core</artifactId>
   <name>im-sdk-core</name>
-  <version>1.0.14.1-SNAPSHOT</version>
+  <version>1.0.15-SNAPSHOT</version>
   <description>OpenAPI Java</description>
   <url>https://github.com/openapitools/openapi-generator</url>
   <developers>

src/main/java/com/easemob/im/ApiClient.java

@@ -115,7 +115,7 @@ public ApiClient() {
     }
 
     public ApiClient(String basePath, Realm realm, String appKey, String clientId, String clientSecret, String appId, String appCert, int maxIdleConnections, int connectKeepAliveMilliSeconds,
-            int connectTimeoutMilliSeconds, EMProxy proxy)
+            int connectTimeoutMilliSeconds, EMProxy proxy, int dispatcherMaxRequests, int dispatcherMaxRequestsPerHost)
             throws ApiException {
         this.basePath = basePath;
         this.realm = realm;
@@ -134,7 +134,8 @@ public ApiClient(String basePath, Realm realm, String appKey, String clientId, S
         );
 
         initHttpClient(Collections.<Interceptor>emptyList(), maxIdleConnections,
-                connectKeepAliveMilliSeconds, connectTimeoutMilliSeconds, proxy);
+                connectKeepAliveMilliSeconds, connectTimeoutMilliSeconds, proxy,
+                dispatcherMaxRequests, dispatcherMaxRequestsPerHost);
         init();
 
         // Setup authentications (key: authentication name, value: authentication).
@@ -159,7 +160,8 @@ public ApiClient(OkHttpClient client) throws ApiException {
     }
 
     private void initHttpClient(List<Interceptor> interceptors, int maxIdleConnections,
-            int connectKeepAliveMilliSeconds, int connectTimeoutMilliSeconds, EMProxy emProxy) {
+            int connectKeepAliveMilliSeconds, int connectTimeoutMilliSeconds, EMProxy emProxy,
+            int dispatcherMaxRequests, int dispatcherMaxRequestsPerHost) {
         OkHttpClient.Builder builder = new OkHttpClient.Builder();
         builder.addNetworkInterceptor(getProgressInterceptor());
         for (Interceptor interceptor: interceptors) {
@@ -192,6 +194,11 @@ private void initHttpClient(List<Interceptor> interceptors, int maxIdleConnectio
             }
         }
 
+        Dispatcher customDispatcher = new Dispatcher();
+        customDispatcher.setMaxRequests(dispatcherMaxRequests);
+        customDispatcher.setMaxRequestsPerHost(dispatcherMaxRequestsPerHost);
+
+        builder.dispatcher(customDispatcher);
         httpClient = builder.build();
     }
 
@@ -353,6 +360,10 @@ public static class Builder {
 
         private int connectTimeoutMilliSeconds = 10000;
 
+        private int dispatcherMaxRequests = 200;
+
+        private int dispatcherMaxRequestsPerHost = 50;
+
         private EMProxy proxy;
 
         public Builder setRealm(Realm realm) {
@@ -442,6 +453,22 @@ public Builder setConnectTimeoutMilliSeconds(int connectTimeoutMilliSeconds) {
             return this;
         }
 
+        public Builder setDispatcherMaxRequests(int dispatcherMaxRequests) {
+            if (dispatcherMaxRequests < 0) {
+                throw new IllegalArgumentException("dispatcherMaxRequests must not be negative");
+            }
+            this.dispatcherMaxRequests = dispatcherMaxRequests;
+            return this;
+        }
+
+        public Builder setDispatcherMaxRequestsPerHost(int dispatcherMaxRequestsPerHost) {
+            if (dispatcherMaxRequestsPerHost < 0) {
+                throw new IllegalArgumentException("dispatcherMaxRequestsPerHost must not be negative");
+            }
+            this.dispatcherMaxRequestsPerHost = dispatcherMaxRequestsPerHost;
+            return this;
+        }
+
         public Builder setProxy(EMProxy proxy) {
             this.proxy = proxy;
             return this;
@@ -480,7 +507,7 @@ public ApiClient build() throws ApiException {
                 return new ApiClient(this.basePath, this.realm, this.appKey, this.clientId,
                         this.clientSecret, null, null, this.maxIdleConnections,
                         this.connectKeepAliveMilliSeconds, this.connectTimeoutMilliSeconds,
-                        this.proxy);
+                        this.proxy, this.dispatcherMaxRequests, this.dispatcherMaxRequestsPerHost);
             } else if (this.realm.equals(Realm.AGORA_REALM)) {
                 if (this.appId == null) {
                     throw new IllegalArgumentException("appId not set");
@@ -494,7 +521,7 @@ public ApiClient build() throws ApiException {
 
                 return new ApiClient(this.basePath, this.realm, this.appKey, null, null, this.appId,
                         this.appCert, this.maxIdleConnections, this.connectKeepAliveMilliSeconds,
-                        this.connectTimeoutMilliSeconds, this.proxy);
+                        this.connectTimeoutMilliSeconds, this.proxy, this.dispatcherMaxRequests, this.dispatcherMaxRequestsPerHost);
             } else {
                 throw new IllegalArgumentException(
                         String.format("invalid realm type %s", this.realm));

src/test/java/com/easemob/im/api/AbstractTest.java

@@ -15,6 +15,8 @@ public abstract class AbstractTest {
                     .setAppKey(System.getenv("IM_APPKEY"))
                     .setClientId(System.getenv("IM_CLIENT_ID"))
                     .setClientSecret(System.getenv("IM_CLIENT_SECRET"))
+                            .setDispatcherMaxRequests(300)
+                            .setDispatcherMaxRequestsPerHost(60)
                     .build());
         } catch (ApiException e) {
             throw new RuntimeException(e);