Update Handler.scala

Fix #2372
zio · Sep 22, 2023 · 5159fbc · 5159fbc
1 parent 8780eb9
commit 5159fbc
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/zio-http/src/main/scala/zio/http/Handler.scala b/zio-http/src/main/scala/zio/http/Handler.scala
@@ -660,6 +660,16 @@ sealed trait Handler[-R, +Err, -In, +Out] { self =>
 
 object Handler {
 
+  def asStreamBounded(request: Request, limit: Int): Handler[Any, Throwable, Any, Chunk[Byte]] =
+    Handler.fromZIO(
+      request.body.asStream
+        .chunks
+        .runFoldZIO(Chunk.empty[Byte])((acc, bytes) =>
+          ZIO.succeed(acc ++ bytes)
+            .filterOrFail(_.sizeIs < limit)(new Exception("Too large input"))
+        )
+    )
+
   /**
    * Attempts to create a Handler that succeeds with the provided value,
    * capturing all exceptions on it's way.