minor

Author: tzolov

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

@@ -203,8 +203,6 @@ public Mono<Void> closeGracefully() {
 			this.sessions.clear();
 			logger.debug("Graceful shutdown completed");
 		}).then().doOnSuccess(v -> {
-			logger.debug("Graceful shutdown completed");
-			sessions.clear();
 			if (this.keepAliveScheduler != null) {
 				this.keepAliveScheduler.shutdown();
 			}

mcp/src/main/java/io/modelcontextprotocol/util/KeepAliveScheduler.java

@@ -50,6 +50,9 @@ public class KeepAliveScheduler {
 	/** The current subscription for the keepAlive calls */
 	private Disposable currentSubscription;
 
+	// TODO Currently we do not support the streams (streamable http session created by
+	// http post/get)
+
 	/** Supplier for reactive McpSession instances */
 	private final Supplier<Flux<McpSession>> mcpSessions;
 
@@ -95,9 +98,10 @@ public Disposable start() {
 				})
 				.doOnCancel(() -> this.isRunning.set(false))
 				.doOnComplete(() -> this.isRunning.set(false))
-				.doOnError(error -> {
+				.onErrorComplete(error -> {
 					logger.error("KeepAlive scheduler error", error);
 					this.isRunning.set(false);
+					return true;
 				})
 				.subscribe();