Add waitUntilComplete() method to Server (#26)

adamwulf · mattt · web-flow · commit 97b756c0f4f9 · 2025-03-29T04:26:16.000-07:00
* Add waitUntilComplete() method to Server

* Update Sources/MCP/Server/Server.swift

Co-authored-by: Mattt &lt;mattt@me.com&gt;

---------

Co-authored-by: Mattt &lt;mattt@me.com&gt;
diff --git a/Sources/MCP/Server/Server.swift b/Sources/MCP/Server/Server.swift
@@ -221,6 +221,7 @@ public actor Server {
                 await logger?.error(
                     "Fatal error in message handling loop", metadata: ["error": "\(error)"])
             }
+            await logger?.info("Server finished", metadata: [:])
         }
     }
 
@@ -234,6 +235,10 @@ public actor Server {
         connection = nil
     }
 
+    public func waitUntilCompleted() async {
+        await task?.value
+    }
+
     // MARK: - Registration
 
     /// Register a method handler

Original file line number	Diff line number	Diff line change
`@@ -221,6 +221,7 @@ public actor Server {`
`221`	`221`	`await logger?.error(`
`222`	`222`	`"Fatal error in message handling loop", metadata: ["error": "\(error)"])`
`223`	`223`	`}`
	`224`	`+ await logger?.info("Server finished", metadata: [:])`
`224`	`225`	`}`
`225`	`226`	`}`
`226`	`227`
`@@ -234,6 +235,10 @@ public actor Server {`
`234`	`235`	`connection = nil`
`235`	`236`	`}`
`236`	`237`
	`238`	`+ public func waitUntilCompleted() async {`
	`239`	`+ await task?.value`
	`240`	`+ }`
	`241`	`+`
`237`	`242`	`// MARK: - Registration`
`238`	`243`
`239`	`244`	`/// Register a method handler`