Examples with RaftInvocationManager com.hazelcast.cp.internal.RaftInvocationManager used on opensource projects

Example 11 with RaftInvocationManager

use of com.hazelcast.cp.internal.RaftInvocationManager in project hazelcast by hazelcast.

the class AbstractSemaphoreFailureTest method testRetriedAcquireReceivesPermitsOnlyOnce.

@Test(timeout = 300_000)
public void testRetriedAcquireReceivesPermitsOnlyOnce() throws InterruptedException, ExecutionException {
    semaphore.init(1);
    semaphore.acquire();
    semaphore.release();
    // if the session-aware semaphore is used, we guarantee that there is a session id now...
    RaftGroupId groupId = getGroupId(semaphore);
    long sessionId = getSessionId(proxyInstance, groupId);
    long threadId = getThreadId(groupId);
    UUID invUid1 = newUnsecureUUID();
    RaftInvocationManager invocationManager = getRaftInvocationManager(proxyInstance);
    InternalCompletableFuture<Object> f1 = invocationManager.invoke(groupId, new AcquirePermitsOp(objectName, sessionId, threadId, invUid1, 2, MINUTES.toMillis(5)));
    assertTrueEventually(() -> {
        SemaphoreService service = getNodeEngineImpl(primaryInstance).getService(SemaphoreService.SERVICE_NAME);
        SemaphoreRegistry registry = service.getRegistryOrNull(groupId);
        assertNotNull(registry);
        assertEquals(1, registry.getWaitTimeouts().size());
    });
    spawn(() -> {
        try {
            semaphore.tryAcquire(20, 5, TimeUnit.MINUTES);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    });
    assertTrueEventually(() -> {
        SemaphoreService service = getNodeEngineImpl(primaryInstance).getService(SemaphoreService.SERVICE_NAME);
        SemaphoreRegistry registry = service.getRegistryOrNull(groupId);
        assertEquals(2, registry.getWaitTimeouts().size());
    });
    InternalCompletableFuture<Object> f2 = invocationManager.invoke(groupId, new AcquirePermitsOp(objectName, sessionId, threadId, invUid1, 2, MINUTES.toMillis(5)));
    assertTrueEventually(() -> {
        SemaphoreService service = getNodeEngineImpl(primaryInstance).getService(SemaphoreService.SERVICE_NAME);
        SemaphoreRegistry registry = service.getRegistryOrNull(groupId);
        Semaphore semaphore = registry.getResourceOrNull(objectName);
        assertEquals(2, semaphore.getInternalWaitKeysMap().size());
    });
    spawn(() -> semaphore.increasePermits(3)).get();
    f1.joinInternal();
    f2.joinInternal();
    assertEquals(2, semaphore.availablePermits());
}

Example 12 with RaftInvocationManager

use of com.hazelcast.cp.internal.RaftInvocationManager in project hazelcast by hazelcast.

the class AbstractCPMessageTask method invoke.

protected void invoke(CPGroupId groupId, RaftOp op) {
    RaftInvocationManager invocationManager = getInvocationManager();
    InternalCompletableFuture<Object> future = invocationManager.invoke(groupId, op, false);
    future.whenCompleteAsync(this);
}

Example 13 with RaftInvocationManager

use of com.hazelcast.cp.internal.RaftInvocationManager in project hazelcast by hazelcast.

the class AbstractFencedLockFailureTest method testRetriedUnlockIsSuccessfulAfterLockedByAnotherEndpoint.

@Test(timeout = 300_000)
public void testRetriedUnlockIsSuccessfulAfterLockedByAnotherEndpoint() {
    lock.lock();
    RaftGroupId groupId = lock.getGroupId();
    long sessionId = getSessionManager().getSession(groupId);
    RaftInvocationManager invocationManager = getRaftInvocationManager();
    UUID invUid = newUnsecureUUID();
    invocationManager.invoke(groupId, new UnlockOp(objectName, sessionId, getThreadId(), invUid)).joinInternal();
    lockByOtherThread();
    invocationManager.invoke(groupId, new UnlockOp(objectName, sessionId, getThreadId(), invUid)).joinInternal();
}

Example 14 with RaftInvocationManager

use of com.hazelcast.cp.internal.RaftInvocationManager in project hazelcast by hazelcast.

the class AbstractFencedLockFailureTest method testExpiredAndRetriedTryLockRequestReceivesFailureResponse.

@Test(timeout = 300_000)
public void testExpiredAndRetriedTryLockRequestReceivesFailureResponse() {
    final CountDownLatch unlockLatch = new CountDownLatch(1);
    spawn(() -> {
        lock.lock();
        assertOpenEventually(unlockLatch);
        lock.unlock();
    });
    assertTrueEventually(() -> assertTrue(lock.isLocked()));
    final RaftGroupId groupId = lock.getGroupId();
    long sessionId = getSessionManager().getSession(groupId);
    RaftInvocationManager invocationManager = getRaftInvocationManager(proxyInstance);
    UUID invUid = newUnsecureUUID();
    InternalCompletableFuture<Long> f1 = invocationManager.invoke(groupId, new TryLockOp(objectName, sessionId, getThreadId(), invUid, SECONDS.toMillis(5)));
    long fence1 = f1.joinInternal();
    assertEquals(INVALID_FENCE, fence1);
    unlockLatch.countDown();
    assertTrueEventually(() -> assertFalse(lock.isLocked()));
    InternalCompletableFuture<Long> f2 = invocationManager.invoke(groupId, new TryLockOp(objectName, sessionId, getThreadId(), invUid, SECONDS.toMillis(5)));
    long fence2 = f2.joinInternal();
    assertEquals(INVALID_FENCE, fence2);
}

Example 15 with RaftInvocationManager

use of com.hazelcast.cp.internal.RaftInvocationManager in project hazelcast by hazelcast.

the class AbstractFencedLockFailureTest method testRetriedTryLockWithTimeoutDoesNotCancelPendingLockRequest.

@Test(timeout = 300_000)
public void testRetriedTryLockWithTimeoutDoesNotCancelPendingLockRequest() {
    lockByOtherThread();
    // there is a session id now
    RaftGroupId groupId = lock.getGroupId();
    long sessionId = getSessionManager().getSession(groupId);
    RaftInvocationManager invocationManager = getRaftInvocationManager();
    UUID invUid = newUnsecureUUID();
    invocationManager.invoke(groupId, new TryLockOp(objectName, sessionId, getThreadId(), invUid, MINUTES.toMillis(5)));
    NodeEngineImpl nodeEngine = getNodeEngineImpl(primaryInstance);
    LockService service = nodeEngine.getService(LockService.SERVICE_NAME);
    assertTrueEventually(() -> {
        LockRegistry registry = service.getRegistryOrNull(groupId);
        assertNotNull(registry);
        assertNotNull(registry.getResourceOrNull(objectName));
        assertEquals(1, registry.getWaitTimeouts().size());
    });
    invocationManager.invoke(groupId, new TryLockOp(objectName, sessionId, getThreadId(), invUid, MINUTES.toMillis(5)));
    assertTrueEventually(() -> {
        RaftService raftService = getNodeEngineImpl(primaryInstance).getService(RaftService.SERVICE_NAME);
        int partitionId = raftService.getCPGroupPartitionId(groupId);
        LockRegistry registry = service.getRegistryOrNull(groupId);
        boolean[] verified = new boolean[1];
        CountDownLatch latch = new CountDownLatch(1);
        OperationServiceImpl operationService = nodeEngine.getOperationService();
        operationService.execute(new PartitionSpecificRunnable() {

            @Override
            public int getPartitionId() {
                return partitionId;
            }

            @Override
            public void run() {
                Lock lock = registry.getResourceOrNull(objectName);
                Map<Object, WaitKeyContainer<LockInvocationKey>> waitKeys = lock.getInternalWaitKeysMap();
                verified[0] = (waitKeys.size() == 1 && waitKeys.values().iterator().next().retryCount() == 1);
                latch.countDown();
            }
        });
        latch.await(60, SECONDS);
        assertTrue(verified[0]);
    });
}