Update MCPApplication.java

m0ver · m0ver · commit b49e191628e7 · 2025-05-20T00:08:14.000+08:00
diff --git a/src/main/java/org/tinystruct/mcp/MCPApplication.java b/src/main/java/org/tinystruct/mcp/MCPApplication.java
@@ -103,7 +103,7 @@ public String shutdown(Response response) throws ApplicationException {
                 return jsonRpcHandler.createErrorResponse("Not in ready state", ErrorCodes.NOT_INITIALIZED);
             }
 
-            sseHandler.closeAllConnections();
+            sseHandler.closeAll();
             sessionState = SessionState.DISCONNECTED;
             
             JsonRpcResponse jsonResponse = new JsonRpcResponse();
@@ -131,13 +131,12 @@ public String events(Response response, Request request) throws ApplicationExcep
             }
 
             String clientId = UUID.randomUUID().toString();
-            SSEHandler.SSESession session = sseHandler.createSession(clientId);
+            sseHandler.registerClient(clientId, response);
             sseHandler.setupSSEHeaders(response);
-
-            // Send initial connection event
-            session.sendEvent(Events.CONNECTED, "{\"clientId\":\"" + clientId + "\"}");
-            return session.getOutput();
-
+            // 发送初始连接事件
+            sseHandler.sendEvent(clientId, Events.CONNECTED, "{\"clientId\":\"" + clientId + "\"}");
+            // SSE 连接应保持打开，返回空字符串即可
+            return "";
         } catch (Exception e) {
             LOGGER.log(Level.SEVERE, "SSE connection failed", e);
             response.setStatus(ResponseStatus.INTERNAL_SERVER_ERROR);
@@ -239,7 +238,7 @@ protected void handleShutdown(JsonRpcRequest request, JsonRpcResponse response)
             return;
         }
 
-        sseHandler.closeAllConnections();
+        sseHandler.closeAll();
         sessionState = SessionState.DISCONNECTED;
         
         response.setId(request.getId());
