From 005ff4f7dcfd820e5fb5e3ca7a0d6072d23c3e8a Mon Sep 17 00:00:00 2001
From: zyyang <zyyang@taosdata.com>
Date: Sat, 20 Feb 2021 11:50:44 +0800
Subject: [PATCH] change

---
 .../src/main/java/com/taosdata/jdbc/AbstractConnection.java | 6 +++++-
 .../src/test/java/com/taosdata/jdbc/TSDBConnectionTest.java | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/connector/jdbc/src/main/java/com/taosdata/jdbc/AbstractConnection.java b/src/connector/jdbc/src/main/java/com/taosdata/jdbc/AbstractConnection.java
index 9d20ab1b37..ea26b6ddcd 100644
--- a/src/connector/jdbc/src/main/java/com/taosdata/jdbc/AbstractConnection.java
+++ b/src/connector/jdbc/src/main/java/com/taosdata/jdbc/AbstractConnection.java
@@ -311,13 +311,17 @@ public abstract class AbstractConnection extends WrapperImpl implements Connecti
 
         boolean status = false;
         try {
-            status = future.get(timeout, TimeUnit.MILLISECONDS);
+            if (timeout == 0)
+                status = future.get();
+            else
+                status = future.get(timeout, TimeUnit.SECONDS);
         } catch (InterruptedException e) {
             e.printStackTrace();
         } catch (ExecutionException e) {
             e.printStackTrace();
         } catch (TimeoutException e) {
             future.cancel(true);
+            status = false;
         } finally {
             executor.shutdownNow();
         }
diff --git a/src/connector/jdbc/src/test/java/com/taosdata/jdbc/TSDBConnectionTest.java b/src/connector/jdbc/src/test/java/com/taosdata/jdbc/TSDBConnectionTest.java
index 6c6f90c60d..fa4433d654 100644
--- a/src/connector/jdbc/src/test/java/com/taosdata/jdbc/TSDBConnectionTest.java
+++ b/src/connector/jdbc/src/test/java/com/taosdata/jdbc/TSDBConnectionTest.java
@@ -194,7 +194,7 @@ public class TSDBConnectionTest {
 
     @Test
     public void isValid() throws SQLException {
-        Assert.assertTrue(conn.isValid(1000));
+        Assert.assertTrue(conn.isValid(5));
         conn.isValid(0);
     }