Skip to content

Commit 656b6e3

Browse files
committed
Return void from McpStatelessSyncServer#closeGracefully instead of Mono
Signed-off-by: Christian Tzolov <christian.tzolov@broadcom.com>
1 parent df75857 commit 656b6e3

2 files changed

Lines changed: 5 additions & 7 deletions

File tree

mcp-core/src/main/java/io/modelcontextprotocol/server/McpStatelessSyncServer.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,10 @@ public McpSchema.Implementation getServerInfo() {
5050

5151
/**
5252
* Gracefully closes the server, allowing any in-progress operations to complete.
53-
* @return A Mono that completes when the server has been closed
53+
*
5454
*/
55-
public Mono<Void> closeGracefully() {
56-
return this.asyncServer.closeGracefully();
55+
public void closeGracefully() {
56+
this.asyncServer.closeGracefully().block();
5757
}
5858

5959
/**

mcp-test/src/main/java/io/modelcontextprotocol/AbstractStatelessIntegrationTests.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44

55
package io.modelcontextprotocol;
66

7-
import static io.modelcontextprotocol.util.ToolsUtils.EMPTY_JSON_SCHEMA;
8-
97
import java.net.URI;
108
import java.net.http.HttpClient;
119
import java.net.http.HttpRequest;
@@ -31,9 +29,9 @@
3129
import net.javacrumbs.jsonunit.core.Option;
3230
import org.junit.jupiter.params.ParameterizedTest;
3331
import org.junit.jupiter.params.provider.MethodSource;
34-
import org.junit.jupiter.params.provider.ValueSource;
3532
import reactor.core.publisher.Mono;
3633

34+
import static io.modelcontextprotocol.util.ToolsUtils.EMPTY_JSON_SCHEMA;
3735
import static net.javacrumbs.jsonunit.assertj.JsonAssertions.assertThatJson;
3836
import static net.javacrumbs.jsonunit.assertj.JsonAssertions.json;
3937
import static org.assertj.core.api.Assertions.assertThat;
@@ -128,7 +126,7 @@ void testToolCallSuccess(String clientType) {
128126
assertThat(response).isNotNull().isEqualTo(callResponse);
129127
}
130128
finally {
131-
mcpServer.closeGracefully().block();
129+
mcpServer.closeGracefully();
132130
}
133131
}
134132

0 commit comments

Comments
 (0)