sse mvc session close async when complete or timeout

Author: taobaorun

mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransportProvider.java

@@ -264,13 +264,11 @@ private ServerResponse handleSseConnection(ServerRequest request) {
 			sseBuilder.onComplete(() -> {
 				logger.debug("SSE connection completed for session: {}", sessionId);
 				// explicitly close the session when the SSE connection is completed
-				session.close();
-				sessions.remove(sessionId);
+				session.closeGracefully().doOnSuccess(v -> sessions.remove(sessionId)).subscribe();
 			});
 			sseBuilder.onTimeout(() -> {
 				logger.debug("SSE connection timed out for session: {}", sessionId);
-				session.close();
-				sessions.remove(sessionId);
+				session.closeGracefully().doOnSuccess(v -> sessions.remove(sessionId)).subscribe();
 			});
 			this.sessions.put(sessionId, session);