Added payload metrics to sink

Signed-off-by: Kondaka <[email protected]>
opensearch-project · Nov 18, 2024 · 0d1cbb1 · 0d1cbb1
1 parent dd0c3a4
commit 0d1cbb1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...s/aws-lambda/src/main/java/org/opensearch/dataprepper/plugins/lambda/sink/LambdaSink.java b/...s/aws-lambda/src/main/java/org/opensearch/dataprepper/plugins/lambda/sink/LambdaSink.java
@@ -161,6 +161,8 @@ public void doOutput(final Collection<Record<Event>> records) {
           lambdaLatencyMetric.record(latency.toMillis(), TimeUnit.MILLISECONDS);
           numberOfRecordsSuccessCounter.increment(inputBuffer.getEventCount());
           numberOfRequestsSuccessCounter.increment();
+          requestPayloadMetric.record(inputBuffer.getPayloadRequestSize());
+          responsePayloadMetric.record(invokeResponse.payload().asByteArray().length);
           releaseEventHandlesPerBatch(true, inputBuffer);
           return new ArrayList<>();
         },
@@ -169,6 +171,7 @@ public void doOutput(final Collection<Record<Event>> records) {
           lambdaLatencyMetric.record(latency.toMillis(), TimeUnit.MILLISECONDS);
           numberOfRecordsFailedCounter.increment(inputBuffer.getEventCount());
           numberOfRequestsFailedCounter.increment();
+          requestPayloadMetric.record(inputBuffer.getPayloadRequestSize());
           handleFailure(new RuntimeException("failed"), inputBuffer);
           return new ArrayList<>();
         });