modelcontextprotocol/java-sdk
Watch
1
0
Fork
You've already forked java-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove closed sessions from list in WebMvcSseServerTransport

Browse source
This commit is contained in:
Dennis Kawurek 2025-02-22 15:33:32 +01:00 • committed by Christian Tzolov
commit bb6514e808
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransport.java
View file

@ -218,6 +218,14 @@ public class WebMvcSseServerTransport implements ServerMcpTransport {
// Send initial endpoint event // Send initial endpoint event
try { try {
return ServerResponse.sse(sseBuilder -> { return ServerResponse.sse(sseBuilder -> {
sseBuilder.onComplete(() -> {
logger.debug("SSE connection completed for session: {}", sessionId);
sessions.remove(sessionId);
});
sseBuilder.onTimeout(() -> {
logger.debug("SSE connection timed out for session: {}", sessionId);
sessions.remove(sessionId);
});
ClientSession session = new ClientSession(sessionId, sseBuilder); ClientSession session = new ClientSession(sessionId, sseBuilder);
this.sessions.put(sessionId, session); this.sessions.put(sessionId, session);