feat: honor EOF as graceful exit

quarkiverse · Jan 23, 2025 · 3548b77 · 3548b77
1 parent e61179d
commit 3548b77
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/...runtime/src/main/java/io/quarkiverse/mcp/server/stdio/runtime/StdioMcpMessageHandler.java b/...runtime/src/main/java/io/quarkiverse/mcp/server/stdio/runtime/StdioMcpMessageHandler.java
@@ -61,10 +61,15 @@ void initialize(PrintStream stdout, McpRuntimeConfig config) {
 
             @Override
             public void run() {
-                BufferedReader reader = new BufferedReader(new InputStreamReader(in));
-                try {
-                    String line;
-                    while ((line = reader.readLine()) != null) {
+
+                try (BufferedReader reader = new BufferedReader(new InputStreamReader(in))) {
+                    while (true) {
+                        String line = reader.readLine();
+                        if (line == null) {
+                            LOG.debug("EOF received, exiting");
+                            io.quarkus.runtime.Quarkus.asyncExit(0);
+                            return;
+                        }
                         try {
                             JsonObject message;
                             try {