Search in sources :

Example 1 with RegisterStreamConsumerRequest

use of software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerRequest in project aws-doc-sdk-examples by awsdocs.

the class RegisterStreamConsumer method regConsumer.

public static String regConsumer(KinesisClient kinesisClient, String streamARN) {
    try {
        RegisterStreamConsumerRequest regCon = RegisterStreamConsumerRequest.builder().consumerName("MyConsumer").streamARN(streamARN).build();
        RegisterStreamConsumerResponse resp = kinesisClient.registerStreamConsumer(regCon);
        return resp.consumer().consumerARN();
    } catch (KinesisException e) {
        System.err.println(e.getMessage());
        System.exit(1);
    }
    return "";
}
Also used : RegisterStreamConsumerRequest(software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerRequest) KinesisException(software.amazon.awssdk.services.kinesis.model.KinesisException) RegisterStreamConsumerResponse(software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerResponse)

Example 2 with RegisterStreamConsumerRequest

use of software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerRequest in project flink by apache.

the class KinesisProxyV2Test method testRegisterStreamConsumer.

@Test
public void testRegisterStreamConsumer() throws Exception {
    KinesisAsyncClient client = mock(KinesisAsyncClient.class);
    KinesisProxyV2 proxy = new KinesisProxyV2(client, mock(SdkAsyncHttpClient.class), createConfiguration(), mock(FullJitterBackoff.class));
    RegisterStreamConsumerResponse expected = RegisterStreamConsumerResponse.builder().build();
    ArgumentCaptor<RegisterStreamConsumerRequest> requestCaptor = ArgumentCaptor.forClass(RegisterStreamConsumerRequest.class);
    when(client.registerStreamConsumer(requestCaptor.capture())).thenReturn(CompletableFuture.completedFuture(expected));
    RegisterStreamConsumerResponse actual = proxy.registerStreamConsumer("arn", "name");
    assertEquals(expected, actual);
    RegisterStreamConsumerRequest request = requestCaptor.getValue();
    assertEquals("arn", request.streamARN());
    assertEquals("name", request.consumerName());
}
Also used : KinesisAsyncClient(software.amazon.awssdk.services.kinesis.KinesisAsyncClient) RegisterStreamConsumerRequest(software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerRequest) SdkAsyncHttpClient(software.amazon.awssdk.http.async.SdkAsyncHttpClient) RegisterStreamConsumerResponse(software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerResponse) Test(org.junit.Test)

Aggregations

RegisterStreamConsumerRequest (software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerRequest)2 RegisterStreamConsumerResponse (software.amazon.awssdk.services.kinesis.model.RegisterStreamConsumerResponse)2 Test (org.junit.Test)1 SdkAsyncHttpClient (software.amazon.awssdk.http.async.SdkAsyncHttpClient)1 KinesisAsyncClient (software.amazon.awssdk.services.kinesis.KinesisAsyncClient)1 KinesisException (software.amazon.awssdk.services.kinesis.model.KinesisException)1