Update isOneShot check and reset stream if possible

microsoft · Aug 22, 2024 · e4dd84e · e4dd84e
1 parent ed319bb
commit e4dd84e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/components/http/okHttp/src/main/java/com/microsoft/kiota/http/OkHttpRequestAdapter.java b/components/http/okHttp/src/main/java/com/microsoft/kiota/http/OkHttpRequestAdapter.java
@@ -901,7 +901,7 @@ public MediaType contentType() {
 
                                 @Override
                                 public boolean isOneShot() {
-                                    return true;
+                                    return !requestInfo.content.markSupported();
                                 }
 
                                 @Override
@@ -928,6 +928,9 @@ public long contentLength() throws IOException {
                                 @Override
                                 public void writeTo(@Nonnull BufferedSink sink) throws IOException {
                                     sink.writeAll(Okio.source(requestInfo.content));
+                                    if (!isOneShot()) {
+                                        requestInfo.content.reset();
+                                    }
                                 }
                             };