Skip to content

Commit c9590a2

Browse files
committed
Polish webflux server transports
Use `RouterFunction<ServerResponse>` instead of `RouterFunction<?>` to align with webmvc server transports Signed-off-by: Yanming Zhou <[email protected]>
1 parent 2ee5853 commit c9590a2

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

mcp-spring/mcp-spring-webflux/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ static class MyConfig {
2222
// Router function for SSE transport used by Spring WebFlux to start an HTTP
2323
// server.
2424
@Bean
25-
public RouterFunction<?> mcpRouterFunction(WebFluxSseServerTransport transport) {
25+
public RouterFunction<ServerResponse> mcpRouterFunction(WebFluxSseServerTransport transport) {
2626
return transport.getRouterFunction();
2727
}
2828

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ public class WebFluxSseServerTransportProvider implements McpServerTransportProv
109109

110110
private final String sseEndpoint;
111111

112-
private final RouterFunction<?> routerFunction;
112+
private final RouterFunction<ServerResponse> routerFunction;
113113

114114
private McpServerSession.Factory sessionFactory;
115115

@@ -256,7 +256,7 @@ public Mono<Void> closeGracefully() {
256256
* </ul>
257257
* @return The configured {@link RouterFunction} for handling HTTP requests
258258
*/
259-
public RouterFunction<?> getRouterFunction() {
259+
public RouterFunction<ServerResponse> getRouterFunction() {
260260
return this.routerFunction;
261261
}
262262

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ public class WebFluxStatelessServerTransport implements McpStatelessServerTransp
3838

3939
private final String mcpEndpoint;
4040

41-
private final RouterFunction<?> routerFunction;
41+
private final RouterFunction<ServerResponse> routerFunction;
4242

4343
private McpStatelessServerHandler mcpHandler;
4444

@@ -83,7 +83,7 @@ public Mono<Void> closeGracefully() {
8383
* </ul>
8484
* @return The configured {@link RouterFunction} for handling HTTP requests
8585
*/
86-
public RouterFunction<?> getRouterFunction() {
86+
public RouterFunction<ServerResponse> getRouterFunction() {
8787
return this.routerFunction;
8888
}
8989

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ public class WebFluxStreamableServerTransportProvider implements McpStreamableSe
5555

5656
private final boolean disallowDelete;
5757

58-
private final RouterFunction<?> routerFunction;
58+
private final RouterFunction<ServerResponse> routerFunction;
5959

6060
private McpStreamableServerSession.Factory sessionFactory;
6161

@@ -152,7 +152,7 @@ public Mono<Void> closeGracefully() {
152152
* </ul>
153153
* @return The configured {@link RouterFunction} for handling HTTP requests
154154
*/
155-
public RouterFunction<?> getRouterFunction() {
155+
public RouterFunction<ServerResponse> getRouterFunction() {
156156
return this.routerFunction;
157157
}
158158

0 commit comments

Comments
 (0)