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

Send Pong via scheduler for the session

Browse Source 
Single threaded scheduler for serial sending, avoiding concurrent
sends that are not allowed by the WebSocket runtime.

Closes gh-793
This commit is contained in:
rstoyanchev
2023-09-15 16:59:19 +01:00
parent e48004f0bf
commit 9d34402545
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -213,7 +213,15 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
.publishOn(state.getScheduler()) // Serial blocking send via single thread
.subscribe(new SendMessageSubscriber(id, session, state));
}
case PING -> session.sendMessage(encode(GraphQlWebSocketMessage.pong(null)));
case PING ->
state.getScheduler().schedule(() -> {
try {
session.sendMessage(encode(GraphQlWebSocketMessage.pong(null)));
}
catch (IOException ex) {
ExceptionWebSocketHandlerDecorator.tryCloseWithError(session, ex, logger);
}
});
case COMPLETE -> {
if (id != null) {
Subscription subscription = state.getSubscriptions().remove(id);