Skip to content

Commit 44f4329

Browse files
committed
Add method to get the MCP handler
This will allow setting custom MCP handlers that defer to the handler that is set by the framework
1 parent 9bbfbe5 commit 44f4329

File tree

4 files changed

+19
-2
lines changed

4 files changed

+19
-2
lines changed

mcp-core/src/main/java/io/modelcontextprotocol/server/transport/HttpServletStatelessServerTransport.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,11 @@ public void setMcpHandler(McpStatelessServerHandler mcpHandler) {
7474
this.mcpHandler = mcpHandler;
7575
}
7676

77+
@Override
78+
public McpStatelessServerHandler getMcpHandler() {
79+
return mcpHandler;
80+
}
81+
7782
@Override
7883
public Mono<Void> closeGracefully() {
7984
return Mono.fromRunnable(() -> this.isClosing = true);

mcp-core/src/main/java/io/modelcontextprotocol/spec/McpStatelessServerTransport.java

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

55
package io.modelcontextprotocol.spec;
66

7-
import java.util.List;
8-
97
import io.modelcontextprotocol.server.McpStatelessServerHandler;
108
import reactor.core.publisher.Mono;
119

10+
import java.util.List;
11+
1212
public interface McpStatelessServerTransport {
1313

1414
void setMcpHandler(McpStatelessServerHandler mcpHandler);
1515

16+
McpStatelessServerHandler getMcpHandler();
17+
1618
/**
1719
* Immediately closes all the transports with connected clients and releases any
1820
* associated resources.

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

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,11 @@ public void setMcpHandler(McpStatelessServerHandler mcpHandler) {
6666
this.mcpHandler = mcpHandler;
6767
}
6868

69+
@Override
70+
public McpStatelessServerHandler getMcpHandler() {
71+
return mcpHandler;
72+
}
73+
6974
@Override
7075
public Mono<Void> closeGracefully() {
7176
return Mono.fromRunnable(() -> this.isClosing = true);

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

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,11 @@ public void setMcpHandler(McpStatelessServerHandler mcpHandler) {
7070
this.mcpHandler = mcpHandler;
7171
}
7272

73+
@Override
74+
public McpStatelessServerHandler getMcpHandler() {
75+
return mcpHandler;
76+
}
77+
7378
@Override
7479
public Mono<Void> closeGracefully() {
7580
return Mono.fromRunnable(() -> this.isClosing = true);

0 commit comments

Comments
 (0)