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

Polishing

Browse Source
This commit is contained in:
rstoyanchev
2022-09-01 16:26:53 +01:00
parent 304b3afd3b
commit 2b829dbe07
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -175,11 +175,10 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
}
@SuppressWarnings({"unused", "try"})
@SuppressWarnings("try")
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage webSocketMessage) throws Exception {
ContextSnapshot snapshot = this.contextHandshakeInterceptor.getContextSnapshot(session);
try (AutoCloseable closeable = snapshot.setThreadLocalValues()) {
try (AutoCloseable closeable = ContextHandshakeInterceptor.setThreadLocals(session)) {
handleInternal(session, webSocketMessage);
}
}
@@ -354,12 +353,14 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
*/
private static class ContextHandshakeInterceptor implements HandshakeInterceptor {
private static final String KEY = ContextSnapshot.class.getName();
@Override
public boolean beforeHandshake(
ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
Map<String, Object> attributes) {
attributes.put(ContextSnapshot.class.getName(), ContextSnapshot.capture());
attributes.put(KEY, ContextSnapshot.capture());
return true;
}
@@ -369,10 +370,10 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
@Nullable Exception exception) {
}
public ContextSnapshot getContextSnapshot(WebSocketSession session) {
ContextSnapshot snapshot = (ContextSnapshot) session.getAttributes().get(ContextSnapshot.class.getName());
Assert.notNull(snapshot, "No ContextSnapshot in WebSocketSession attributes");
return snapshot;
public static AutoCloseable setThreadLocals(WebSocketSession session) {
ContextSnapshot snapshot = (ContextSnapshot) session.getAttributes().get(KEY);
Assert.notNull(snapshot, "Expected ContextSnapshot in WebSocketSession attributes");
return snapshot.setThreadLocalValues();
}
}