From 1a2e88c656aeb02aa1b4e775ae3e840b944fdaed Mon Sep 17 00:00:00 2001
From: Krishna Kondaka <krishkdk@amazon.com>
Date: Fri, 7 Feb 2025 19:50:20 +0000
Subject: [PATCH] Addressed review comments

Signed-off-by: Krishna Kondaka <krishkdk@amazon.com>
---
 data-prepper-plugins/otel-proto-common/build.gradle        | 2 +-
 .../plugins/otel/codec/OTelLogsProtoBufDecoder.java        | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/data-prepper-plugins/otel-proto-common/build.gradle b/data-prepper-plugins/otel-proto-common/build.gradle
index 7c27e0fbd0..40636b049c 100644
--- a/data-prepper-plugins/otel-proto-common/build.gradle
+++ b/data-prepper-plugins/otel-proto-common/build.gradle
@@ -8,7 +8,7 @@ plugins {
 }
 
 test {
-    jvmArgs '-Xmx4g'
+    jvmArgs '-Xmx512m'
 }
 
 dependencies {
diff --git a/data-prepper-plugins/otel-proto-common/src/main/java/org/opensearch/dataprepper/plugins/otel/codec/OTelLogsProtoBufDecoder.java b/data-prepper-plugins/otel-proto-common/src/main/java/org/opensearch/dataprepper/plugins/otel/codec/OTelLogsProtoBufDecoder.java
index 30195b02da..0bd691b994 100644
--- a/data-prepper-plugins/otel-proto-common/src/main/java/org/opensearch/dataprepper/plugins/otel/codec/OTelLogsProtoBufDecoder.java
+++ b/data-prepper-plugins/otel-proto-common/src/main/java/org/opensearch/dataprepper/plugins/otel/codec/OTelLogsProtoBufDecoder.java
@@ -50,8 +50,9 @@ public void parse(InputStream inputStream, Instant timeReceivedMs, Consumer<Reco
         // Each request is written in a separate S3 object, so, no legth preceeding the actual data
         // Same with Kafka exporter too. Each message is written as a separate message to Kafka
         if (!lengthPrefixedEncoding) {
-            if (inputStream.available() > MAX_REQUEST_LEN) {
-                throw new IOException("buffer length exceeds max allowed buffer length of "+ MAX_REQUEST_LEN);
+            int available = inputStream.available();
+            if (available > MAX_REQUEST_LEN) {
+                throw new IOException("buffer length " + available + " exceeds max allowed buffer length of "+ MAX_REQUEST_LEN);
             }
             byte[] buffer = inputStream.readAllBytes();
             parseRequest(buffer, timeReceivedMs, eventConsumer);
@@ -65,7 +66,7 @@ public void parse(InputStream inputStream, Instant timeReceivedMs, Consumer<Reco
             ByteBuffer lengthBuffer = ByteBuffer.wrap(lenBytes);
             int len = lengthBuffer.getInt();
             if (len > MAX_REQUEST_LEN) {
-                throw new IOException("buffer length exceeds max allowed buffer length of "+ MAX_REQUEST_LEN);
+                throw new IOException("buffer length " + len + " exceeds max allowed buffer length of "+ MAX_REQUEST_LEN);
             }
             byte[] buffer = new byte[len];
             if (inputStream.read(buffer, 0, len) != len) {