AJ-566: fix the connection in connection pool can connect twice

Author: lucaijun

src/com/xxdb/SimpleDBConnectionPool.java

@@ -3,6 +3,8 @@
 import com.xxdb.data.BasicInt;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Objects;
 import java.util.concurrent.CopyOnWriteArrayList;
@@ -95,7 +97,9 @@ protected class SimpleDBConnectionPoolImpl {
             try {
                 for (int i = 0; i < initialPoolSize; i++) {
                     PoolEntry poolEntry = new PoolEntry(useSSL, compress, usePython, String.format("DolphinDBConnection_%d", i + 1));
-                    if (!poolEntry.connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, loadBalance)) {
+                    if (poolEntry.connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, loadBalance)) {
+                        poolEntry.isFirstLogin.set(false);
+                    }else {
                         log.error(String.format("Connection %s connect failure.", poolEntry.connectionName));
                     }
                     poolEntryArrayList.add(poolEntry);
@@ -156,6 +160,8 @@ boolean isClosed() {
 
     class PoolEntry extends DBConnection {
         AtomicBoolean inUse = new AtomicBoolean(false);
+        AtomicBoolean isFirstConnect = new AtomicBoolean(true);
+        AtomicBoolean isFirstLogin = new AtomicBoolean(true);
         String connectionName;
 
         PoolEntry(boolean useSSL, boolean compress, boolean usePython, String connectionName) {
@@ -171,6 +177,27 @@ boolean isIdle() {
             return !this.inUse.get();
         }
 
+        @Override
+        public void setLoadBalance(boolean loadBalance) {
+            throw new RuntimeException("The loadBalance configuration of connection in connection pool can only be set in SimpleDBConnectionPoolConfig.");
+        }
+
+        @Override
+        public boolean connect(String hostName, int port, String userId, String password, String initialScript, boolean enableHighAvailability, String[] highAvailabilitySites, boolean reconnect, boolean enableLoadBalance) throws IOException {
+            if (isFirstConnect.compareAndSet(true, false))
+                return super.connect(hostName, port, userId, password, initialScript, enableHighAvailability, highAvailabilitySites, reconnect, enableLoadBalance);
+            else
+                throw new RuntimeException("The connection in connection pool can only connect by pool.");
+        }
+
+        @Override
+        public void login(String userId, String password, boolean enableEncryption) throws IOException {
+            if (isFirstLogin.get())
+                super.login(userId, password, enableEncryption);
+            else
+                throw new RuntimeException("The connection in connection pool can only login by pool.");
+        }
+
         @Override
         public void close() {
             if (isBusy())