AJ-729: fix issue about DBConnection;

Author: chengyitian

src/com/xxdb/DBConnection.java

@@ -725,25 +725,25 @@ public boolean connect(String hostName, int port, String userId, String password
             return connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, false);
     }
 
-    public boolean connect(String hostName, int port, String userId, String password, String initialScript, boolean enableHighAvailability, String[] highAvailabilitySites, boolean reconnect, boolean enableLoadBalance, int tryReconnectNums) throws IOException {
-        if (tryReconnectNums <= 0) {
-            this.tryReconnectNums = -1;
-            log.warn("If the param 'tryReconnectNums' less than or equal to 0, when reconnect will be unlimited attempts.");
-        } else {
-            this.tryReconnectNums = tryReconnectNums;
-        }
-
-        return connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, enableLoadBalance);
+    public boolean connect(String hostName, int port, String userId, String password, String initialScript, boolean enableHighAvailability, String[] highAvailabilitySites, boolean reconnect, boolean enableLoadBalance) throws IOException {
+        return connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, enableLoadBalance, -1);
     }
 
-    public boolean connect(String hostName, int port, String userId, String password, String initialScript, boolean enableHighAvailability, String[] highAvailabilitySites, boolean reconnect, boolean enableLoadBalance) throws IOException {
+    public boolean connect(String hostName, int port, String userId, String password, String initialScript, boolean enableHighAvailability, String[] highAvailabilitySites, boolean reconnect, boolean enableLoadBalance, int tryReconnectNums) throws IOException {
         mutex_.lock();
         try {
             this.uid_ = userId;
             this.pwd_ = password;
             this.initialScript_ = initialScript;
             this.enableHighAvailability_ = enableHighAvailability;
             this.loadBalance_ = enableLoadBalance;
+            if (tryReconnectNums <= 0) {
+                this.tryReconnectNums = -1;
+                log.warn("If the param 'tryReconnectNums' less than or equal to 0, when reconnect will be unlimited attempts.");
+            } else {
+                this.tryReconnectNums = tryReconnectNums;
+            }
+
             if (this.loadBalance_ && !this.enableHighAvailability_)
                 throw new RuntimeException("Cannot only enable loadbalance but not enable highAvailablity.");
 
@@ -923,7 +923,7 @@ public void switchDataNode(Node node) throws IOException{
             }
         } while (!closed_ && (tryReconnectNums == -1 || attempt < tryReconnectNums));
 
-        if (!closed_)
+        if (!closed_ && tryReconnectNums > 0)
             throw new RuntimeException("Connect to " + node.hostName + ":" + node.port + " failed after " + attempt + " reconnect attemps.");
 
         if (initialScript_!=null && initialScript_.length() > 0){