HBASE-19544 Add UTs for testing concurrent modifications on replication peer

Signed-off-by: zhangduo <zhangduo@apache.org>
This commit is contained in:
Guanghao Zhang 2018-01-02 17:07:41 +08:00 committed by zhangduo
parent d357920988
commit d95ee41583
1 changed files with 69 additions and 0 deletions

View File

@ -31,6 +31,7 @@ import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.atomic.AtomicLong;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
@ -55,6 +56,8 @@ import org.junit.Rule;
import org.junit.Test; import org.junit.Test;
import org.junit.experimental.categories.Category; import org.junit.experimental.categories.Category;
import org.junit.rules.TestName; import org.junit.rules.TestName;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/** /**
* Unit testing of ReplicationAdmin * Unit testing of ReplicationAdmin
@ -62,6 +65,8 @@ import org.junit.rules.TestName;
@Category({MediumTests.class, ClientTests.class}) @Category({MediumTests.class, ClientTests.class})
public class TestReplicationAdmin { public class TestReplicationAdmin {
private static final Logger LOG = LoggerFactory.getLogger(TestReplicationAdmin.class);
private final static HBaseTestingUtility TEST_UTIL = private final static HBaseTestingUtility TEST_UTIL =
new HBaseTestingUtility(); new HBaseTestingUtility();
@ -110,6 +115,70 @@ public class TestReplicationAdmin {
} }
} }
@Test
public void testConcurrentPeerOperations() throws Exception {
int threadNum = 5;
AtomicLong successCount = new AtomicLong(0);
// Test concurrent add peer operation
Thread[] addPeers = new Thread[threadNum];
for (int i = 0; i < threadNum; i++) {
addPeers[i] = new Thread(() -> {
try {
hbaseAdmin.addReplicationPeer(ID_ONE,
ReplicationPeerConfig.newBuilder().setClusterKey(KEY_ONE).build());
successCount.incrementAndGet();
} catch (Exception e) {
LOG.debug("Got exception when add replication peer", e);
}
});
addPeers[i].start();
}
for (Thread addPeer : addPeers) {
addPeer.join();
}
assertEquals(1, successCount.get());
// Test concurrent remove peer operation
successCount.set(0);
Thread[] removePeers = new Thread[threadNum];
for (int i = 0; i < threadNum; i++) {
removePeers[i] = new Thread(() -> {
try {
hbaseAdmin.removeReplicationPeer(ID_ONE);
successCount.incrementAndGet();
} catch (Exception e) {
LOG.debug("Got exception when remove replication peer", e);
}
});
removePeers[i].start();
}
for (Thread removePeer : removePeers) {
removePeer.join();
}
assertEquals(1, successCount.get());
// Test concurrent add peer operation again
successCount.set(0);
addPeers = new Thread[threadNum];
for (int i = 0; i < threadNum; i++) {
addPeers[i] = new Thread(() -> {
try {
hbaseAdmin.addReplicationPeer(ID_ONE,
ReplicationPeerConfig.newBuilder().setClusterKey(KEY_ONE).build());
successCount.incrementAndGet();
} catch (Exception e) {
LOG.debug("Got exception when add replication peer", e);
}
});
addPeers[i].start();
}
for (Thread addPeer : addPeers) {
addPeer.join();
}
assertEquals(1, successCount.get());
}
@Test @Test
public void testAddInvalidPeer() { public void testAddInvalidPeer() {
ReplicationPeerConfigBuilder builder = ReplicationPeerConfig.newBuilder(); ReplicationPeerConfigBuilder builder = ReplicationPeerConfig.newBuilder();