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

Throw more specific networking errors in NetworkTransport (#5)

Browse source
This commit is contained in:
Mattt 2025-03-04 06:18:44 -08:00 • committed by GitHub
commit d0364a936a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
Sources/MCP/Base/Transports.swift
View file

@ -362,6 +362,12 @@ public actor StdioTransport: Transport {
messageContinuation.yield(message) messageContinuation.yield(message)
} }
} }
} catch let error as NWError {
if !Task.isCancelled {
logger.error("Network error occurred", metadata: ["error": "\(error)"])
messageContinuation.finish(throwing: MCP.Error.transportError(error))
}
break
} catch { } catch {
if !Task.isCancelled { if !Task.isCancelled {
logger.error("Receive error: \(error)") logger.error("Receive error: \(error)")
@ -375,7 +381,6 @@ public actor StdioTransport: Transport {
} }
private func receiveData() async throws -> Data { private func receiveData() async throws -> Data {
// Use a local actor-isolated variable to track continuation state
var receiveContinuationResumed = false var receiveContinuationResumed = false
return try await withCheckedThrowingContinuation { return try await withCheckedThrowingContinuation {
@ -391,7 +396,13 @@ public actor StdioTransport: Transport {
if !receiveContinuationResumed { if !receiveContinuationResumed {
receiveContinuationResumed = true receiveContinuationResumed = true
if let error = error { if let error = error {
continuation.resume(throwing: error) if let nwError = error as? NWError {
continuation.resume(throwing: MCP.Error.transportError(nwError))
} else {
continuation.resume(
throwing: MCP.Error.internalError("Receive error: \(error)")
)
}
} else if let content = content { } else if let content = content {
continuation.resume(returning: content) continuation.resume(returning: content)
} else { } else {