an example for zio/zio-http#2836

stanislav-chetvertkov · May 9, 2024 · beb943c · beb943c
1 parent d5c107a
commit beb943c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/main/resources/petstore-minimal.json b/src/main/resources/petstore-minimal.json
@@ -126,6 +126,16 @@
               }
             }
           },
+          "404": {
+            "description": "Not Found",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/Error"
+                }
+              }
+            }
+          },
           "default": {
             "description": "unexpected error",
             "content": {

diff --git a/src/main/scala/openapi/generator/example/ServerExample.scala b/src/main/scala/openapi/generator/example/ServerExample.scala
@@ -4,13 +4,16 @@ import base.Pets
 import base.component.Pet
 import zio.http.endpoint.Endpoint
 import zio.http.endpoint.EndpointMiddleware.None
-import zio.http.{Handler, Route, Server}
+import zio.http.{Handler, Response, Route, Server, Status}
 import zio.{Scope, ZIO, ZIOAppArgs, ZIOAppDefault, ZNothing}
 
 object ServerExample extends ZIOAppDefault {
 
   val endpoint: Endpoint[Unit, Pet, ZNothing, Unit, None] = Pets.createPets
-  val route: Route[Any, Nothing] = endpoint.implement(Handler.fromFunctionZIO(pet => {ZIO.logInfo("Pet: " + pet.toString)}))
+  val route: Route[Any, Nothing] = endpoint.implement(
+    Handler.fromFunctionZIO(pet => {ZIO.logInfo("Pet: " + pet.toString)})
+    .mapError(e => Response.text("Error: " + e.getMessage).status(Status.InternalServerError))
+  )
 
   // create zio http app from the generated endpoint
   val httpApp = route.toHttpApp