From 97b756c0f4f9e81ce7a7cb92fb975e6c28cebf7f Mon Sep 17 00:00:00 2001 From: Adam Wulf Date: Sat, 29 Mar 2025 06:26:16 -0500 Subject: [PATCH] Add waitUntilComplete() method to Server (#26) * Add waitUntilComplete() method to Server * Update Sources/MCP/Server/Server.swift Co-authored-by: Mattt --------- Co-authored-by: Mattt --- Sources/MCP/Server/Server.swift | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/Sources/MCP/Server/Server.swift b/Sources/MCP/Server/Server.swift index 66dd6ba..48f9e33 100644 --- 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