Set maxQps on OKHttp dispatcher to not be limited to the default conf…

…iguration
googlemaps · Apr 10, 2017 · fe422e9 · fe422e9
1 parent a8e19cc
commit fe422e9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/com/google/maps/OkHttpRequestHandler.java b/src/main/java/com/google/maps/OkHttpRequestHandler.java
@@ -42,10 +42,12 @@ public class OkHttpRequestHandler implements GeoApiContext.RequestHandler {
   private static final MediaType JSON = MediaType.parse("application/json; charset=utf-8");
   private final OkHttpClient client = new OkHttpClient();
   private final RateLimitExecutorService rateLimitExecutorService;
+  private final Dispatcher dispatcher;
 
   public OkHttpRequestHandler() {
     rateLimitExecutorService = new RateLimitExecutorService();
-    client.setDispatcher(new Dispatcher(rateLimitExecutorService));
+    dispatcher = new Dispatcher(rateLimitExecutorService);
+    client.setDispatcher(dispatcher);
   }
 
   @Override
@@ -95,11 +97,15 @@ public void setWriteTimeout(long timeout, TimeUnit unit) {
 
   @Override
   public void setQueriesPerSecond(int maxQps) {
+    dispatcher.setMaxRequests(maxQps);
+    dispatcher.setMaxRequestsPerHost(maxQps);
     rateLimitExecutorService.setQueriesPerSecond(maxQps);
   }
 
   @Override
   public void setQueriesPerSecond(int maxQps, int minimumInterval) {
+    dispatcher.setMaxRequests(maxQps);
+    dispatcher.setMaxRequestsPerHost(maxQps);
     rateLimitExecutorService.setQueriesPerSecond(maxQps, minimumInterval);
   }