添加對countdown的捕獲

Author: dapeng

rdb/rdb-side/src/main/java/com/dtstack/flink/sql/side/rdb/async/RdbAsyncReqRow.java

@@ -119,19 +119,18 @@ private void connectWithRetry(Map<String, Object> inputParams, CRow input, Resul
         while(!finishFlag.get()){
             CountDownLatch latch = new CountDownLatch(1);
             rdbSqlClient.getConnection(conn -> {
-                if(conn.failed()){
-                    if(failCounter.getAndIncrement() % 1000 == 0){
-                        logger.error("getConnection error", conn.cause());
-                    }
-                    if(failCounter.get() >= sideInfo.getSideTableInfo().getAsyncFailMaxNum(3)){
-                        resultFuture.completeExceptionally(conn.cause());
-                        finishFlag.set(true);
-                    }
-                    latch.countDown();
-                    conn.result().close();
-                    return;
-                }
                 try {
+                    if(conn.failed()){
+                        if(failCounter.getAndIncrement() % 1000 == 0){
+                            logger.error("getConnection error", conn.cause());
+                        }
+                        if(failCounter.get() >= sideInfo.getSideTableInfo().getAsyncFailMaxNum(3)){
+                            resultFuture.completeExceptionally(conn.cause());
+                            finishFlag.set(true);
+                        }
+                        conn.result().close();
+                        return;
+                    }
                     CONN_STATUS.set(true);
                     ScheduledFuture<?> timerFuture = registerTimer(input, resultFuture);
                     cancelTimerWhenComplete(resultFuture, timerFuture);