diff --git a/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/DataRepository.java b/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/DataRepository.java index ddf24816..49a6c182 100644 --- a/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/DataRepository.java +++ b/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/DataRepository.java @@ -29,25 +29,25 @@ import org.springframework.stereotype.Repository; @Repository public class DataRepository { - public String getBasic(DataFetchingEnvironment environment) { + public String getBasic() { return "Hello world!"; } - public Mono getGreeting(DataFetchingEnvironment environment) { + public Mono getGreeting() { return Mono.deferContextual(context -> { Object name = context.get("name"); return Mono.delay(Duration.ofMillis(50)).map(aLong -> "Hello " + name); }); } - public Flux getGreetings(DataFetchingEnvironment environment) { + public Flux getGreetings() { return Mono.delay(Duration.ofMillis(50)).flatMapMany(aLong -> Flux.deferContextual(context -> { String name = context.get("name"); return Flux.just("Hi", "Bonjour", "Hola", "Ciao", "Zdravo").map(s -> s + " " + name); })); } - public Flux getGreetingsStream(DataFetchingEnvironment environment) { + public Flux getGreetingsStream() { return Mono.delay(Duration.ofMillis(50)).flatMapMany(aLong -> Flux.deferContextual(context -> { String name = context.get("name"); return Flux.just("Hi", "Bonjour", "Hola", "Ciao", "Zdravo").map(s -> s + " " + name); diff --git a/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/SampleWiring.java b/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/SampleWiring.java index d2fefd67..6264e9bd 100644 --- a/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/SampleWiring.java +++ b/samples/webflux-websocket/src/main/java/io/spring/sample/graphql/SampleWiring.java @@ -32,10 +32,10 @@ public class SampleWiring implements RuntimeWiringBuilderCustomizer { @Override public void customize(RuntimeWiring.Builder wiringBuilder) { - wiringBuilder.type("Query", builder -> builder.dataFetcher("greeting", this.repository::getBasic)); - wiringBuilder.type("Query", builder -> builder.dataFetcher("greetingMono", this.repository::getGreeting)); - wiringBuilder.type("Query", builder -> builder.dataFetcher("greetingsFlux", this.repository::getGreetings)); - wiringBuilder.type("Subscription", builder -> builder.dataFetcher("greetings", this.repository::getGreetingsStream)); + wiringBuilder.type("Query", builder -> builder.dataFetcher("greeting", env -> this.repository.getBasic())); + wiringBuilder.type("Query", builder -> builder.dataFetcher("greetingMono", env -> this.repository.getGreeting())); + wiringBuilder.type("Query", builder -> builder.dataFetcher("greetingsFlux", env -> this.repository.getGreetings())); + wiringBuilder.type("Subscription", builder -> builder.dataFetcher("greetings", env -> this.repository.getGreetingsStream())); } }