From df853a9649476fcf4d4aa6ecf19d165726566138 Mon Sep 17 00:00:00 2001
From: zyyang <zyyang@taosdata.com>
Date: Thu, 4 Feb 2021 09:37:58 +0800
Subject: [PATCH] change

---
 .../cases/InvalidResultSetPointerTest.java    | 158 ++++++++++++++++++
 1 file changed, 158 insertions(+)
 create mode 100644 src/connector/jdbc/src/test/java/com/taosdata/jdbc/cases/InvalidResultSetPointerTest.java

diff --git a/src/connector/jdbc/src/test/java/com/taosdata/jdbc/cases/InvalidResultSetPointerTest.java b/src/connector/jdbc/src/test/java/com/taosdata/jdbc/cases/InvalidResultSetPointerTest.java
new file mode 100644
index 0000000000..569d45c52a
--- /dev/null
+++ b/src/connector/jdbc/src/test/java/com/taosdata/jdbc/cases/InvalidResultSetPointerTest.java
@@ -0,0 +1,158 @@
+package com.taosdata.jdbc.cases;
+
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.sql.*;
+import java.util.Properties;
+
+public class InvalidResultSetPointerTest {
+
+    private static String host = "127.0.0.1";
+    private static String driverType = "jni";
+    private static final String dbName = "test";
+    private static final String stbName = "weather";
+    private static final String tbName = "subweather";
+    private static Connection connection;
+    private static int numOfSTb = 300000;
+    private static int numOfTb = 3;
+    private static int numOfThreads = 1;
+
+    @Test
+    public void test() throws SQLException {
+        execute("use " + dbName);
+
+        int a = numOfSTb / numOfThreads;
+        if (a < 1) {
+            numOfThreads = numOfSTb;
+            a = 1;
+        }
+
+        int b = 0;
+        if (numOfThreads != 0) {
+            b = numOfSTb % numOfThreads;
+        }
+
+        multiThreadingClass instance[] = new multiThreadingClass[numOfThreads];
+
+        int last = 0;
+        for (int i = 0; i < numOfThreads; i++) {
+            instance[i] = new multiThreadingClass();
+            instance[i].id = i;
+            instance[i].from = last;
+            if (i < b) {
+                instance[i].to = last + a;
+            } else {
+                instance[i].to = last + a - 1;
+            }
+
+            last = instance[i].to + 1;
+            instance[i].numOfTb = numOfTb;
+            instance[i].connection = connection;
+            instance[i].dbName = dbName;
+            instance[i].tbName = tbName;
+            instance[i].start();
+        }
+
+        for (int i = 0; i < numOfThreads; i ++) {
+            try {
+                instance[i].join();
+            } catch (InterruptedException ie) {
+                ie.printStackTrace();
+            }
+        }
+
+        if (connection != null) {
+            this.connection.close();
+            System.out.println("connection closed.");
+        }
+    }
+
+    @BeforeClass
+    public static void beforeClass() {
+        try {
+            String url = "jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata";
+            if (driverType.equals("restful")) {
+                Class.forName("com.taosdata.jdbc.rs.RestfulDriver");
+                url = "jdbc:TAOS-RS://" + host + ":6041/?user=root&password=taosdata";
+            } else {
+                Class.forName("com.taosdata.jdbc.TSDBDriver");
+            }
+            Properties properties = new Properties();
+            properties.setProperty("charset", "UTF-8");
+            properties.setProperty("locale", "en_US.UTF-8");
+            properties.setProperty("timezone", "UTC-8");
+            System.out.println("get connection starting...");
+            connection = DriverManager.getConnection(url, properties);
+            if (connection != null)
+                System.out.println("[ OK ] Connection established.");
+        } catch (ClassNotFoundException | SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private void execute(String sql) {
+        try (Statement statement = connection.createStatement()) {
+            long start = System.currentTimeMillis();
+            boolean execute = statement.execute(sql);
+            long end = System.currentTimeMillis();
+            printSql(sql, execute, (end - start));
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private void printSql(String sql, boolean succeed, long cost) {
+        System.out.println("[ " + (succeed ? "OK" : "ERROR!") + " ] time cost: " + cost + " ms, execute statement ====> " + sql);
+    }
+
+    private void executeQuery(String sql) {
+        try (Statement statement = connection.createStatement()) {
+            long start = System.currentTimeMillis();
+            ResultSet resultSet = statement.executeQuery(sql);
+            long end = System.currentTimeMillis();
+            printSql(sql, true, (end - start));
+//            printResult(resultSet);
+            resultSet.close();
+            statement.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    class multiThreadingClass extends Thread {
+        public int id;
+        public int from, to;
+        public int numOfTb;
+        public Connection connection;
+        public String dbName, stbName, tbName;
+
+        public void run() {
+            System.out.println("ID: " + id + " from: " + from + " to: " + to);
+            try {
+                Thread.sleep(1000);
+            } catch (InterruptedException e) {
+                System.out.println("Thread " + id + " interrupted.");
+            }
+
+            for (int i = from; i < to; i++) {
+                for (int j = 0; j < numOfTb; j++) {
+                    if (j % 1000 == 0) {
+                        try {
+                            System.out.print(id + "s.");
+                            Thread.sleep(1);
+                        } catch (InterruptedException e) {
+                            System.out.println("Thread " + id + " interrupted.");
+                        }
+                    }
+                    final String sql = "select last_row(humidity) from " + dbName + "." + tbName + i + "_" + j;
+//              System.out.println(sql);
+                    executeQuery(sql);
+                }
+            }
+        }
+
+
+    }
+
+}
-- 
GitLab