diff --git a/data-prepper-plugins/cloudwatch-logs/src/main/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/CloudWatchLogsDispatcher.java b/data-prepper-plugins/cloudwatch-logs/src/main/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/CloudWatchLogsDispatcher.java index 102b511f14..c9693fdc8f 100644 --- a/data-prepper-plugins/cloudwatch-logs/src/main/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/CloudWatchLogsDispatcher.java +++ b/data-prepper-plugins/cloudwatch-logs/src/main/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/CloudWatchLogsDispatcher.java @@ -86,6 +86,7 @@ public void dispatchLogs(List inputLogEvents, Collection eventHandles; + private final int totalEventCount; private final int retryCount; private final long backOffTimeBase; @@ -132,10 +134,10 @@ public void upload() { if (failedToTransmit) { - cloudWatchLogsMetrics.increaseLogEventFailCounter(eventHandles.size()); + cloudWatchLogsMetrics.increaseLogEventFailCounter(totalEventCount); releaseEventHandles(false, eventHandles); } else { - cloudWatchLogsMetrics.increaseLogEventSuccessCounter(eventHandles.size()); + cloudWatchLogsMetrics.increaseLogEventSuccessCounter(totalEventCount); releaseEventHandles(true, eventHandles); } } diff --git a/data-prepper-plugins/cloudwatch-logs/src/test/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/UploaderTest.java b/data-prepper-plugins/cloudwatch-logs/src/test/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/UploaderTest.java index 19629b2952..83cc9fa98a 100644 --- a/data-prepper-plugins/cloudwatch-logs/src/test/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/UploaderTest.java +++ b/data-prepper-plugins/cloudwatch-logs/src/test/java/org/opensearch/dataprepper/plugins/sink/cloudwatch_logs/client/UploaderTest.java @@ -49,6 +49,7 @@ CloudWatchLogsDispatcher.Uploader getUploader() { .cloudWatchLogsMetrics(mockCloudWatchLogsMetrics) .putLogEventsRequest(getMockPutLogEventsRequest()) .eventHandles(getTestEventHandles()) + .totalEventCount(ThresholdConfig.DEFAULT_BATCH_SIZE) .retryCount(ThresholdConfig.DEFAULT_RETRY_COUNT) .backOffTimeBase(ThresholdConfig.DEFAULT_BACKOFF_TIME) .build();