Make the chunk size used to convert InputStreams to ZIO response bodies configurable (#4780)

Closes #4762

Author: adamw

server/zio-http-server/src/main/scala/sttp/tapir/server/ziohttp/ZioHttpInterpreter.scala

@@ -28,7 +28,7 @@ trait ZioHttpInterpreter[R] {
     val widenedSes = ses.map(_.widen[R & R2])
     val widenedServerOptions = zioHttpServerOptions.widen[R & R2]
     val zioHttpRequestBody = new ZioHttpRequestBody(widenedServerOptions)
-    val zioHttpResponseBody = new ZioHttpToResponseBody
+    val zioHttpResponseBody = new ZioHttpToResponseBody(zioHttpServerOptions.inputStreamChunkSize)
     val interceptors = RejectInterceptor.disableWhenSingleEndpoint(widenedServerOptions.interceptors, widenedSes)
 
     def handleRequest(req: Request, filteredEndpoints: List[ZServerEndpoint[R & R2, ZioStreams with WebSockets]]) =

server/zio-http-server/src/main/scala/sttp/tapir/server/ziohttp/ZioHttpServerOptions.scala

@@ -6,12 +6,15 @@ import sttp.tapir.server.interceptor.{CustomiseInterceptors, Interceptor}
 import sttp.tapir.{Defaults, TapirFile}
 import zio.http.WebSocketConfig
 import zio.{Cause, RIO, Task, ZIO}
+import zio.stream.ZStream
 
 case class ZioHttpServerOptions[R](
     createFile: ServerRequest => Task[TapirFile],
     deleteFile: TapirFile => RIO[R, Unit],
     interceptors: List[Interceptor[RIO[R, *]]],
-    customWebSocketConfig: ServerRequest => Option[WebSocketConfig]
+    customWebSocketConfig: ServerRequest => Option[WebSocketConfig],
+    /** #4762: the chunk size used when converting an input stream to a response body. */
+    inputStreamChunkSize: Int = ZStream.DefaultChunkSize
 ) {
   def prependInterceptor(i: Interceptor[RIO[R, *]]): ZioHttpServerOptions[R] =
     copy(interceptors = i :: interceptors)

server/zio-http-server/src/main/scala/sttp/tapir/server/ziohttp/ZioHttpToResponseBody.scala

@@ -14,7 +14,7 @@ import java.io.InputStream
 import java.nio.ByteBuffer
 import java.nio.charset.Charset
 
-class ZioHttpToResponseBody extends ToResponseBody[ZioResponseBody, ZioStreams] {
+class ZioHttpToResponseBody(inputStreamChunkSize: Int) extends ToResponseBody[ZioResponseBody, ZioStreams] {
   override val streams: ZioStreams = ZioStreams
 
   override def fromRawValue[R](
@@ -49,16 +49,16 @@ class ZioHttpToResponseBody extends ToResponseBody[ZioResponseBody, ZioStreams]
         val buffer: ByteBuffer = r
         ZioRawHttpResponseBody(Chunk.fromByteBuffer(buffer), Some(buffer.remaining()))
       case RawBodyType.InputStreamBody =>
-        ZioStreamHttpResponseBody(ZStream.fromInputStream(r), None)
+        ZioStreamHttpResponseBody(ZStream.fromInputStream(r, inputStreamChunkSize), None)
       case RawBodyType.InputStreamRangeBody =>
         r.range
           .map(range =>
             ZioStreamHttpResponseBody(
-              ZStream.fromInputStream(r.inputStreamFromRangeStart()).take(range.contentLength),
+              ZStream.fromInputStream(r.inputStreamFromRangeStart(), inputStreamChunkSize).take(range.contentLength),
               Some(range.contentLength)
             )
           )
-          .getOrElse(ZioStreamHttpResponseBody(ZStream.fromInputStream(r.inputStream()), None))
+          .getOrElse(ZioStreamHttpResponseBody(ZStream.fromInputStream(r.inputStream(), inputStreamChunkSize), None))
       case RawBodyType.FileBody =>
         val tapirFile = r
         tapirFile.range
@@ -118,14 +118,14 @@ class ZioHttpToResponseBody extends ToResponseBody[ZioResponseBody, ZioStreams]
       case RawBodyType.InputStreamBody =>
         FormField.streamingBinaryField(
           part.name,
-          ZStream.fromInputStream(part.body).orDie,
+          ZStream.fromInputStream(part.body, inputStreamChunkSize).orDie,
           mediaType.getOrElse(MediaType.application.`octet-stream`),
           filename = part.fileName
         )
       case RawBodyType.InputStreamRangeBody =>
         FormField.streamingBinaryField(
           part.name,
-          ZStream.fromInputStream(part.body.inputStream()).orDie,
+          ZStream.fromInputStream(part.body.inputStream(), inputStreamChunkSize).orDie,
           mediaType.getOrElse(MediaType.application.`octet-stream`),
           filename = part.fileName
         )