Examples with GetStorageContainerEndpointRequest org.apache.bookkeeper.stream.proto.storage.GetStorageContainerEndpointRequest used on opensource projects

Example 1 with GetStorageContainerEndpointRequest

use of org.apache.bookkeeper.stream.proto.storage.GetStorageContainerEndpointRequest in project bookkeeper by apache.

the class TestLocationClientImpl method testLocateStorageContainersSucceedAfterRetried.

@Test
public void testLocateStorageContainersSucceedAfterRetried() throws Exception {
    serviceRegistry.removeService(locationServiceDefinition);
    final AtomicInteger retries = new AtomicInteger(3);
    StatusRuntimeException statusException = new StatusRuntimeException(Status.INTERNAL);
    StorageContainerServiceImplBase locationServiceWithFailures = new StorageContainerServiceImplBase() {

        @Override
        public void getStorageContainerEndpoint(GetStorageContainerEndpointRequest request, StreamObserver<GetStorageContainerEndpointResponse> responseObserver) {
            if (retries.decrementAndGet() == 0) {
                locationService.getStorageContainerEndpoint(request, responseObserver);
                return;
            }
            responseObserver.onError(statusException);
        }
    };
    serviceRegistry.addService(locationServiceWithFailures.bindService());
    testLocateStorageContainersSuccess();
    assertEquals(0, retries.get());
}

Example 2 with GetStorageContainerEndpointRequest

use of org.apache.bookkeeper.stream.proto.storage.GetStorageContainerEndpointRequest in project bookkeeper by apache.

the class TestLocationClientImpl method testLocateStorageContainersFailureAfterRetried.

@Test
public void testLocateStorageContainersFailureAfterRetried() throws Exception {
    serviceRegistry.removeService(locationServiceDefinition);
    final AtomicInteger retries = new AtomicInteger(3);
    StatusRuntimeException statusException = new StatusRuntimeException(Status.INTERNAL);
    StorageContainerServiceImplBase locationServiceWithFailures = new StorageContainerServiceImplBase() {

        @Override
        public void getStorageContainerEndpoint(GetStorageContainerEndpointRequest request, StreamObserver<GetStorageContainerEndpointResponse> responseObserver) {
            if (retries.decrementAndGet() == 0) {
                responseObserver.onError(new StatusRuntimeException(Status.INVALID_ARGUMENT));
                return;
            }
            responseObserver.onError(statusException);
        }
    };
    serviceRegistry.addService(locationServiceWithFailures.bindService());
    CompletableFuture<List<OneStorageContainerEndpointResponse>> future = locationClient.locateStorageContainers(Lists.newArrayList(Revisioned.of(1L, IRevisioned.ANY_REVISION)));
    try {
        future.get();
        fail("should fail with exception");
    } catch (ExecutionException ee) {
        assertNotNull(ee.getCause());
        assertTrue(ee.getCause() instanceof StatusRuntimeException);
        assertEquals(Status.INVALID_ARGUMENT, ((StatusRuntimeException) ee.getCause()).getStatus());
    }
    assertEquals(0, retries.get());
}

Example 3 with GetStorageContainerEndpointRequest

use of org.apache.bookkeeper.stream.proto.storage.GetStorageContainerEndpointRequest in project bookkeeper by apache.

the class TestProtocolInternalUtils method testCreateGetStorageContainerEndpointRequest.

// 
// Test Location Server Requests
// 
@Test
public void testCreateGetStorageContainerEndpointRequest() {
    List<Revisioned<Long>> scs = Lists.newArrayList(Revisioned.of(1000L, 1L), Revisioned.of(2000L, 2L), Revisioned.of(3000L, 3L));
    GetStorageContainerEndpointRequest request = createGetStorageContainerEndpointRequest(scs);
    assertEquals(3, request.getRequestsCount());
    int i = 1;
    for (OneStorageContainerEndpointRequest oneRequest : request.getRequestsList()) {
        assertEquals(1000L * i, oneRequest.getStorageContainer());
        assertEquals(1L * i, oneRequest.getRevision());
        ++i;
    }
}

Example 4 with GetStorageContainerEndpointRequest

use of org.apache.bookkeeper.stream.proto.storage.GetStorageContainerEndpointRequest in project bookkeeper by apache.

the class GrpcClientTestBase method setUp.

@Before
public void setUp() throws Exception {
    fakeServer = InProcessServerBuilder.forName(serverName).fallbackHandlerRegistry(serviceRegistry).directExecutor().build().start();
    scheduler = OrderedScheduler.newSchedulerBuilder().name("scheduler-" + getClass()).numThreads(Runtime.getRuntime().availableProcessors()).build();
    settings = StorageClientSettings.newBuilder().managedChannelBuilder(InProcessChannelBuilder.forName(serverName).directExecutor()).usePlaintext(true).build();
    serverManager = new StorageServerClientManagerImpl(settings, resources.scheduler(), endpoint -> new StorageServerChannel(InProcessChannelBuilder.forName(serverName).directExecutor().build(), Optional.empty()));
    StorageContainerServiceImplBase scService = new StorageContainerServiceImplBase() {

        @Override
        public void getStorageContainerEndpoint(GetStorageContainerEndpointRequest request, StreamObserver<GetStorageContainerEndpointResponse> responseObserver) {
            GetStorageContainerEndpointResponse.Builder respBuilder = GetStorageContainerEndpointResponse.newBuilder();
            respBuilder.setStatusCode(StatusCode.SUCCESS);
            for (OneStorageContainerEndpointRequest oneReq : request.getRequestsList()) {
                OneStorageContainerEndpointResponse oneResp = OneStorageContainerEndpointResponse.newBuilder().setEndpoint(StorageContainerEndpoint.newBuilder().setStorageContainerId(oneReq.getStorageContainer()).setRevision(oneReq.getRevision() + 1).setRwEndpoint(ENDPOINT)).build();
                respBuilder.addResponses(oneResp);
            }
            responseObserver.onNext(respBuilder.build());
            responseObserver.onCompleted();
        }
    };
    serviceRegistry.addService(scService.bindService());
    doSetup();
}