modelcontextprotocol/java-sdk
Watch
1
0
Fork
You've already forked java-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(sse): expand accepted HTTP status codes for SSE stream connection

Browse source 
Accept HTTP status codes 200, 201, 202, and 206 as valid responses
for SSE stream connection. Improves error message to include actual
status code. Removes unused main method.

Resolves #68
This commit is contained in:
Christian Tzolov 2025-01-24 18:25:35 +01:00
commit d75adb68c7
1 changed files with 3 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

31
mcp/src/main/java/org/springframework/ai/mcp/client/transport/FlowSseClient.java
View file

@ -198,8 +198,9 @@ public class FlowSseClient {
info -> subscriberFactory.apply(lineSubscriber));
future.thenAccept(response -> {
if (response.statusCode() != 200) {
throw new RuntimeException("Failed to connect to SSE stream: " + response.statusCode());
int status = response.statusCode();
if (status != 200 && status != 201 && status != 202 && status != 206) {
throw new RuntimeException("Failed to connect to SSE stream. Unexpected status code: " + status);
}
}).exceptionally(throwable -> {
eventHandler.onError(throwable);
@ -207,30 +208,4 @@ public class FlowSseClient {
});
}
// public static void main(String[] args) {
// FlowSseClient sseClient = new FlowSseClient(
// HttpClient.newBuilder().connectTimeout(Duration.ofSeconds(10)).build());
// var eventHandler = new SSEEventHandler() {
// @Override
// public void onEvent(SSEEvent event) {
// System.out.println("Received event: " + event);
// }
// @Override
// public void onError(Throwable error) {
// System.out.println("Error: " + error);
// }
// };
// sseClient.subscribe("http://localhost:8080/sse", eventHandler);
// try {
// Thread.sleep(100000);
// }
// catch (InterruptedException e) {
// e.printStackTrace();
// }
// }
}