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 "";
}
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());
}
Aggregations