Skip to content

Commit 2a54f6b

Browse files
committed
Rename kinesis metrics config to kcl_metrics_enabled and default to true
Signed-off-by: RashmiRam <ras.xena@gmail.com>
1 parent 496230d commit 2a54f6b

File tree

5 files changed

+9
-9
lines changed

5 files changed

+9
-9
lines changed

data-prepper-plugins/kinesis-source/src/main/java/org/opensearch/dataprepper/plugins/kinesis/source/KinesisService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ public Scheduler createScheduler(final Buffer<Record<Event>> buffer) {
200200
kinesisSourceConfig.getPollingConfig().getIdleTimeBetweenReads().toMillis()));
201201
}
202202

203-
MetricsConfig metricsConfig = kinesisSourceConfig.isMetrics()
203+
MetricsConfig metricsConfig = kinesisSourceConfig.isKclMetricsEnabled()
204204
? configsBuilder.metricsConfig()
205205
: configsBuilder.metricsConfig().metricsFactory(new NullMetricsFactory());
206206

data-prepper-plugins/kinesis-source/src/main/java/org/opensearch/dataprepper/plugins/kinesis/source/configuration/KinesisSourceConfig.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,8 @@ public Duration getShardAcknowledgmentTimeout() {
8181
private Duration initializationBackoffTime = DEFAULT_INITIALIZATION_BACKOFF_TIME;
8282

8383
@Getter
84-
@JsonProperty("metrics")
85-
private boolean metrics = false;
84+
@JsonProperty("kcl_metrics_enabled")
85+
private boolean kclMetricsEnabled = true;
8686
}
8787

8888

data-prepper-plugins/kinesis-source/src/test/java/org/opensearch/dataprepper/plugins/kinesis/source/KinesisServiceTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ void testCreateScheduler() {
262262

263263
@Test
264264
void testCreateSchedulerUsesNullMetricsFactoryWhenMetricsDisabled() {
265-
when(kinesisSourceConfig.isMetrics()).thenReturn(false);
265+
when(kinesisSourceConfig.isKclMetricsEnabled()).thenReturn(false);
266266
KinesisService kinesisService = new KinesisService(kinesisSourceConfig, kinesisClientFactory, pluginMetrics, pluginFactory,
267267
pipelineDescription, acknowledgementSetManager, kinesisLeaseConfigSupplier, workerIdentifierGenerator);
268268
Scheduler schedulerObjectUnderTest = kinesisService.createScheduler(buffer);
@@ -274,7 +274,7 @@ void testCreateSchedulerUsesNullMetricsFactoryWhenMetricsDisabled() {
274274

275275
@Test
276276
void testCreateSchedulerUsesCloudWatchMetricsFactoryWhenMetricsEnabled() {
277-
when(kinesisSourceConfig.isMetrics()).thenReturn(true);
277+
when(kinesisSourceConfig.isKclMetricsEnabled()).thenReturn(true);
278278
KinesisService kinesisService = new KinesisService(kinesisSourceConfig, kinesisClientFactory, pluginMetrics, pluginFactory,
279279
pipelineDescription, acknowledgementSetManager, kinesisLeaseConfigSupplier, workerIdentifierGenerator);
280280
Scheduler schedulerObjectUnderTest = kinesisService.createScheduler(buffer);

data-prepper-plugins/kinesis-source/src/test/java/org/opensearch/dataprepper/plugins/kinesis/source/configuration/KinesisSourceConfigTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ void testSourceConfig() {
8282
assertEquals(KinesisSourceConfig.DEFAULT_MAX_INITIALIZATION_ATTEMPTS, kinesisSourceConfig.getMaxInitializationAttempts());
8383
assertEquals(KinesisSourceConfig.DEFAULT_INITIALIZATION_BACKOFF_TIME, kinesisSourceConfig.getInitializationBackoffTime());
8484
assertTrue(kinesisSourceConfig.isAcknowledgments());
85-
assertFalse(kinesisSourceConfig.isMetrics());
85+
assertTrue(kinesisSourceConfig.isKclMetricsEnabled());
8686
assertEquals(KinesisSourceConfig.DEFAULT_SHARD_ACKNOWLEDGEMENT_TIMEOUT, kinesisSourceConfig.getShardAcknowledgmentTimeout());
8787
assertThat(kinesisSourceConfig.getAwsAuthenticationConfig(), notNullValue());
8888
assertEquals(kinesisSourceConfig.getAwsAuthenticationConfig().getAwsRegion(), Region.US_EAST_1);
@@ -116,7 +116,7 @@ void testSourceConfigWithStreamCodec() {
116116
assertEquals(KinesisSourceConfig.DEFAULT_MAX_INITIALIZATION_ATTEMPTS, kinesisSourceConfig.getMaxInitializationAttempts());
117117
assertEquals(KinesisSourceConfig.DEFAULT_INITIALIZATION_BACKOFF_TIME, kinesisSourceConfig.getInitializationBackoffTime());
118118
assertFalse(kinesisSourceConfig.isAcknowledgments());
119-
assertFalse(kinesisSourceConfig.isMetrics());
119+
assertTrue(kinesisSourceConfig.isKclMetricsEnabled());
120120
assertEquals(KinesisSourceConfig.DEFAULT_SHARD_ACKNOWLEDGEMENT_TIMEOUT, kinesisSourceConfig.getShardAcknowledgmentTimeout());
121121
assertThat(kinesisSourceConfig.getAwsAuthenticationConfig(), notNullValue());
122122
assertEquals(kinesisSourceConfig.getAwsAuthenticationConfig().getAwsRegion(), Region.US_EAST_1);
@@ -243,7 +243,7 @@ void testSourceConfigWithStreamArnConsumerArn() {
243243
void testSourceConfigWithMetricsEnabled() {
244244

245245
assertThat(kinesisSourceConfig, notNullValue());
246-
assertTrue(kinesisSourceConfig.isMetrics());
246+
assertTrue(kinesisSourceConfig.isKclMetricsEnabled());
247247
assertEquals(KinesisSourceConfig.DEFAULT_NUMBER_OF_RECORDS_TO_ACCUMULATE, kinesisSourceConfig.getNumberOfRecordsToAccumulate());
248248
assertEquals(KinesisSourceConfig.DEFAULT_TIME_OUT_IN_MILLIS, kinesisSourceConfig.getBufferTimeout());
249249
assertFalse(kinesisSourceConfig.isAcknowledgments());

data-prepper-plugins/kinesis-source/src/test/resources/pipeline_with_metrics_enabled.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,5 @@ source:
99
aws:
1010
sts_role_arn: "arn:aws:iam::123456789012:role/OSI-PipelineRole"
1111
region: "us-east-1"
12-
metrics: true
12+
kcl_metrics_enabled: true
1313

0 commit comments

Comments
 (0)