fix: McpAsyncServer: addResource use the description as the key problem (failure) to find

mcp-solon/mcp-solon-webrx/src/main/java/io/modelcontextprotocol/server/transport/WebRxSseServerTransportProvider.java

@@ -92,7 +92,6 @@ public class WebRxSseServerTransportProvider implements McpServerTransportProvid
 	 * Map of active client sessions, keyed by session ID.
 	 */
 	private final ConcurrentHashMap<String, McpServerSession> sessions = new ConcurrentHashMap<>();
-	private final ConcurrentHashMap<String, WebRxMcpSessionTransport> sessionTransports = new ConcurrentHashMap<>();
 	/**
 	 * Flag indicating if the transport is shutting down.
 	 */
@@ -117,9 +116,9 @@ public WebRxSseServerTransportProvider(ObjectMapper objectMapper, String message
 		this.sseEndpoint = sseEndpoint;
 	}
 
-	public void sendHeartbeat(){
-		for (WebRxMcpSessionTransport transport : sessionTransports.values()) {
-			transport.sendHeartbeat();
+	public void sendHeartbeat() {
+		for (McpServerSession session : sessions.values()) {
+			((WebRxMcpSessionTransport) session.getTransport()).sendHeartbeat();
 		}
 	}
 
@@ -230,14 +229,13 @@ public void handleSseConnection(Context ctx) throws Throwable{
 		}
 
 		Flux<SseEvent> publisher = Flux.create(sink -> {
-			WebRxMcpSessionTransport sessionTransport = new WebRxMcpSessionTransport(sink);
+			WebRxMcpSessionTransport sessionTransport = new WebRxMcpSessionTransport(ctx, sink);
 
 			McpServerSession session = sessionFactory.create(sessionTransport);
 			String sessionId = session.getId();
 
 			logger.debug("Created new SSE connection for session: {}", sessionId);
 			sessions.put(sessionId, session);
-			sessionTransports.put(sessionId, sessionTransport);
 
 			// Send initial endpoint event
 			logger.debug("Sending initial endpoint event to session: {}", sessionId);
@@ -247,7 +245,6 @@ public void handleSseConnection(Context ctx) throws Throwable{
 			sink.onCancel(() -> {
 				logger.debug("Session {} cancelled", sessionId);
 				sessions.remove(sessionId);
-				sessionTransports.remove(sessionId);
 			});
 		});
 
@@ -277,13 +274,16 @@ public void handleMessage(Context ctx) throws Throwable {
 			return;
 		}
 
-		if (Utils.isEmpty(ctx.param("sessionId"))) {
+		String sessionId = ctx.param("sessionId");
+
+		if (Utils.isEmpty(sessionId)) {
 			ctx.status(404);
 			ctx.render(new McpError("Session ID missing in message endpoint"));
 			return;
 		}
 
-		McpServerSession session = sessions.get(ctx.param("sessionId"));
+
+		McpServerSession session = sessions.get(sessionId);
 
 		String body = ctx.body();
 		try {
@@ -309,14 +309,19 @@ public void handleMessage(Context ctx) throws Throwable {
 		}
 	}
 
-	private class WebRxMcpSessionTransport implements McpServerTransport {
-
+	public class WebRxMcpSessionTransport implements McpServerTransport {
+		private final Context context;
 		private final FluxSink<SseEvent> sink;
 
-		public WebRxMcpSessionTransport(FluxSink<SseEvent> sink) {
+		public WebRxMcpSessionTransport(Context context,FluxSink<SseEvent> sink) {
+			this.context = context;
 			this.sink = sink;
 		}
 
+		public Context getContext() {
+			return context;
+		}
+
 		public void sendHeartbeat() {
 			sink.next(new SseEvent().comment("heartbeat"));
 		}

mcp/src/main/java/io/modelcontextprotocol/server/McpAsyncServer.java

@@ -646,9 +646,9 @@ public Mono<Void> addResource(McpServerFeatures.AsyncResourceSpecification resou
 			}
 
 			return Mono.defer(() -> {
-				if (this.resources.putIfAbsent(resourceSpecification.getResource().getDescription(), resourceSpecification) != null) {
+				if (this.resources.putIfAbsent(resourceSpecification.getResource().getUri(), resourceSpecification) != null) {
 					return Mono.error(new McpError(
-							"Resource with URI '" + resourceSpecification.getResource().getDescription() + "' already exists"));
+							"Resource with URI '" + resourceSpecification.getResource().getUri() + "' already exists"));
 				}
 				logger.debug("Added resource handler: {}", resourceSpecification.getResource().getUri());
 				if (this.serverCapabilities.getResources().getListChanged()) {

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

@@ -81,6 +81,10 @@ public McpServerSession(String id, McpServerTransport transport, InitRequestHand
 		this.notificationHandlers = notificationHandlers;
 	}
 
+	public McpServerTransport getTransport() {
+		return transport;
+	}
+
 	/**
 	 * Retrieve the session id.
 	 * @return session id

pom.xml

@@ -92,7 +92,7 @@
 		<awaitility.version>4.2.0</awaitility.version>
 		<bnd-maven-plugin.version>5.0.1</bnd-maven-plugin.version>
 		<json-unit-assertj.version>2.40.1</json-unit-assertj.version>
-		<solon.version>3.2.1</solon.version>
+		<solon.version>3.3.0</solon.version>
 
 	</properties>