fix

Author: KirillKurdyukov

shedlock-ydb/pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>tech.ydb.dialects</groupId>
     <artifactId>shedlock-ydb</artifactId>
-    <version>0.2.0</version>
+    <version>0.3.0</version>
 
     <packaging>jar</packaging>
 

shedlock-ydb/src/main/java/tech/ydb/lock/provider/YdbJDBCLockProvider.java

@@ -75,10 +75,17 @@ private record YdbJDBCLock(String name, DataSource dataSource) implements Simple
         @Override
         public void unlock() {
             try (var connection = dataSource.getConnection()) {
-                var ps = connection.prepareStatement(
-                        "UPDATE shedlock SET lock_until = CurrentUtcTimestamp() WHERE name = ?");
-                ps.setObject(1, name);
-                ps.execute();
+                var autoCommit = connection.getAutoCommit();
+
+                try {
+                    connection.setAutoCommit(true);
+                    var ps = connection.prepareStatement(
+                            "UPDATE shedlock SET lock_until = CurrentUtcTimestamp() WHERE name = ?");
+                    ps.setObject(1, name);
+                    ps.execute();
+                } finally {
+                    connection.setAutoCommit(autoCommit);
+                }
             } catch (SQLException e) {
                 LOGGER.error(String.format("Instance[{%s}] release lock is failed", LOCKED_BY), e);
 

shedlock-ydb/src/test/java/tech/ydb/lock/provider/YdbLockProviderTest.java

@@ -3,9 +3,11 @@
 import java.sql.SQLException;
 import java.time.Duration;
 import java.time.Instant;
+import java.util.ArrayList;
 import java.util.Optional;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Executors;
+import java.util.concurrent.Future;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
 import javax.sql.DataSource;
@@ -67,12 +69,13 @@ public void integrationTest() throws ExecutionException, InterruptedException, S
                             ");");
         }
 
+        var lockFutures = new ArrayList<Future<?>>();
         var executorServer = Executors.newFixedThreadPool(10);
         var atomicInt = new AtomicInteger();
         var locked = new AtomicBoolean();
 
         for (int i = 0; i < 10; i++) {
-            executorServer.submit(() -> {
+            lockFutures.add(executorServer.submit(() -> {
                 Optional<SimpleLock> optinal = Optional.empty();
 
                 while (optinal.isEmpty()) {
@@ -96,9 +99,13 @@ public void integrationTest() throws ExecutionException, InterruptedException, S
                         simpleLock.unlock();
                     });
                 }
-            }).get();
+            }));
         }
 
-        Assertions.assertEquals(500, atomicInt.get());
+        for (var future : lockFutures) {
+            future.get();
+        }
+
+        Assertions.assertEquals(4950, atomicInt.get());
     }
 }