Add PoolingOptions method to set both core and max connections (JAVA-662)

olim7t · olim7t · commit a2cd65955034 · 2015-06-22T17:26:21.000+02:00
diff --git a/changelog/README.md b/changelog/README.md
@@ -15,6 +15,8 @@
 - [improvement] Create values() function for Insert builder using List (JAVA-225)
 - [improvement] Warn when ReplicationStrategy encounters invalid
   replication factors (JAVA-702)
+- [improvement] Add PoolingOptions method to set both core and max
+  connections (JAVA-662).
 
 Merged from 2.0.10_fixes branch:
 
diff --git a/driver-core/src/main/java/com/datastax/driver/core/PoolingOptions.java b/driver-core/src/main/java/com/datastax/driver/core/PoolingOptions.java
@@ -230,6 +230,30 @@ public synchronized PoolingOptions setMaxConnectionsPerHost(HostDistance distanc
         return this;
     }
 
+    /**
+     * Sets the core and maximum number of connections per host in one call.
+     * <p>
+     * This is a convenience method that is equivalent to calling {@link #setCoreConnectionsPerHost(HostDistance, int)}
+     * and {@link #setMaxConnectionsPerHost(HostDistance, int)}.
+     *
+     * @param distance the {@code HostDistance} for which to set these threshold.
+     * @param core the core number of connections.
+     * @param max the max number of connections.
+     * @return this {@code PoolingOptions}.
+     *
+     * @throws IllegalArgumentException if {@code distance == HostDistance.IGNORED},
+     * or if {@code core} > {@code max}.
+     */
+    public synchronized PoolingOptions setConnectionsPerHost(HostDistance distance, int core, int max) {
+        if (distance == HostDistance.IGNORED)
+            throw new IllegalArgumentException("Cannot set connections per host for " + distance + " hosts");
+
+        checkConnectionsPerHostOrder(core, max, distance);
+        coreConnections[distance.ordinal()] = core;
+        maxConnections[distance.ordinal()] = max;
+        return this;
+    }
+
     /**
      * Returns the timeout before an idle connection is removed.
      *
