Skip to content

Commit bb6514e

Browse files
denniskawurektzolov
authored andcommitted
Remove closed sessions from list in WebMvcSseServerTransport
1 parent dac5c54 commit bb6514e

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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

+8
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,14 @@ private ServerResponse handleSseConnection(ServerRequest request) {
218218
// Send initial endpoint event
219219
try {
220220
return ServerResponse.sse(sseBuilder -> {
221+
sseBuilder.onComplete(() -> {
222+
logger.debug("SSE connection completed for session: {}", sessionId);
223+
sessions.remove(sessionId);
224+
});
225+
sseBuilder.onTimeout(() -> {
226+
logger.debug("SSE connection timed out for session: {}", sessionId);
227+
sessions.remove(sessionId);
228+
});
221229

222230
ClientSession session = new ClientSession(sessionId, sseBuilder);
223231
this.sessions.put(sessionId, session);

0 commit comments

Comments
 (0)