Added webflux methods on the new TaskService.

leangen · Jan 4, 2019 · eab165c · eab165c
1 parent 9ceec7f
commit eab165c
Showing 1 changed file with 11 additions and 0 deletions.
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
@@ -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);