fix todo clear pending responses

Author: taobaorun

mcp-core/src/main/java/io/modelcontextprotocol/spec/McpServerSession.java

@@ -4,27 +4,27 @@
 
 package io.modelcontextprotocol.spec;
 
-import java.time.Duration;
-import java.util.Map;
-import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.atomic.AtomicBoolean;
-import java.util.concurrent.atomic.AtomicInteger;
-import java.util.concurrent.atomic.AtomicLong;
-import java.util.concurrent.atomic.AtomicReference;
-
 import io.modelcontextprotocol.common.McpTransportContext;
+import io.modelcontextprotocol.json.TypeRef;
 import io.modelcontextprotocol.server.McpAsyncServerExchange;
 import io.modelcontextprotocol.server.McpInitRequestHandler;
 import io.modelcontextprotocol.server.McpNotificationHandler;
 import io.modelcontextprotocol.server.McpRequestHandler;
-import io.modelcontextprotocol.json.TypeRef;
 import io.modelcontextprotocol.util.Assert;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Mono;
 import reactor.core.publisher.MonoSink;
 import reactor.core.publisher.Sinks;
 
+import java.time.Duration;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicLong;
+import java.util.concurrent.atomic.AtomicReference;
+
 /**
  * Represents a Model Context Protocol (MCP) session on the server side. It manages
  * bidirectional JSON-RPC communication with the client.
@@ -37,7 +37,9 @@ public class McpServerSession implements McpLoggableSession {
 
 	private final String id;
 
-	/** Duration to wait for request responses before timing out */
+	/**
+	 * Duration to wait for request responses before timing out
+	 */
 	private final Duration requestTimeout;
 
 	private final AtomicLong requestCounter = new AtomicLong(0);
@@ -348,8 +350,9 @@ private MethodNotFoundError getMethodNotFoundError(String method) {
 
 	@Override
 	public Mono<Void> closeGracefully() {
-		// TODO: clear pendingResponses and emit errors?
 		if (this.closed.compareAndSet(false, true)) {
+			this.pendingResponses.forEach((id, response) -> response.error(new RuntimeException("Session closed")));
+			this.pendingResponses.clear();
 			return this.transport.closeGracefully();
 		}
 		else {
@@ -359,8 +362,9 @@ public Mono<Void> closeGracefully() {
 
 	@Override
 	public void close() {
-		// TODO: clear pendingResponses and emit errors?
 		if (this.closed.compareAndSet(false, true)) {
+			this.pendingResponses.forEach((id, response) -> response.error(new RuntimeException("Session closed")));
+			this.pendingResponses.clear();
 			this.transport.close();
 		}
 	}