package com.hazelcast.concurrent.countdownlatch;
import com.hazelcast.config.Config;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.ICountDownLatch;
import com.hazelcast.core.LifecycleEvent;
import com.hazelcast.core.LifecycleListener;
import com.hazelcast.core.MemberAttributeEvent;
import com.hazelcast.core.MembershipEvent;
import com.hazelcast.core.MembershipListener;
import com.hazelcast.instance.GroupProperties;
import com.hazelcast.instance.HazelcastInstanceFactory;
import com.hazelcast.instance.Node;
import com.hazelcast.instance.TestUtil;
import com.hazelcast.test.HazelcastTestSupport;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import java.io.IOException;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
public class CountDownLatchSplitBrainTest extends HazelcastTestSupport {
@Before
@After
public void killAllHazelcastInstances() throws IOException {
HazelcastInstanceFactory.shutdownAll();
}
@Test
public void testCountDownLatchSplitBrain() throws InterruptedException {
Config config = newConfig();
HazelcastInstance h1 = Hazelcast.newHazelcastInstance(config);
HazelcastInstance h2 = Hazelcast.newHazelcastInstance(config);
HazelcastInstance h3 = Hazelcast.newHazelcastInstance(config);
final String name = generateKeyOwnedBy(h3);
ICountDownLatch countDownLatch = h3.getCountDownLatch(name);
countDownLatch.trySetCount(5);
TestMemberShipListener memberShipListener = new TestMemberShipListener(2);
h3.getCluster().addMembershipListener(memberShipListener);
TestLifeCycleListener lifeCycleListener = new TestLifeCycleListener(1);
h3.getLifecycleService().addLifecycleListener(lifeCycleListener);
countDownLatch.countDown();
closeConnectionBetween(h1, h3);
closeConnectionBetween(h2, h3);
assertOpenEventually(memberShipListener.latch);
assertClusterSizeEventually(2, h1);
assertClusterSizeEventually(2, h2);
assertClusterSizeEventually(1, h3);
ICountDownLatch countDownLatch1 = h1.getCountDownLatch(name);
countDownLatch1.countDown();
countDownLatch.countDown();
countDownLatch.countDown();
assertOpenEventually(lifeCycleListener.latch);
assertClusterSizeEventually(3, h1);
assertClusterSizeEventually(3, h2);
assertClusterSizeEventually(3, h3);
ICountDownLatch countDownLatchTest = h3.getCountDownLatch(name);
assertEquals(3, countDownLatchTest.getCount());
}
private void closeConnectionBetween(HazelcastInstance h1, HazelcastInstance h2) {
if (h1 == null || h2 == null) return;
final Node n1 = TestUtil.getNode(h1);
final Node n2 = TestUtil.getNode(h2);
n1.clusterService.removeAddress(n2.address);
n2.clusterService.removeAddress(n1.address);
}
private Config newConfig() {
Config config = new Config();
config.setProperty(GroupProperties.PROP_MERGE_FIRST_RUN_DELAY_SECONDS, "30");
config.setProperty(GroupProperties.PROP_MERGE_NEXT_RUN_DELAY_SECONDS, "3");
return config;
}
private class TestLifeCycleListener implements LifecycleListener {
CountDownLatch latch;
TestLifeCycleListener(int countdown) {
latch = new CountDownLatch(countdown);
}
@Override
public void stateChanged(LifecycleEvent event) {
if (event.getState() == LifecycleEvent.LifecycleState.MERGED) {
latch.countDown();
}
}
}
private class TestMemberShipListener implements MembershipListener {
final CountDownLatch latch;
TestMemberShipListener(int countdown) {
latch = new CountDownLatch(countdown);
}
@Override
public void memberAdded(MembershipEvent membershipEvent) {
}
@Override
public void memberRemoved(MembershipEvent membershipEvent) {
latch.countDown();
}
@Override
public void memberAttributeChanged(MemberAttributeEvent memberAttributeEvent) {
}
}
}