feat: fix

Author: KirillKurdyukov

shaded-lock-ydb/pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>tech.ydb.dialects</groupId>
     <artifactId>shaded-lock-ydb</artifactId>
-    <version>0.9.1</version>
+    <version>0.1.0</version>
 
     <packaging>jar</packaging>
 

shaded-lock-ydb/src/main/java/tech/ydb/lock/provider/YdbCoordinationServiceLockProvider.java

@@ -57,6 +57,7 @@ public Optional<SimpleLock> lock(LockConfiguration lockConfiguration) {
         if (semaphoreLease.isSuccess()) {
             logger.debug("Semaphore acquired");
 
+            System.out.println(semaphoreLease.getStatus());
             return Optional.of(new YdbSimpleLock(semaphoreLease.getValue()));
         } else {
             return Optional.empty();

shaded-lock-ydb/src/main/java/tech/ydb/lock/provider/YdbLockProviderConfiguration.java

@@ -14,6 +14,10 @@
 public class YdbLockProviderConfiguration {
     @Bean
     public YdbCoordinationServiceLockProvider ydbLockProvider(DataSource dataSource) throws SQLException {
-        return new YdbCoordinationServiceLockProvider(dataSource.getConnection().unwrap(YdbConnection.class));
+        var provider = new YdbCoordinationServiceLockProvider(dataSource.getConnection().unwrap(YdbConnection.class));
+
+        provider.init();
+
+        return provider;
     }
 }

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

@@ -7,6 +7,7 @@
 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 net.javacrumbs.shedlock.core.LockConfiguration;
 import net.javacrumbs.shedlock.core.SimpleLock;
@@ -53,27 +54,32 @@ private static String jdbcUrl() {
     public void integrationTest() throws ExecutionException, InterruptedException {
         var executorServer = Executors.newFixedThreadPool(10);
         var atomicInt = new AtomicInteger();
+        var locked = new AtomicBoolean();
         var futures = new ArrayList<Future<?>>();
 
         for (int i = 0; i < 100; i++) {
             final var ii = i;
             futures.add(executorServer.submit(() -> {
-                lockProvider.init();
-
                 Optional<SimpleLock> optinal = Optional.empty();
 
                 while (optinal.isEmpty()) {
                     optinal = lockProvider.lock(
                             new LockConfiguration(Instant.now(), "semaphore", Duration.ZERO, Duration.ZERO));
 
                     optinal.ifPresent(simpleLock -> {
+                        if (locked.get()) {
+                            throw new RuntimeException();
+                        }
+                        locked.set(true);
+
                         try {
                             Thread.sleep(100);
                         } catch (InterruptedException e) {
                             throw new RuntimeException(e);
                         }
 
                         atomicInt.addAndGet(ii);
+                        locked.set(false);
                         simpleLock.unlock();
                     });
                 }