Examples with RedisLockRegistry org.springframework.integration.redis.util.RedisLockRegistry used on opensource projects

Example 1 with RedisLockRegistry

use of org.springframework.integration.redis.util.RedisLockRegistry in project spring-integration by spring-projects.

the class RedisLockRegistryLeaderInitiatorTests method testDistributedLeaderElection.

@Test
@RedisAvailable
public void testDistributedLeaderElection() throws Exception {
    CountDownLatch granted = new CountDownLatch(1);
    CountingPublisher countingPublisher = new CountingPublisher(granted);
    List<LockRegistryLeaderInitiator> initiators = new ArrayList<>();
    for (int i = 0; i < 2; i++) {
        RedisLockRegistry registry = new RedisLockRegistry(getConnectionFactoryForTest(), "LeaderInitiator");
        LockRegistryLeaderInitiator initiator = new LockRegistryLeaderInitiator(registry, new DefaultCandidate("foo", "bar"));
        initiator.setLeaderEventPublisher(countingPublisher);
        initiators.add(initiator);
    }
    for (LockRegistryLeaderInitiator initiator : initiators) {
        initiator.start();
    }
    assertThat(granted.await(10, TimeUnit.SECONDS), is(true));
    LockRegistryLeaderInitiator initiator1 = countingPublisher.initiator;
    LockRegistryLeaderInitiator initiator2 = null;
    for (LockRegistryLeaderInitiator initiator : initiators) {
        if (initiator != initiator1) {
            initiator2 = initiator;
            break;
        }
    }
    assertNotNull(initiator2);
    assertThat(initiator1.getContext().isLeader(), is(true));
    assertThat(initiator2.getContext().isLeader(), is(false));
    final CountDownLatch granted1 = new CountDownLatch(1);
    final CountDownLatch granted2 = new CountDownLatch(1);
    CountDownLatch revoked1 = new CountDownLatch(1);
    CountDownLatch revoked2 = new CountDownLatch(1);
    CountDownLatch acquireLockFailed1 = new CountDownLatch(1);
    CountDownLatch acquireLockFailed2 = new CountDownLatch(1);
    initiator1.setLeaderEventPublisher(new CountingPublisher(granted1, revoked1, acquireLockFailed1) {

        @Override
        public void publishOnRevoked(Object source, Context context, String role) {
            try {
                // It's difficult to see round-robin election, so block one initiator until the second is elected.
                assertThat(granted2.await(10, TimeUnit.SECONDS), is(true));
            } catch (InterruptedException e) {
            // No op
            }
            super.publishOnRevoked(source, context, role);
        }
    });
    initiator2.setLeaderEventPublisher(new CountingPublisher(granted2, revoked2, acquireLockFailed2) {

        @Override
        public void publishOnRevoked(Object source, Context context, String role) {
            try {
                // It's difficult to see round-robin election, so block one initiator until the second is elected.
                assertThat(granted1.await(10, TimeUnit.SECONDS), is(true));
            } catch (InterruptedException e) {
            // No op
            }
            super.publishOnRevoked(source, context, role);
        }
    });
    initiator1.getContext().yield();
    assertThat(revoked1.await(10, TimeUnit.SECONDS), is(true));
    assertThat(initiator2.getContext().isLeader(), is(true));
    assertThat(initiator1.getContext().isLeader(), is(false));
    initiator2.getContext().yield();
    assertThat(revoked2.await(10, TimeUnit.SECONDS), is(true));
    assertThat(initiator1.getContext().isLeader(), is(true));
    assertThat(initiator2.getContext().isLeader(), is(false));
    initiator2.stop();
    CountDownLatch revoked11 = new CountDownLatch(1);
    initiator1.setLeaderEventPublisher(new CountingPublisher(new CountDownLatch(1), revoked11, new CountDownLatch(1)));
    initiator1.getContext().yield();
    assertThat(revoked11.await(10, TimeUnit.SECONDS), is(true));
    assertThat(initiator1.getContext().isLeader(), is(false));
    initiator1.stop();
}