Fix tests

Signed-off-by: Kondaka <[email protected]>
kkondaka · Nov 17, 2024 · cd72a82 · cd72a82
1 parent 8fe70b9
commit cd72a82
Show file tree

Hide file tree

Showing 2 changed files with 240 additions and 270 deletions.
diff --git a/...da/src/main/java/org/opensearch/dataprepper/plugins/lambda/processor/LambdaProcessor.java b/...da/src/main/java/org/opensearch/dataprepper/plugins/lambda/processor/LambdaProcessor.java
@@ -142,7 +142,8 @@ public Collection<Record<Event>> doExecute(Collection<Record<Event>> records) {
             }
             recordsToLambda.add(record);
         }
-        resultRecords.addAll(
+        try {
+            resultRecords.addAll(
                 lambdaCommonHandler.sendRecords(recordsToLambda, lambdaProcessorConfig, lambdaAsyncClient,
                         (inputBuffer, response) -> {
                             Duration latency = inputBuffer.stopLatencyWatch();
@@ -158,7 +159,12 @@ public Collection<Record<Event>> doExecute(Collection<Record<Event>> records) {
                             numberOfRequestsFailedCounter.increment();
                             addFailureTags(inputBuffer, outputRecords);
                         })
-        );
+
+            );
+        } catch (Exception e) {
+            numberOfRecordsFailedCounter.increment(recordsToLambda.size());
+            resultRecords.addAll(recordsToLambda);
+        }
         return resultRecords;
     }