[java] Return new input stream to allow multiple reads (#11249)

java/src/org/openqa/selenium/remote/ProtocolHandshake.java

@@ -34,7 +34,6 @@
 import org.openqa.selenium.remote.http.HttpRequest;
 import org.openqa.selenium.remote.http.HttpResponse;
 
-import java.io.BufferedInputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStreamWriter;
@@ -43,6 +42,7 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.function.Function;
+import java.util.function.Supplier;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 import java.util.stream.Collectors;
@@ -91,21 +91,23 @@ public Result createSession(HttpHandler client, Command command) throws IOExcept
 
   public Either<SessionNotCreatedException, Result> createSession(HttpHandler client, NewSessionPayload payload) throws IOException {
     int threshold = (int) Math.min(Runtime.getRuntime().freeMemory() / 10, Integer.MAX_VALUE);
-    FileBackedOutputStream os = new FileBackedOutputStream(threshold);
+    FileBackedOutputStream os = new FileBackedOutputStream(threshold, true);
 
     try (CountingOutputStream counter = new CountingOutputStream(os);
          Writer writer = new OutputStreamWriter(counter, UTF_8)) {
       payload.writeTo(writer);
-
-      try (InputStream contentStream = os.asByteSource().openBufferedStream()) {
-        return createSession(client, contentStream, counter.getCount());
-      }
-    } finally {
-      os.reset();
+      Supplier<InputStream> contentSupplier = () -> {
+        try {
+          return os.asByteSource().openBufferedStream();
+        } catch (IOException e) {
+          throw new RuntimeException(e);
+        }
+      };
+      return createSession(client, contentSupplier, counter.getCount());
     }
   }
 
-  private Either<SessionNotCreatedException, Result> createSession(HttpHandler client, InputStream newSessionBlob, long size) {
+  private Either<SessionNotCreatedException, Result> createSession(HttpHandler client, Supplier<InputStream> contentSupplier, long size) {
     // Create the http request and send it
     HttpRequest request = new HttpRequest(HttpMethod.POST, "/session");
 
@@ -117,7 +119,7 @@ private Either<SessionNotCreatedException, Result> createSession(HttpHandler cli
     // session e.g. with profiles.
     request.setHeader(CONTENT_LENGTH, String.valueOf(size));
     request.setHeader(CONTENT_TYPE, JSON_UTF_8);
-    request.setContent(() -> newSessionBlob);
+    request.setContent(contentSupplier);
 
     response = client.execute(request);
     long time = System.currentTimeMillis() - start;