Examples with OperationContinuationHandlerContext software.amazon.awssdk.eventstreamrpc.OperationContinuationHandlerContext used on opensource projects

Example 1 with OperationContinuationHandlerContext

use of software.amazon.awssdk.eventstreamrpc.OperationContinuationHandlerContext in project aws-greengrass-cli by aws-greengrass.

the class CLIServiceTest method verifyHandlersRegisteredForAllOperations.

private void verifyHandlersRegisteredForAllOperations() {
    OperationContinuationHandlerContext mockContext = mock(OperationContinuationHandlerContext.class);
    ArgumentCaptor<Function> argumentCaptor = ArgumentCaptor.forClass(Function.class);
    verify(greengrassCoreIPCService).setGetComponentDetailsHandler(argumentCaptor.capture());
    verify(greengrassCoreIPCService).setListComponentsHandler(argumentCaptor.capture());
    argumentCaptor.getAllValues().stream().forEach(handler -> handler.apply(mockContext));
    verify(cliEventStreamAgent).getGetComponentDetailsHandler(mockContext);
    verify(cliEventStreamAgent).getListComponentsHandler(mockContext);
}

Example 2 with OperationContinuationHandlerContext

use of software.amazon.awssdk.eventstreamrpc.OperationContinuationHandlerContext in project aws-iot-device-sdk-java-v2 by aws.

the class TestIpcServiceHandler method getOperationHandler.

@Override
public Function<OperationContinuationHandlerContext, ? extends ServerConnectionContinuationHandler> getOperationHandler(final String operationName) {
    final EventStreamRPCServiceModel svcModel = this.serviceModel;
    System.out.println("Operation handler retrieved for operation name: " + operationName);
    return (context) -> {
        final TestOperationContinuationHandler handler = new TestOperationContinuationHandler(context) {

            @Override
            public OperationModelContext getOperationModelContext() {
                return svcModel.getOperationModelContext(operationName);
            }

            @Override
            protected void onStreamClosed() {
                streamClosed = true;
            }

            @Override
            public EventStreamJsonMessage handleRequest(EventStreamJsonMessage request) {
                System.out.println("Handling request...");
                return requestHandler.apply(request);
            }

            @Override
            public void handleStreamEvent(EventStreamJsonMessage streamRequestEvent) {
                System.out.println("Handling request stream event...");
            }
        };
        continuationHandlerQueue.offer(new AbstractMap.SimpleImmutableEntry<>(operationName, handler));
        return handler;
    };
}