Pipe: Fix unnecessary client shutdown logic and reduce handshake waiting time (apache#16341)

* Pipe: Fix unnecessary client shutdown logic and reduce handshake waiting time

* spotless

* update

Author: luoluoyuyu

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/pipe/sink/client/IoTDBDataNodeAsyncClientManager.java

@@ -252,6 +252,9 @@ public void onComplete(final TPipeTransferResp response) {
             }
 
             isHandshakeFinished.set(true);
+            synchronized (isHandshakeFinished) {
+              isHandshakeFinished.notifyAll();
+            }
           }
 
           @Override
@@ -265,6 +268,9 @@ public void onError(final Exception e) {
             exception.set(e);
 
             isHandshakeFinished.set(true);
+            synchronized (isHandshakeFinished) {
+              isHandshakeFinished.notifyAll();
+            }
           }
         };
 
@@ -350,14 +356,13 @@ public void onError(final Exception e) {
 
   private void waitHandshakeFinished(final AtomicBoolean isHandshakeFinished) {
     try {
-      final long startTime = System.currentTimeMillis();
       while (!isHandshakeFinished.get()) {
-        if (isClosed
-            || System.currentTimeMillis() - startTime
-                > PipeConfig.getInstance().getPipeConnectorHandshakeTimeoutMs() * 2L) {
+        if (isClosed) {
           throw new PipeConnectionException("Timed out when waiting for client handshake finish.");
         }
-        Thread.sleep(10);
+        synchronized (isHandshakeFinished) {
+          isHandshakeFinished.wait(1);
+        }
       }
     } catch (final InterruptedException e) {
       Thread.currentThread().interrupt();