implement available. (Azure#27907)

AzureSDKAutomation · Mar 28, 2022 · 8218284 · 8218284
1 parent d108c7d
commit 8218284
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/common/perf-test-core/src/main/java/com/azure/perf/test/core/RepeatingInputStream.java b/common/perf-test-core/src/main/java/com/azure/perf/test/core/RepeatingInputStream.java
@@ -3,6 +3,7 @@
 
 package com.azure.perf.test.core;
 
+import java.io.IOException;
 import java.io.InputStream;
 import java.util.Random;
 
@@ -84,5 +85,15 @@ public boolean markSupported() {
     public synchronized void reset() {
         this.pos = this.mark;
     }
+
+    @Override
+    public int available() throws IOException {
+        long remaining = this.size - this.pos;
+        if (remaining > Integer.MAX_VALUE) {
+            return Integer.MAX_VALUE;
+        } else {
+            return (int) remaining;
+        }
+    }
 }