|
47 | 47 | import software.amazon.awssdk.services.kinesis.KinesisAsyncClient; |
48 | 48 | import software.amazon.kinesis.common.InitialPositionInStream; |
49 | 49 | import software.amazon.kinesis.common.InitialPositionInStreamExtended; |
| 50 | +import software.amazon.kinesis.metrics.MetricsLevel; |
50 | 51 | import software.amazon.kinesis.producer.KinesisProducer; |
51 | 52 | import software.amazon.kinesis.producer.KinesisProducerConfiguration; |
52 | 53 |
|
@@ -136,12 +137,15 @@ public KinesisProducerConfiguration kinesisProducerConfiguration() { |
136 | 137 | .getEndpointOverride(LocalStackContainer.Service.KINESIS); |
137 | 138 | URI cloudWatchUri = LocalstackContainerTest.LOCAL_STACK_CONTAINER |
138 | 139 | .getEndpointOverride(LocalStackContainer.Service.CLOUDWATCH); |
| 140 | + URI stsUri = LocalstackContainerTest.LOCAL_STACK_CONTAINER |
| 141 | + .getEndpointOverride(LocalStackContainer.Service.STS); |
139 | 142 |
|
140 | 143 | return new KinesisProducerConfiguration() |
141 | 144 | .setCredentialsProvider(LocalstackContainerTest.credentialsProvider()) |
142 | 145 | .setRegion(LocalstackContainerTest.LOCAL_STACK_CONTAINER.getRegion()) |
143 | 146 | .setKinesisEndpoint(kinesisUri.getHost()).setKinesisPort(kinesisUri.getPort()) |
144 | 147 | .setCloudwatchEndpoint(cloudWatchUri.getHost()).setCloudwatchPort(cloudWatchUri.getPort()) |
| 148 | + .setStsEndpoint(stsUri.getHost()).setStsPort(stsUri.getPort()) |
145 | 149 | .setVerifyCertificate(false); |
146 | 150 | } |
147 | 151 |
|
@@ -172,6 +176,9 @@ public KclMessageDrivenChannelAdapter kclMessageDrivenChannelAdapter(PollableCha |
172 | 176 | adapter.setStreamInitialSequence( |
173 | 177 | InitialPositionInStreamExtended.newInitialPosition(InitialPositionInStream.TRIM_HORIZON)); |
174 | 178 | adapter.setBindSourceRecord(true); |
| 179 | + adapter.setMetricsLevel(MetricsLevel.NONE); |
| 180 | + adapter.setLeaseManagementConfigCustomizer(leaseManagementConfig -> leaseManagementConfig |
| 181 | + .workerUtilizationAwareAssignmentConfig().disableWorkerMetrics(true)); |
175 | 182 | return adapter; |
176 | 183 | } |
177 | 184 |
|
|
0 commit comments