Spring/spring-graphql
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add WebSocketInterceptor

Browse Source 
See gh-162
This commit is contained in:
Rossen Stoyanchev
2021-10-20 21:21:31 +01:00
parent 811d32b647
commit 20bae75ed8
23 changed files with 361 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
samples/webflux-websocket/src/test/java/io/spring/sample/graphql/QueryTests.java
View File

@@ -36,7 +36,7 @@ public class QueryTests {
@BeforeEach
public void setUp(@Autowired WebGraphQlHandler handler) {
this.graphQlTester = WebGraphQlTester.create(webInput ->
handler.handle(webInput).contextWrite(context -> context.put("name", "James")));
handler.handleRequest(webInput).contextWrite(context -> context.put("name", "James")));
}
@Test

2
samples/webflux-websocket/src/test/java/io/spring/sample/graphql/SubscriptionTests.java
View File

@@ -38,7 +38,7 @@ public class SubscriptionTests {
@BeforeEach
public void setUp(@Autowired WebGraphQlHandler handler) {
this.graphQlTester = WebGraphQlTester.create(webInput ->
handler.handle(webInput).contextWrite(context -> context.put("name", "James")));
handler.handleRequest(webInput).contextWrite(context -> context.put("name", "James")));
}
@Test

4
samples/webmvc-http-security/src/main/java/io/spring/sample/graphql/SampleApplication.java
View File

@@ -34,9 +34,9 @@ public class SampleApplication {
@Bean
public WebInterceptor interceptor() {
return (input, next) -> {
return (webInput, interceptorChain) -> {
// Switch threads to prove ThreadLocal context propagation works
return Mono.delay(Duration.ofMillis(10)).flatMap(aLong -> next.handle(input));
return Mono.delay(Duration.ofMillis(10)).flatMap(aLong -> interceptorChain.next(webInput));
};
}