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

Handle PONG messages in WebSocket handlers

Browse Source 
See gh-534
This commit is contained in:
rstoyanchev
2024-04-11 16:59:42 +01:00
parent 74688ea4ff
commit 10ea73d150
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spring-graphql/src/main/java/org/springframework/graphql/server/webflux/GraphQlWebSocketHandler.java
View File

@@ -183,6 +183,9 @@ public class GraphQlWebSocketHandler implements WebSocketHandler {
case PING -> { case PING -> {
return Flux.just(this.codecDelegate.encode(session, GraphQlWebSocketMessage.pong(null))); return Flux.just(this.codecDelegate.encode(session, GraphQlWebSocketMessage.pong(null)));
} }
case PONG -> {
return Flux.empty();
}
case COMPLETE -> { case COMPLETE -> {
if (id != null) { if (id != null) {
Subscription subscription = subscriptions.remove(id); Subscription subscription = subscriptions.remove(id);

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

@@ -248,6 +248,8 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
ExceptionWebSocketHandlerDecorator.tryCloseWithError(session, ex, logger); ExceptionWebSocketHandlerDecorator.tryCloseWithError(session, ex, logger);
} }
}); });
case PONG -> {
}
case COMPLETE -> { case COMPLETE -> {
if (id != null) { if (id != null) {
Subscription subscription = state.getSubscriptions().remove(id); Subscription subscription = state.getSubscriptions().remove(id);