Added tests

Author: alex268

jdbc/src/main/java/tech/ydb/jdbc/context/TableTxExecutor.java

@@ -2,6 +2,8 @@
 
 import java.sql.SQLException;
 import java.util.concurrent.locks.ReentrantLock;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 import com.google.common.cache.Cache;
 import com.google.common.hash.Hashing;
@@ -31,6 +33,7 @@
  * @author mzinal
  */
 public class TableTxExecutor extends QueryServiceExecutor {
+    private static final Logger LOGGER = Logger.getLogger(TableTxExecutor.class.getName());
     private static final ReentrantLock VALIDATE_LOCK = new ReentrantLock();
 
     private static final String CREATE_SQL = ""
@@ -147,6 +150,7 @@ public static void validate(YdbContext ctx, String tablePath, Cache<String, Tabl
 
         VALIDATE_LOCK.lock();
         try {
+            LOGGER.log(Level.INFO, "Validate TxTableExecutor {0}", tablePath);
             if (cache.getIfPresent(tablePath) != null) {
                 return;
             }
@@ -158,6 +162,7 @@ public static void validate(YdbContext ctx, String tablePath, Cache<String, Tabl
 
             // validate table name
             Result<TableDescription> res = retryCtx.supplyResult(s -> s.describeTable(tablePath)).join();
+            LOGGER.log(Level.INFO, "Describe TxTableExecutor {0} -> {1}", new Object[] {tablePath, res.getStatus()});
             if (res.isSuccess()) {
                 cache.put(tablePath, res.getValue());
                 return;
@@ -171,12 +176,14 @@ public static void validate(YdbContext ctx, String tablePath, Cache<String, Tabl
             // Try to create a table
             String query = String.format(CREATE_SQL, tablePath);
             Status status = retryCtx.supplyStatus(session -> session.executeSchemeQuery(query)).join();
+            LOGGER.log(Level.INFO, "Create  TxTableExecutor {0} -> {1}", new Object[] {tablePath, status});
             if (!status.isSuccess()) {
                 throw ExceptionFactory.createException("Cannot initialize TableTxExecutor with tx table " + tablePath,
                         new UnexpectedResultException("Cannot create table", status));
             }
 
             Result<TableDescription> res2 = retryCtx.supplyResult(s -> s.describeTable(tablePath)).join();
+            LOGGER.log(Level.INFO, "Validate TxTableExecutor {0} -> {1}", new Object[] {tablePath, res2.getStatus()});
             if (!res2.isSuccess()) {
                 throw ExceptionFactory.createException("Cannot initialize TableTxExecutor with tx table " + tablePath,
                         new UnexpectedResultException("Cannot describe after creating", res2.getStatus()));

jdbc/src/test/java/tech/ydb/jdbc/YdbDriverTxValidateTest.java

@@ -5,6 +5,9 @@
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.sql.Statement;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.CompletableFuture;
 
 import org.junit.Assert;
 import org.junit.jupiter.api.Assertions;
@@ -14,6 +17,7 @@
 import tech.ydb.core.Status;
 import tech.ydb.core.StatusCode;
 import tech.ydb.core.UnexpectedResultException;
+import tech.ydb.jdbc.context.YdbContext;
 import tech.ydb.jdbc.impl.YdbTracerImpl;
 import tech.ydb.jdbc.impl.helper.ExceptionAssert;
 import tech.ydb.jdbc.impl.helper.JdbcConnectionExtention;
@@ -94,6 +98,33 @@ public void basicUsageTest() throws SQLException {
         }
     }
 
+    @Test
+    public void testContextCacheConncurrent() throws SQLException {
+        String url = jdbcURL.withArg("withTxValidationTable", "tx2_store").build();
+        List<CompletableFuture<YdbConnection>> list = new ArrayList<>();
+
+        for (int idx = 0; idx < 20; idx++) {
+            list.add(CompletableFuture.supplyAsync(() -> {
+                try {
+                    Connection connection = DriverManager.getConnection(url);
+                    return connection.unwrap(YdbConnection.class);
+                } catch (SQLException ex) {
+                    throw new RuntimeException("Cannot connect", ex);
+                }
+            }));
+        }
+
+        YdbContext first = list.get(0).join().getCtx();
+
+        for (CompletableFuture<YdbConnection> future: list) {
+            Assertions.assertEquals(first, future.join().getCtx());
+        }
+
+        for (CompletableFuture<YdbConnection> future: list) {
+            future.join().close();
+        }
+    }
+
     @Test
     public void commitedTxTest() throws SQLException {
         String url = jdbcURL.withArg("withTxValidationTable", "tx1_store").build();