You are viewing a plain text version of this content. The canonical link for it is here.
Posted to notifications@shardingsphere.apache.org by zh...@apache.org on 2022/10/29 13:48:53 UTC

[shardingsphere] branch master updated: Rename DistributedLockProvider (#21839)

This is an automated email from the ASF dual-hosted git repository.

zhaojinchao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new a1e490dc9ed Rename DistributedLockProvider (#21839)
a1e490dc9ed is described below

commit a1e490dc9ed7c68013d8f02e0a79ead44dc4b740
Author: Liang Zhang <zh...@apache.org>
AuthorDate: Sat Oct 29 21:48:46 2022 +0800

    Rename DistributedLockProvider (#21839)
---
 ...tedLockProvider.java => DistributedLockHolder.java} |  4 ++--
 .../repository/cluster/consul/ConsulRepository.java    | 12 ++++++------
 ...kProvider.java => ConsulDistributedLockHolder.java} |  6 +++---
 .../cluster/consul/ConsulRepositoryTest.java           |  4 ++--
 .../mode/repository/cluster/etcd/EtcdRepository.java   |  8 ++++----
 ...ockProvider.java => EtcdDistributedLockHolder.java} |  6 +++---
 .../cluster/zookeeper/CuratorZookeeperRepository.java  | 12 ++++++------
 ...java => CuratorZookeeperDistributedLockHolder.java} |  6 +++---
 .../zookeeper/CuratorZookeeperRepositoryTest.java      | 18 +++++++++---------
 9 files changed, 38 insertions(+), 38 deletions(-)

diff --git a/mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockProvider.java b/mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockHolder.java
similarity index 93%
rename from mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockProvider.java
rename to mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockHolder.java
index e9a5bf7c7cf..ce388370c2b 100644
--- a/mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockProvider.java
+++ b/mode/type/cluster/repository/api/src/main/java/org/apache/shardingsphere/mode/repository/cluster/lock/DistributedLockHolder.java
@@ -18,9 +18,9 @@
 package org.apache.shardingsphere.mode.repository.cluster.lock;
 
 /**
- * Distributed lock provider.
+ * Distributed lock holder.
  */
-public interface DistributedLockProvider {
+public interface DistributedLockHolder {
     
     /**
      * Get distributed lock.
diff --git a/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepository.java b/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepository.java
index 6aa7bbcab4f..8c541263fa6 100644
--- a/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepository.java
+++ b/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepository.java
@@ -27,7 +27,7 @@ import com.ecwid.consul.v1.session.model.Session;
 import org.apache.shardingsphere.infra.instance.metadata.InstanceMetaData;
 import org.apache.shardingsphere.mode.repository.cluster.ClusterPersistRepository;
 import org.apache.shardingsphere.mode.repository.cluster.ClusterPersistRepositoryConfiguration;
-import org.apache.shardingsphere.mode.repository.cluster.consul.lock.ConsulDistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.consul.lock.ConsulDistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.consul.props.ConsulProperties;
 import org.apache.shardingsphere.mode.repository.cluster.consul.props.ConsulPropertyKey;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent;
@@ -50,7 +50,7 @@ public class ConsulRepository implements ClusterPersistRepository {
     
     private ConsulProperties consulProps;
     
-    private ConsulDistributedLockProvider consulDistributedLockProvider;
+    private ConsulDistributedLockHolder consulDistributedLockHolder;
     
     private Map<String, Collection<String>> watchKeyMap;
     
@@ -58,7 +58,7 @@ public class ConsulRepository implements ClusterPersistRepository {
     public void init(final ClusterPersistRepositoryConfiguration config, final InstanceMetaData instanceMetaData) {
         consulClient = new ShardingSphereConsulClient(new ConsulRawClient(config.getServerLists()));
         consulProps = new ConsulProperties(config.getProps());
-        consulDistributedLockProvider = new ConsulDistributedLockProvider(consulClient, consulProps);
+        consulDistributedLockHolder = new ConsulDistributedLockHolder(consulClient, consulProps);
         watchKeyMap = new HashMap<>(6, 1);
     }
     
@@ -106,7 +106,7 @@ public class ConsulRepository implements ClusterPersistRepository {
         PutParams putParams = new PutParams();
         putParams.setAcquireSession(sessionId);
         consulClient.setKVValue(key, value, putParams);
-        consulDistributedLockProvider.generatorFlushSessionTtlTask(consulClient, sessionId);
+        consulDistributedLockHolder.generatorFlushSessionTtlTask(consulClient, sessionId);
     }
     
     private NewSession createNewSession(final String key) {
@@ -124,12 +124,12 @@ public class ConsulRepository implements ClusterPersistRepository {
     
     @Override
     public boolean tryLock(final String lockKey, final long timeoutMillis) {
-        return consulDistributedLockProvider.getDistributedLock(lockKey).tryLock(timeoutMillis);
+        return consulDistributedLockHolder.getDistributedLock(lockKey).tryLock(timeoutMillis);
     }
     
     @Override
     public void unlock(final String lockKey) {
-        consulDistributedLockProvider.getDistributedLock(lockKey).unlock();
+        consulDistributedLockHolder.getDistributedLock(lockKey).unlock();
     }
     
     @Override
diff --git a/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockProvider.java b/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockHolder.java
similarity index 95%
rename from mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockProvider.java
rename to mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockHolder.java
index 2d0b7af782b..cd67c6eb4f2 100644
--- a/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockProvider.java
+++ b/mode/type/cluster/repository/provider/consul/src/main/java/org/apache/shardingsphere/mode/repository/cluster/consul/lock/ConsulDistributedLockHolder.java
@@ -24,7 +24,7 @@ import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.shardingsphere.mode.repository.cluster.consul.props.ConsulProperties;
 import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLock;
-import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockHolder;
 
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
@@ -32,11 +32,11 @@ import java.util.concurrent.ScheduledThreadPoolExecutor;
 import java.util.concurrent.TimeUnit;
 
 /**
- * Consul distributed lock provider.
+ * Consul distributed lock holder.
  */
 @RequiredArgsConstructor
 @Slf4j
-public class ConsulDistributedLockProvider implements DistributedLockProvider {
+public class ConsulDistributedLockHolder implements DistributedLockHolder {
     
     private static final ScheduledThreadPoolExecutor SESSION_FLUSH_EXECUTOR = new ScheduledThreadPoolExecutor(2);
     
diff --git a/mode/type/cluster/repository/provider/consul/src/test/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepositoryTest.java b/mode/type/cluster/repository/provider/consul/src/test/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepositoryTest.java
index ba734a2e14f..7a626d7f6d1 100644
--- a/mode/type/cluster/repository/provider/consul/src/test/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepositoryTest.java
+++ b/mode/type/cluster/repository/provider/consul/src/test/java/org/apache/shardingsphere/mode/repository/cluster/consul/ConsulRepositoryTest.java
@@ -23,7 +23,7 @@ import com.ecwid.consul.v1.kv.model.GetValue;
 import com.ecwid.consul.v1.kv.model.PutParams;
 import com.ecwid.consul.v1.session.model.NewSession;
 import lombok.SneakyThrows;
-import org.apache.shardingsphere.mode.repository.cluster.consul.lock.ConsulDistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.consul.lock.ConsulDistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.consul.props.ConsulProperties;
 import org.junit.Before;
 import org.junit.Ignore;
@@ -98,7 +98,7 @@ public final class ConsulRepositoryTest {
         // when(responseGetValueList.getValue()).thenReturn(getValueList);
         when(client.setKVValue(any(String.class), any(String.class))).thenReturn(responseBoolean);
         Plugins.getMemberAccessor().set(repository.getClass().getDeclaredField("consulClient"), repository, client);
-        Plugins.getMemberAccessor().set(repository.getClass().getDeclaredField("consulDistributedLockProvider"), repository, mock(ConsulDistributedLockProvider.class));
+        Plugins.getMemberAccessor().set(repository.getClass().getDeclaredField("consulDistributedLockHolder"), repository, mock(ConsulDistributedLockHolder.class));
     }
     
     @SneakyThrows(ReflectiveOperationException.class)
diff --git a/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/EtcdRepository.java b/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/EtcdRepository.java
index e49a6043616..0a2df21a5a1 100644
--- a/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/EtcdRepository.java
+++ b/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/EtcdRepository.java
@@ -35,13 +35,13 @@ import lombok.SneakyThrows;
 import org.apache.shardingsphere.infra.instance.metadata.InstanceMetaData;
 import org.apache.shardingsphere.mode.repository.cluster.ClusterPersistRepository;
 import org.apache.shardingsphere.mode.repository.cluster.ClusterPersistRepositoryConfiguration;
-import org.apache.shardingsphere.mode.repository.cluster.etcd.lock.EtcdDistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.etcd.lock.EtcdDistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.etcd.props.EtcdProperties;
 import org.apache.shardingsphere.mode.repository.cluster.etcd.props.EtcdPropertyKey;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent.Type;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEventListener;
-import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockHolder;
 
 import java.nio.charset.StandardCharsets;
 import java.util.List;
@@ -57,7 +57,7 @@ public final class EtcdRepository implements ClusterPersistRepository {
     
     private EtcdProperties etcdProps;
     
-    private DistributedLockProvider distributedLockHolder;
+    private DistributedLockHolder distributedLockHolder;
     
     @Override
     public void init(final ClusterPersistRepositoryConfiguration config, final InstanceMetaData instanceMetaData) {
@@ -66,7 +66,7 @@ public final class EtcdRepository implements ClusterPersistRepository {
                 .namespace(ByteSequence.from(config.getNamespace(), StandardCharsets.UTF_8))
                 .maxInboundMessageSize((int) 32e9)
                 .build();
-        distributedLockHolder = new EtcdDistributedLockProvider(client, etcdProps);
+        distributedLockHolder = new EtcdDistributedLockHolder(client, etcdProps);
     }
     
     @SneakyThrows({InterruptedException.class, ExecutionException.class})
diff --git a/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockProvider.java b/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockHolder.java
similarity index 94%
rename from mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockProvider.java
rename to mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockHolder.java
index 99369227468..0525485c96a 100644
--- a/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockProvider.java
+++ b/mode/type/cluster/repository/provider/etcd/src/main/java/org/apache/shardingsphere/mode/repository/cluster/etcd/lock/EtcdDistributedLockHolder.java
@@ -23,17 +23,17 @@ import lombok.extern.slf4j.Slf4j;
 import org.apache.shardingsphere.mode.repository.cluster.etcd.props.EtcdProperties;
 import org.apache.shardingsphere.mode.repository.cluster.etcd.props.EtcdPropertyKey;
 import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLock;
-import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockHolder;
 
 import java.util.HashMap;
 import java.util.Map;
 
 /**
- * Etcd distributed lock provider.
+ * Etcd distributed lock holder.
  */
 @RequiredArgsConstructor
 @Slf4j
-public final class EtcdDistributedLockProvider implements DistributedLockProvider {
+public final class EtcdDistributedLockHolder implements DistributedLockHolder {
     
     private final Map<String, EtcdDistributedLock> locks = new HashMap<>();
     
diff --git a/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepository.java b/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepository.java
index 798385a1499..75e17d9051a 100644
--- a/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepository.java
+++ b/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepository.java
@@ -36,10 +36,10 @@ import org.apache.shardingsphere.mode.repository.cluster.exception.ClusterPersis
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent.Type;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEventListener;
-import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.handler.CuratorZookeeperExceptionHandler;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.listener.SessionConnectionListener;
-import org.apache.shardingsphere.mode.repository.cluster.zookeeper.lock.CuratorZookeeperDistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.zookeeper.lock.CuratorZookeeperDistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.props.ZookeeperProperties;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.props.ZookeeperPropertyKey;
 import org.apache.zookeeper.CreateMode;
@@ -67,7 +67,7 @@ public final class CuratorZookeeperRepository implements ClusterPersistRepositor
     
     private CuratorFramework client;
     
-    private DistributedLockProvider distributedLockProvider;
+    private DistributedLockHolder distributedLockHolder;
     
     private InstanceMetaData instanceMetaData;
     
@@ -76,7 +76,7 @@ public final class CuratorZookeeperRepository implements ClusterPersistRepositor
         this.instanceMetaData = instanceMetaData;
         ZookeeperProperties zookeeperProps = new ZookeeperProperties(config.getProps());
         client = buildCuratorClient(config, zookeeperProps);
-        distributedLockProvider = new CuratorZookeeperDistributedLockProvider(client);
+        distributedLockHolder = new CuratorZookeeperDistributedLockHolder(client);
         initCuratorClient(zookeeperProps);
     }
     
@@ -276,12 +276,12 @@ public final class CuratorZookeeperRepository implements ClusterPersistRepositor
     
     @Override
     public boolean tryLock(final String lockKey, final long timeoutMillis) {
-        return distributedLockProvider.getDistributedLock(lockKey).tryLock(timeoutMillis);
+        return distributedLockHolder.getDistributedLock(lockKey).tryLock(timeoutMillis);
     }
     
     @Override
     public void unlock(final String lockKey) {
-        distributedLockProvider.getDistributedLock(lockKey).unlock();
+        distributedLockHolder.getDistributedLock(lockKey).unlock();
     }
     
     @Override
diff --git a/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockProvider.java b/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockHolder.java
similarity index 91%
rename from mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockProvider.java
rename to mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockHolder.java
index 26afce0601f..91b4184f767 100644
--- a/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockProvider.java
+++ b/mode/type/cluster/repository/provider/zookeeper-curator/src/main/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/lock/CuratorZookeeperDistributedLockHolder.java
@@ -21,16 +21,16 @@ import lombok.RequiredArgsConstructor;
 import org.apache.curator.framework.CuratorFramework;
 import org.apache.curator.framework.recipes.locks.InterProcessMutex;
 import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLock;
-import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.lock.DistributedLockHolder;
 
 import java.util.HashMap;
 import java.util.Map;
 
 /**
- * Curator ZooKeeper distributed lock provider.
+ * Curator ZooKeeper distributed lock holder.
  */
 @RequiredArgsConstructor
-public final class CuratorZookeeperDistributedLockProvider implements DistributedLockProvider {
+public final class CuratorZookeeperDistributedLockHolder implements DistributedLockHolder {
     
     private final Map<String, CuratorZookeeperDistributedLock> locks = new HashMap<>();
     
diff --git a/mode/type/cluster/repository/provider/zookeeper-curator/src/test/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepositoryTest.java b/mode/type/cluster/repository/provider/zookeeper-curator/src/test/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepositoryTest.java
index 2583e033e5f..7f4a65cd7e5 100644
--- a/mode/type/cluster/repository/provider/zookeeper-curator/src/test/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepositoryTest.java
+++ b/mode/type/cluster/repository/provider/zookeeper-curator/src/test/java/org/apache/shardingsphere/mode/repository/cluster/zookeeper/CuratorZookeeperRepositoryTest.java
@@ -40,7 +40,7 @@ import org.apache.shardingsphere.mode.repository.cluster.ClusterPersistRepositor
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent;
 import org.apache.shardingsphere.mode.repository.cluster.listener.DataChangedEvent.Type;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.lock.CuratorZookeeperDistributedLock;
-import org.apache.shardingsphere.mode.repository.cluster.zookeeper.lock.CuratorZookeeperDistributedLockProvider;
+import org.apache.shardingsphere.mode.repository.cluster.zookeeper.lock.CuratorZookeeperDistributedLockHolder;
 import org.apache.shardingsphere.mode.repository.cluster.zookeeper.props.ZookeeperPropertyKey;
 import org.apache.zookeeper.CreateMode;
 import org.apache.zookeeper.data.Stat;
@@ -123,7 +123,7 @@ public final class CuratorZookeeperRepositoryTest {
         mockBuilder();
         ClusterPersistRepositoryConfiguration config = new ClusterPersistRepositoryConfiguration(REPOSITORY.getType(), "governance", SERVER_LISTS, new Properties());
         REPOSITORY.init(config, new ProxyInstanceMetaData("foo_id", 3307));
-        mockInternalLockHolder();
+        mockDistributedLockHolder();
     }
     
     @SneakyThrows({ReflectiveOperationException.class, InterruptedException.class})
@@ -143,14 +143,14 @@ public final class CuratorZookeeperRepositoryTest {
     }
     
     @SneakyThrows(ReflectiveOperationException.class)
-    private void mockInternalLockHolder() {
-        Field internalLockProviderFiled = CuratorZookeeperRepository.class.getDeclaredField("distributedLockProvider");
-        internalLockProviderFiled.setAccessible(true);
-        CuratorZookeeperDistributedLockProvider distributedLockProvider = new CuratorZookeeperDistributedLockProvider(client);
-        Field locksFiled = CuratorZookeeperDistributedLockProvider.class.getDeclaredField("locks");
+    private void mockDistributedLockHolder() {
+        Field distributedLockHolderField = CuratorZookeeperRepository.class.getDeclaredField("distributedLockHolder");
+        distributedLockHolderField.setAccessible(true);
+        CuratorZookeeperDistributedLockHolder distributedLockHolder = new CuratorZookeeperDistributedLockHolder(client);
+        Field locksFiled = CuratorZookeeperDistributedLockHolder.class.getDeclaredField("locks");
         locksFiled.setAccessible(true);
-        locksFiled.set(distributedLockProvider, Collections.singletonMap("/locks/glock", new CuratorZookeeperDistributedLock(interProcessLock)));
-        internalLockProviderFiled.set(REPOSITORY, distributedLockProvider);
+        locksFiled.set(distributedLockHolder, Collections.singletonMap("/locks/glock", new CuratorZookeeperDistributedLock(interProcessLock)));
+        distributedLockHolderField.set(REPOSITORY, distributedLockHolder);
     }
     
     private void mockBuilder() {