Merge pull request apache#340 from olvesh/seeds_to_many_ips

olim7t · olim7t · commit 0bef855ba2b5 · 2015-06-17T16:56:42.000+02:00
New addContactPoints method for resolving multiple InetAddress from one hostname
diff --git a/driver-core/src/main/java/com/datastax/driver/core/Cluster.java b/driver-core/src/main/java/com/datastax/driver/core/Cluster.java
@@ -770,6 +770,38 @@ public Builder addContactPoints(String... addresses) {
             return this;
         }
 
+        /**
+         * Adds a contact point - or many if it host resolves to multiple <code>InetAddress</code>s (A records).
+         * <p>
+         *
+         * If the host name points to a dns records with multiple a-records, all InetAddresses
+         * returned will be used. Make sure that all resulting <code>InetAddress</code>s returned
+         * points to the same cluster and datacenter.
+         * <p>
+         * See {@link Builder#addContactPoint} for more details on contact
+         * points and thrown exceptions
+         *
+         * @param address address of the nodes to look up InetAddresses from to add as contact points.
+         * @return this Builder.
+         *
+         *
+         * @see Builder#addContactPoint
+         */
+        public Builder addContactPoints(String address) {
+            // We explicitely check for nulls because InetAdress.getByName() will happily
+            // accept it and use localhost (while a null here almost likely mean a user error,
+            // not "connect to localhost")
+            if (address == null)
+                throw new NullPointerException();
+
+            try {
+                addContactPoints(InetAddress.getAllByName(address));
+            } catch (UnknownHostException e) {
+                throw new IllegalArgumentException(e.getMessage());
+            }
+            return this;
+        }
+
         /**
          * Adds contact points.
          * <p>
