feature:add active test flag

2026-09-17 17:29:00 +00:00 · 2019-12-31 15:28:43 +08:00
parent 70599a1656
commit f742f3c6aa
3 changed files with 38 additions and 1 deletions
@@ -57,6 +57,16 @@ public class ConnectionManager {
                        closeConnection(connection);
                        continue;
                    }
+                    if (connection.isNeedActiveTest()) {
+                        boolean isActive = false;
+                        try {
+                            isActive = connection.activeTest();
+                        } catch (IOException e) {
+                            System.err.println("send to server[" + inetSocketAddress.getAddress().getHostAddress() + ":" + inetSocketAddress.getPort() + "] active test error ,emsg:" + e.getMessage());
+                            isActive = false;
+                        }
+                        if (!isActive) continue;
+                    }
                } else if (ClientGlobal.g_connection_pool_max_count_per_entry == 0 || totalCount.get() < ClientGlobal.g_connection_pool_max_count_per_entry) {
                    connection = ConnectionFactory.create(this.inetSocketAddress);
                    totalCount.incrementAndGet();
@@ -107,6 +117,20 @@ public class ConnectionManager {
        }
    }

+    public void setActiveTestFlag() {
+        if (freeCount.get() > 0) {
+            lock.lock();
+            try {
+                for (Connection freeConnection : freeConnections) {
+                    freeConnection.setNeedActiveTest(true);
+                }
+            } finally {
+                lock.unlock();
+            }
+        }
+    }
+
+
    @Override
    public String toString() {
        return "ConnectionManager{" +