HBASE-19686 Use KeyLocker instead of ReentrantLock in PeerProcedureHandlerImpl
Project: http://git-wip-us.apache.org/repos/asf/hbase/repo Commit: http://git-wip-us.apache.org/repos/asf/hbase/commit/6922f80b Tree: http://git-wip-us.apache.org/repos/asf/hbase/tree/6922f80b Diff: http://git-wip-us.apache.org/repos/asf/hbase/diff/6922f80b Branch: refs/heads/HBASE-19397 Commit: 6922f80b1ef15414a832c0f94e99d1cea6f5d9df Parents: eee095c Author: zhangduo <zhang...@apache.org> Authored: Tue Jan 2 16:13:55 2018 +0800 Committer: zhangduo <zhang...@apache.org> Committed: Thu Jan 4 09:22:35 2018 +0800 ---------------------------------------------------------------------- .../regionserver/PeerProcedureHandlerImpl.java | 41 ++++++++++++++------ 1 file changed, 29 insertions(+), 12 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/hbase/blob/6922f80b/hbase-server/src/main/java/org/apache/hadoop/hbase/replication/regionserver/PeerProcedureHandlerImpl.java ---------------------------------------------------------------------- diff --git a/hbase-server/src/main/java/org/apache/hadoop/hbase/replication/regionserver/PeerProcedureHandlerImpl.java b/hbase-server/src/main/java/org/apache/hadoop/hbase/replication/regionserver/PeerProcedureHandlerImpl.java index 1efe180..c09c6a0 100644 --- a/hbase-server/src/main/java/org/apache/hadoop/hbase/replication/regionserver/PeerProcedureHandlerImpl.java +++ b/hbase-server/src/main/java/org/apache/hadoop/hbase/replication/regionserver/PeerProcedureHandlerImpl.java @@ -19,10 +19,10 @@ package org.apache.hadoop.hbase.replication.regionserver; import java.io.IOException; -import java.util.concurrent.locks.ReentrantLock; - +import java.util.concurrent.locks.Lock; import org.apache.hadoop.hbase.replication.ReplicationException; import org.apache.hadoop.hbase.replication.ReplicationPeer.PeerState; +import org.apache.hadoop.hbase.util.KeyLocker; import org.apache.yetus.audience.InterfaceAudience; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -32,7 +32,7 @@ public class PeerProcedureHandlerImpl implements PeerProcedureHandler { private static final Logger LOG = LoggerFactory.getLogger(PeerProcedureHandlerImpl.class); private final ReplicationSourceManager replicationSourceManager; - private final ReentrantLock peersLock = new ReentrantLock(); + private final KeyLocker<String> peersLock = new KeyLocker<>(); public PeerProcedureHandlerImpl(ReplicationSourceManager replicationSourceManager) { this.replicationSourceManager = replicationSourceManager; @@ -40,40 +40,57 @@ public class PeerProcedureHandlerImpl implements PeerProcedureHandler { @Override public void addPeer(String peerId) throws ReplicationException, IOException { - peersLock.lock(); + Lock peerLock = peersLock.acquireLock(peerId); try { replicationSourceManager.addPeer(peerId); } finally { - peersLock.unlock(); + peerLock.unlock(); } } @Override public void removePeer(String peerId) throws ReplicationException, IOException { - peersLock.lock(); + Lock peerLock = peersLock.acquireLock(peerId); try { if (replicationSourceManager.getReplicationPeers().getPeer(peerId) != null) { replicationSourceManager.removePeer(peerId); } } finally { - peersLock.unlock(); + peerLock.unlock(); } } @Override public void disablePeer(String peerId) throws ReplicationException, IOException { - PeerState newState = replicationSourceManager.getReplicationPeers().refreshPeerState(peerId); - LOG.info("disable replication peer, id: " + peerId + ", new state: " + newState); + PeerState newState; + Lock peerLock = peersLock.acquireLock(peerId); + try { + newState = replicationSourceManager.getReplicationPeers().refreshPeerState(peerId); + } finally { + peerLock.unlock(); + } + LOG.info("disable replication peer, id: {}, new state: {}", peerId, newState); } @Override public void enablePeer(String peerId) throws ReplicationException, IOException { - PeerState newState = replicationSourceManager.getReplicationPeers().refreshPeerState(peerId); - LOG.info("enable replication peer, id: " + peerId + ", new state: " + newState); + PeerState newState; + Lock peerLock = peersLock.acquireLock(peerId); + try { + newState = replicationSourceManager.getReplicationPeers().refreshPeerState(peerId); + } finally { + peerLock.unlock(); + } + LOG.info("enable replication peer, id: {}, new state: {}", peerId, newState); } @Override public void updatePeerConfig(String peerId) throws ReplicationException, IOException { - replicationSourceManager.getReplicationPeers().refreshPeerConfig(peerId); + Lock peerLock = peersLock.acquireLock(peerId); + try { + replicationSourceManager.getReplicationPeers().refreshPeerConfig(peerId); + } finally { + peerLock.unlock(); + } } }