From eab165c64be7a07285c9630eb89eb3b2b645506f Mon Sep 17 00:00:00 2001
From: Robert  Dinu <vjroby@gmail.com>
Date: Fri, 4 Jan 2019 16:56:44 +0100
Subject: [PATCH] Added webflux methods on the new TaskService.

---
 .../leangen/graphql/samples/service/TaskService.java  | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/spring-boot-starter-sample/src/main/java/io/leangen/graphql/samples/service/TaskService.java b/spring-boot-starter-sample/src/main/java/io/leangen/graphql/samples/service/TaskService.java
index fe1f661..df48fa6 100644
--- a/spring-boot-starter-sample/src/main/java/io/leangen/graphql/samples/service/TaskService.java
+++ b/spring-boot-starter-sample/src/main/java/io/leangen/graphql/samples/service/TaskService.java
@@ -15,6 +15,7 @@
 import org.springframework.stereotype.Service;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.FluxSink;
+import reactor.core.publisher.Mono;
 
 import java.util.Collection;
 
@@ -51,11 +52,21 @@ public Collection<Task> tasks(String projectCode, Status... statuses) {
         return repo.byProjectCodeAndStatus(projectCode, statuses);
     }
 
+    @GraphQLQuery
+    public Flux<Task> tasksWebflux(String projectCode, Status... statuses) {
+        return Flux.fromIterable(repo.byProjectCodeAndStatus(projectCode, statuses));
+    }
+
     @GraphQLQuery
     public Task task(String code) {
         return repo.byCode(code);
     }
 
+    @GraphQLQuery
+    public Mono<Task> taskWebfux(String code) {
+        return Mono.just(repo.byCode(code));
+    }
+
     @GraphQLSubscription
     public Publisher<Task> taskStatusChanged(String code) {
         return Flux.create(subscriber -> subscribers.add(code, subscriber.onDispose(() -> subscribers.remove(code, subscriber))), FluxSink.OverflowStrategy.LATEST);