Examples with GetClientRequest voldemort.store.socket.clientrequest.GetClientRequest used on opensource projects

Example 1 with GetClientRequest

use of voldemort.store.socket.clientrequest.GetClientRequest in project voldemort by voldemort.

the class ClientRequestExecutorPoolTest method testCloseWithOutstandingQueue.

@Test
public void testCloseWithOutstandingQueue() {
    final int MAX_CONNECTIONS = 2;
    ClientRequestExecutorPool execPool = new ClientRequestExecutorPool(2, MAX_CONNECTIONS, CONNECTION_TIMEOUT_MS, SOCKET_TIMEOUT_MS, IDLE_CONNECTION_TIMEOUT_MS, 32 * 1024, false, true, new String());
    // will always wait in the queue.
    for (int i = 0; i < MAX_CONNECTIONS; i++) {
        execPool.checkout(dest1);
    }
    for (int j = 0; j < 2; j++) {
        GetClientRequest clientRequest = new GetClientRequest("sampleStore", new RequestFormatFactory().getRequestFormat(dest1.getRequestFormatType()), RequestRoutingType.ROUTED, new ByteArray(new byte[] { 1, 2, 3 }), null);
        final AtomicInteger cancelledEvents = new AtomicInteger(0);
        NonblockingStoreCallback callback = new NonblockingStoreCallback() {

            @Override
            public void requestComplete(Object result, long requestTime) {
                if (result instanceof UnreachableStoreException)
                    cancelledEvents.incrementAndGet();
                else
                    fail("The request must have failed with UnreachableException" + result);
            }
        };
        int queuedRequestCount = 20;
        for (int i = 0; i < queuedRequestCount; i++) {
            execPool.submitAsync(dest1, clientRequest, callback, 5000, "get");
        }
        int outstandingQueue = execPool.internalGetQueuedPool().getRegisteredResourceRequestCount(dest1);
        assertEquals("Queued request count should match", queuedRequestCount, outstandingQueue);
        // Now reset the queue, the outstanding requests should fail with
        // UnreachableStoreException
        execPool.close(dest1);
        outstandingQueue = execPool.internalGetQueuedPool().getRegisteredResourceRequestCount(dest1);
        assertEquals("Queued request should have been cleared", 0, outstandingQueue);
        // CancelledEvents should be
        assertEquals("All Queuedrequest must have been cancelled.", queuedRequestCount, cancelledEvents.get());
    }
}

Example 2 with GetClientRequest

use of voldemort.store.socket.clientrequest.GetClientRequest in project voldemort by voldemort.

the class SocketStore method submitGetRequest.

@Override
public void submitGetRequest(ByteArray key, byte[] transforms, NonblockingStoreCallback callback, long timeoutMs) {
    StoreUtils.assertValidKey(key);
    GetClientRequest clientRequest = new GetClientRequest(getName(), requestFormat, requestRoutingType, key, transforms);
    if (logger.isDebugEnabled())
        logger.debug("GET keyRef: " + System.identityHashCode(key) + " requestRef: " + System.identityHashCode(clientRequest));
    requestAsync(clientRequest, callback, timeoutMs, "get");
}

Example 3 with GetClientRequest

use of voldemort.store.socket.clientrequest.GetClientRequest in project voldemort by voldemort.

the class SocketStore method get.

@Override
public List<Versioned<byte[]>> get(ByteArray key, byte[] transforms) throws VoldemortException {
    StoreUtils.assertValidKey(key);
    GetClientRequest clientRequest = new GetClientRequest(getName(), requestFormat, requestRoutingType, key, transforms);
    if (logger.isDebugEnabled())
        logger.debug("GET keyRef: " + System.identityHashCode(key) + " requestRef: " + System.identityHashCode(clientRequest));
    return request(clientRequest, "get");
}