Skip to content

Commit 2a97656

Browse files
committed
sse mvc session close sync when complete or timeout.
1 parent 32562f3 commit 2a97656

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,11 +264,13 @@ private ServerResponse handleSseConnection(ServerRequest request) {
264264
sseBuilder.onComplete(() -> {
265265
logger.debug("SSE connection completed for session: {}", sessionId);
266266
// explicitly close the session when the SSE connection is completed
267-
session.closeGracefully().doOnSuccess(v -> sessions.remove(sessionId)).subscribe();
267+
session.close();
268+
sessions.remove(sessionId);
268269
});
269270
sseBuilder.onTimeout(() -> {
270271
logger.debug("SSE connection timed out for session: {}", sessionId);
271-
session.closeGracefully().doOnSuccess(v -> sessions.remove(sessionId)).subscribe();
272+
session.close();
273+
sessions.remove(sessionId);
272274
});
273275
this.sessions.put(sessionId, session);
274276

0 commit comments

Comments
 (0)