small fixes

p-zalejko · Oct 11, 2019 · 6b97497 · 6b97497
1 parent e9fe16d
commit 6b97497
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/kotlin/com/gmail/pzalejko/myplanner/event/EventController.kt b/src/main/kotlin/com/gmail/pzalejko/myplanner/event/EventController.kt
@@ -2,11 +2,11 @@ package com.gmail.pzalejko.myplanner.event;
 
 import com.mongodb.client.MongoClient
 import io.vertx.axle.core.Vertx
-import java.util.concurrent.CompletableFuture
 import java.util.concurrent.CompletionStage
 import javax.inject.Inject
 import javax.ws.rs.*
 import javax.ws.rs.core.MediaType
+import javax.ws.rs.core.Response
 
 @Path("/events")
 @Produces(MediaType.APPLICATION_JSON)
@@ -28,10 +28,12 @@ class EventController {
     }
 
     @POST
-    fun addNew(name: EventDto): CompletionStage<String> {
+    fun addNew(name: EventDto): CompletionStage<Response> {
         val event = Event(name = name.name)
         val addNew = service.addNew(event)
-        return addNew.thenCompose { d -> CompletableFuture.completedFuture("OK") }
+        return addNew
+                .thenApply(Response::ok)
+                .thenApply(Response.ResponseBuilder::build);
     }
 
     class EventDto {