Update StatementTest.java

Author: Paultagoras

jdbc-v2/src/test/java/com/clickhouse/jdbc/StatementTest.java

@@ -3,6 +3,9 @@
 import com.clickhouse.client.api.Client;
 import com.clickhouse.client.api.ClientConfigProperties;
 import com.clickhouse.client.api.query.GenericRecord;
+import com.clickhouse.client.api.query.QuerySettings;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.testng.Assert;
 import org.testng.annotations.Test;
 
@@ -18,6 +21,7 @@
 import java.util.Arrays;
 import java.util.List;
 import java.util.Properties;
+import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 
 import static org.testng.Assert.*;
@@ -29,6 +33,8 @@
 
 
 public class StatementTest extends JdbcIntegrationTest {
+    private static final Logger log = LoggerFactory.getLogger(StatementTest.class);
+
     @Test(groups = { "integration" })
     public void testExecuteQuerySimpleNumbers() throws Exception {
         try (Connection conn = getJdbcConnection()) {
@@ -471,4 +477,28 @@ public void testConnectionExhaustion() throws Exception {
             }
         }
     }
+
+    @Test(groups = { "integration" })
+    public void testConcurrentCancel() throws Exception {
+        try (Connection conn = getJdbcConnection()) {
+            try (StatementImpl stmt = (StatementImpl) conn.createStatement()) {
+                stmt.executeQuery("SELECT number FROM system.numbers LIMIT 10000000000");
+                stmt.cancel();
+            }
+            try (StatementImpl stmt = (StatementImpl) conn.createStatement()) {
+                new Thread(() -> {
+                    try {
+                        ResultSet rs = stmt.executeQuery("SELECT number FROM system.numbers LIMIT 10000000000");
+                        rs.next();
+                        log.info(rs.getObject(1).toString());
+                    } catch (SQLException e) {
+                        log.error("Error in thread", e);
+                    }
+                }).start();
+
+                Thread.sleep(1000);
+                stmt.cancel();
+            }
+        }
+    }
 }