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

Uncomment assertion for context propagation

Browse Source 
Closes gh-342
This commit is contained in:
rstoyanchev
2022-04-12 21:23:02 +01:00
parent 218b16e022
commit dbba8b17f8
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlWebSocketHandler.java
View File

@@ -365,12 +365,9 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
public Closeable restoreThreadLocalValue(WebSocketSession session) {
if (this.accessor != null) {
Map<String, Object> valuesMap = (Map<String, Object>) session.getAttributes().get(SAVED_CONTEXT_KEY);
// Uncomment when Boot is updated to use HandshakeInterceptor
// Assert.state(valuesMap != null, "No context");
if (valuesMap != null) {
this.accessor.restoreValues(valuesMap);
return () -> this.accessor.resetValues(valuesMap);
}
Assert.state(valuesMap != null, "No ThreadLocal context in WebSocketSession attributes");
this.accessor.restoreValues(valuesMap);
return () -> this.accessor.resetValues(valuesMap);
}
return () -> {};
}