Skip to content

Commit a9f8337

Browse files
committed
chore: Using Mono#then
Signed-off-by: He-Pin <hepin1989@gmail.com>
1 parent 0069c97 commit a9f8337

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/server/transport/WebFluxSseServerTransportProvider.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ private Mono<ServerResponse> handleMessage(ServerRequest request) {
314314
return request.bodyToMono(String.class).flatMap(body -> {
315315
try {
316316
McpSchema.JSONRPCMessage message = McpSchema.deserializeJsonRpcMessage(objectMapper, body);
317-
return session.handle(message).flatMap(response -> ServerResponse.ok().build()).onErrorResume(error -> {
317+
return session.handle(message).then(Mono.defer(() -> ServerResponse.ok().build())).onErrorResume(error -> {
318318
logger.error("Error processing message: {}", error.getMessage());
319319
// TODO: instead of signalling the error, just respond with 200 OK
320320
// - the error is signalled on the SSE connection

0 commit comments

Comments
 (0)