Provide trailingMetadata for streaming response error

sundayfun · Aug 23, 2023 · adb8f8d · adb8f8d
1 parent 279170a
commit adb8f8d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Sources/GRPC/AsyncAwaitSupport/GRPCAsyncClientStreamingCall.swift b/Sources/GRPC/AsyncAwaitSupport/GRPCAsyncClientStreamingCall.swift
@@ -55,7 +55,15 @@ public struct GRPCAsyncClientStreamingCall<Request: Sendable, Response: Sendable
   /// The response returned by the server.
   public var response: Response {
     get async throws {
-      try await self.responseParts.response.get()
+      do {
+        return try await self.responseParts.response.get()
+      } catch {
+        if let grpcStatus = error as? GRPCStatus {
+          throw GRPCAsyncError(status: grpcStatus, trailers: try? await self.trailingMetadata)
+        } else {
+          throw error
+        }
+      }
     }
   }