Examples with WaitKeyCancelledException com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException used on opensource projects

Example 6 with WaitKeyCancelledException

use of com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException in project hazelcast by hazelcast.

the class AbstractSemaphoreFailureTest method testDrainCancelsPendingAcquireRequestWhenNotAcquired.

@Test(timeout = 300_000)
public void testDrainCancelsPendingAcquireRequestWhenNotAcquired() throws InterruptedException {
    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 invUid = newUnsecureUUID();
    RaftInvocationManager invocationManager = getRaftInvocationManager(proxyInstance);
    InternalCompletableFuture<Object> f = invocationManager.invoke(groupId, new AcquirePermitsOp(objectName, sessionId, threadId, invUid, 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());
    });
    semaphore.drainPermits();
    try {
        f.joinInternal();
        fail();
    } catch (WaitKeyCancelledException ignored) {
    }
}

Example 7 with WaitKeyCancelledException

use of com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException in project hazelcast by hazelcast.

the class SessionAwareSemaphoreProxy method tryAcquire.

@Override
public boolean tryAcquire(int permits, long timeout, TimeUnit unit) {
    checkPositive(permits, "Permits must be positive!");
    long timeoutMs = max(0, unit.toMillis(timeout));
    long threadId = getThreadId();
    UUID invocationUid = newUnsecureUUID();
    long start;
    for (; ; ) {
        start = Clock.currentTimeMillis();
        long sessionId = sessionManager.acquireSession(this.groupId, permits);
        try {
            ClientMessage request = SemaphoreAcquireCodec.encodeRequest(groupId, objectName, sessionId, threadId, invocationUid, permits, timeoutMs);
            HazelcastClientInstanceImpl client = getClient();
            ClientMessage response = new ClientInvocation(client, request, objectName).invoke().joinInternal();
            boolean acquired = SemaphoreAcquireCodec.decodeResponse(response);
            if (!acquired) {
                sessionManager.releaseSession(this.groupId, sessionId, permits);
            }
            return acquired;
        } catch (SessionExpiredException e) {
            sessionManager.invalidateSession(this.groupId, sessionId);
            timeoutMs -= (Clock.currentTimeMillis() - start);
            if (timeoutMs <= 0) {
                return false;
            }
        } catch (WaitKeyCancelledException e) {
            sessionManager.releaseSession(this.groupId, sessionId, permits);
            return false;
        } catch (RuntimeException e) {
            sessionManager.releaseSession(this.groupId, sessionId, permits);
            throw e;
        }
    }
}

Example 8 with WaitKeyCancelledException

use of com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException in project hazelcast by hazelcast.

the class SessionAwareSemaphoreProxy method acquire.

@Override
public void acquire(int permits) {
    checkPositive("permits", permits);
    long threadId = getThreadId();
    UUID invocationUid = newUnsecureUUID();
    for (; ; ) {
        long sessionId = sessionManager.acquireSession(this.groupId, permits);
        try {
            ClientMessage request = SemaphoreAcquireCodec.encodeRequest(groupId, objectName, sessionId, threadId, invocationUid, permits, -1);
            HazelcastClientInstanceImpl client = getClient();
            new ClientInvocation(client, request, objectName).invoke().joinInternal();
            return;
        } catch (SessionExpiredException e) {
            sessionManager.invalidateSession(this.groupId, sessionId);
        } catch (WaitKeyCancelledException e) {
            sessionManager.releaseSession(this.groupId, sessionId, permits);
            throw new IllegalStateException("Semaphore[" + objectName + "] not acquired because the acquire call " + "on the CP group is cancelled, possibly because of another indeterminate call from the same thread.");
        } catch (RuntimeException e) {
            sessionManager.releaseSession(this.groupId, sessionId, permits);
            throw e;
        }
    }
}

Example 9 with WaitKeyCancelledException

use of com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException in project hazelcast by hazelcast.

the class SessionlessSemaphoreProxy method doTryAcquire.

private boolean doTryAcquire(int permits, long timeoutMs) {
    long clusterWideThreadId = sessionManager.getOrCreateUniqueThreadId(groupId);
    UUID invocationUid = newUnsecureUUID();
    ClientMessage request = SemaphoreAcquireCodec.encodeRequest(groupId, objectName, NO_SESSION_ID, clusterWideThreadId, invocationUid, permits, timeoutMs);
    try {
        ClientMessage response = new ClientInvocation(getClient(), request, objectName).invoke().joinInternal();
        return SemaphoreAcquireCodec.decodeResponse(response);
    } catch (WaitKeyCancelledException e) {
        throw new IllegalStateException("Semaphore[" + objectName + "] not acquired because the acquire call " + "on the CP group is cancelled, possibly because of another indeterminate call from the same thread.");
    }
}

Example 10 with WaitKeyCancelledException

use of com.hazelcast.cp.internal.datastructures.exception.WaitKeyCancelledException in project hazelcast by hazelcast.

the class SessionAwareSemaphoreProxy method acquire.

@Override
public void acquire(int permits) {
    checkPositive(permits, "Permits must be positive!");
    long threadId = getThreadId();
    UUID invocationUid = newUnsecureUUID();
    for (; ; ) {
        long sessionId = acquireSession(permits);
        RaftOp op = new AcquirePermitsOp(objectName, sessionId, threadId, invocationUid, permits, -1L);
        try {
            invocationManager.invoke(groupId, op).joinInternal();
            return;
        } catch (SessionExpiredException e) {
            invalidateSession(sessionId);
        } catch (WaitKeyCancelledException e) {
            releaseSession(sessionId, permits);
            throw new IllegalStateException("Semaphore[" + objectName + "] not acquired because the acquire call " + "on the CP group is cancelled, possibly because of another indeterminate call from the same thread.");
        } catch (RuntimeException e) {
            releaseSession(sessionId, permits);
            throw e;
        }
    }
}