From bb734bf8949ba6f39e893824eb9a59b34e1afe96 Mon Sep 17 00:00:00 2001
From: rstoyanchev <rossen.stoyanchev@broadcom.com>
Date: Wed, 27 Mar 2024 11:50:26 +0000
Subject: [PATCH] Remove session state from onClose only

Doing the same eagerly from onError has the side effect of not being able
to delegate handleConnectionClosed to interceptors.

Closes gh-872
---
 .../graphql/server/webmvc/GraphQlWebSocketHandler.java    | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlWebSocketHandler.java b/spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlWebSocketHandler.java
index 65e3988a..ef5f43b6 100644
--- a/spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlWebSocketHandler.java
+++ b/spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlWebSocketHandler.java
@@ -326,14 +326,6 @@ public class GraphQlWebSocketHandler extends TextWebSocketHandler implements Sub
 		}
 	}
 
-	@Override
-	public void handleTransportError(WebSocketSession session, Throwable exception) {
-		SessionState info = this.sessionInfoMap.remove(session.getId());
-		if (info != null) {
-			info.dispose();
-		}
-	}
-
 	@Override
 	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) {
 		String id = session.getId();