Merge pull request #538 from tapdata/TAP-6786

Tap 6786

Author: jarad0628

connectors/doris-connector/src/main/java/io/tapdata/connector/doris/DorisTest.java

@@ -19,7 +19,9 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Consumer;
 
 import static io.tapdata.base.ConnectorBase.testItem;
@@ -64,19 +66,31 @@ protected Boolean testVersion() {
     protected Boolean testWritePrivilege() {
         try {
             AtomicInteger beCount = new AtomicInteger(0);
+            AtomicReference<String> invalidBeHost = new AtomicReference<>();
+            AtomicReference<Integer> invalidHttpPort = new AtomicReference<>();
+            AtomicReference<IOException> throwable = new AtomicReference<>();
             try {
                 jdbcContext.normalQuery("show backends", resultSet -> {
+                    boolean hasValidNode = false;
                     while (resultSet.next()) {
                         beCount.incrementAndGet();
                         String beHost = (resultSet.getString("Host"));
                         Integer httpPort = (resultSet.getInt("HttpPort"));
                         if (null == beHost || null == httpPort) continue;
                         try {
-                            NetUtil.validateHostPortWithSocket(beHost, httpPort);
+                            if (!hasValidNode) {
+                                NetUtil.validateHostPortWithSocket(beHost, httpPort);
+                                hasValidNode = true;
+                            }
                         } catch (IOException e) {
-                            throw new TapTestHostPortEx(e, beHost, String.valueOf(httpPort));
+                            invalidBeHost.set(beHost);
+                            invalidHttpPort.set(httpPort);
+                            throwable.set(e);
                         }
                     }
+                    if (!hasValidNode && null != throwable.get()) {
+                        throw new TapTestHostPortEx(throwable.get(), invalidBeHost.get(), String.valueOf(invalidHttpPort.get()));
+                    }
                 });
             } catch (SQLSyntaxErrorException e) {
                 if ("42000".equals(e.getSQLState())) {