Add available() method to the streaming wrappers

mulesoft · Dec 18, 2024 · 021d81f · 021d81f
1 parent 8b485bf
commit 021d81f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/...c/main/java/org/mule/runtime/core/internal/streaming/bytes/AbstractInputStreamBuffer.java b/...c/main/java/org/mule/runtime/core/internal/streaming/bytes/AbstractInputStreamBuffer.java
@@ -47,6 +47,11 @@ public AbstractInputStreamBuffer(InputStream stream, ByteBufferManager bufferMan
     this.bufferManager = bufferManager;
   }
 
+  @Override
+  public int available() throws IOException {
+    return stream.available();
+  }
+
   /**
    * Consumes the stream in order to obtain data that has not been read yet.
    *

diff --git a/core/src/main/java/org/mule/runtime/core/internal/streaming/bytes/BufferedCursorStream.java b/core/src/main/java/org/mule/runtime/core/internal/streaming/bytes/BufferedCursorStream.java
@@ -117,6 +117,11 @@ protected int doRead(byte[] b, int off, int len) throws IOException {
     }
   }
 
+  @Override
+  public int available() throws IOException {
+    return localBuffer.remaining() + streamBuffer.available();
+  }
+
   private int assureDataInLocalBuffer(int len) {
     if (len <= localBuffer.remaining()) {
       return toIntExact(len);

diff --git a/core/src/main/java/org/mule/runtime/core/internal/streaming/bytes/InputStreamBuffer.java b/core/src/main/java/org/mule/runtime/core/internal/streaming/bytes/InputStreamBuffer.java
@@ -6,6 +6,7 @@
  */
 package org.mule.runtime.core.internal.streaming.bytes;
 
+import java.io.IOException;
 import java.nio.ByteBuffer;
 
 /**
@@ -36,4 +37,6 @@ public interface InputStreamBuffer {
    * Releases all the resources held by this buffer
    */
   void close();
+
+  int available() throws IOException;
 }