Optimise zio.http.endpoint.Endpoint.implementHandler (#3236)

Replace `.map(..).catchAll(..)` with `.foldZIO` Co-authored-by: Nabil Abdel-Hafeez <[email protected]>
zio · Dec 14, 2024 · 7db7f7f · 7db7f7f
1 parent 869a230
commit 7db7f7f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/zio-http/shared/src/main/scala/zio/http/endpoint/Endpoint.scala b/zio-http/shared/src/main/scala/zio/http/endpoint/Endpoint.scala
@@ -303,9 +303,10 @@ final case class Endpoint[PathInput, Input, Err, Output, Auth <: AuthType](
               )
               .getOrElse(defaultMediaTypes)
           (endpoint.input ++ authCodec(endpoint.authType)).decodeRequest(request, config).orDie.flatMap { value =>
-            original(value).map(endpoint.output.encodeResponse(_, outputMediaTypes, config)).catchAll { error =>
-              ZIO.succeed(endpoint.error.encodeResponse(error, outputMediaTypes, config))
-            }
+            original(value).foldZIO(
+              success = output => Exit.succeed(endpoint.output.encodeResponse(output, outputMediaTypes, config)),
+              failure = error => Exit.succeed(endpoint.error.encodeResponse(error, outputMediaTypes, config)),
+            )
           }
         } -> condition
       }