fix continuations types for debug

Sources/WebSocketActors/ClientManager.swift

@@ -138,8 +138,14 @@ public final actor ClientManager: Manager {
  if let remoteNodeID = remoteNodeID {
  return remoteNodeID
  }
+#if DEBUG
  return await withCheckedContinuation { continuation in
-  remoteNodeIDContinuations.append(continuation)
+ remoteNodeIDContinuations.append(continuation)
  }
+#else
+ return await withUnsafeContinuation { continuation in
+ remoteNodeIDContinuations.append(continuation)
+ }
+#endif
  }
 }

Sources/WebSocketActors/Continuation.swift

@@ -13,30 +13,34 @@ import Foundation
 #if DEBUG
  typealias Continuation = CheckedContinuation
 
- @inlinable func withContinuation<T>(function: String = #function,
- _ body: (CheckedContinuation<T, Never>) -> Void) async -> T
- {
- await withCheckedContinuation(function: function, body)
+ @inlinable func withContinuation<T>(
+ function: String = #function,
+ _ body: (CheckedContinuation<T, Never>) -> Void
+ ) async -> T {
+ await withCheckedContinuation(function: function, body)
  }
 
- @inlinable func withThrowingContinuation<T>(function: String = #function,
- _ body: (CheckedContinuation<T, Error>) -> Void) async throws -> T
- {
- try await withCheckedThrowingContinuation(function: function, body)
+ @inlinable func withThrowingContinuation<T>(
+ function: String = #function,
+ _ body: (CheckedContinuation<T, Error>) -> Void
+ ) async throws -> T {
+ try await withCheckedThrowingContinuation(function: function, body)
  }
 #else
  typealias Continuation = UnsafeContinuation
 
- @inlinable func withContinuation<T>(function: String = #function,
- _ body: (UnsafeContinuation<T, Never>) -> Void) async -> T
- {
- await withUnsafeContinuation(function: function, body)
+ @inlinable func withContinuation<T>(
+ function: String = #function,
+ _ body: (UnsafeContinuation<T, Never>) -> Void
+ ) async -> T {
+ await withUnsafeContinuation(body)
  }
 
- @inlinable func withThrowingContinuation<T>(function: String = #function,
- _ body: (UnsafeContinuation<T, Error>) -> Void) async throws -> T
- {
- try await withUnsafeThrowingContinuation(function: function, body)
+ @inlinable func withThrowingContinuation<T>(
+ function: String = #function,
+ _ body: (UnsafeContinuation<T, Error>) -> Void
+ ) async throws -> T {
+ try await withUnsafeThrowingContinuation(body)
  }
 #endif
 

Sources/WebSocketActors/PendingReplies.swift

@@ -32,15 +32,28 @@ actor PendingReplies {
  }
 
  nonisolated func sendMessage(_ body: @Sendable @escaping (CallID) async throws -> Void) async throws -> Data {
- try await withCheckedThrowingContinuation { continuation in
-  Task {
-  let callID = await self.expectReply(continuation: continuation)
-  do {
-  try await body(callID)
-  } catch {
-  try await self.receivedError(callID: callID, error: error)
-  }
-  }
+#if DEBUG
+ try await withCheckedThrowingContinuation { continuation in
+ Task {
+ let callID = await self.expectReply(continuation: continuation)
+ do {
+ try await body(callID)
+ } catch {
+ try await self.receivedError(callID: callID, error: error)
+ }
  }
+ }
+#else
+ try await withUnsafeThrowingContinuation { continuation in
+ Task {
+ let callID = await self.expectReply(continuation: continuation)
+ do {
+ try await body(callID)
+ } catch {
+ try await self.receivedError(callID: callID, error: error)
+ }
+ }
+ }
+#endif
  }
 }

Sources/WebSocketActors/ServerManager.swift

@@ -52,9 +52,15 @@ public final actor ServerManager: Manager {
  channel
  }
  else {
- await withCheckedContinuation { continuation in
- waitingForChannel.append(continuation)
- }
+#if DEBUG
+ await withCheckedContinuation { continuation in
+ waitingForChannel.append(continuation)
+ }
+#else
+ await withUnsafeContinuation { continuation in
+ waitingForChannel.append(continuation)
+ }
+#endif
  }
  }