Send status on client for streaming calls

grpc · Feb 27, 2018 · 953b218 · 953b218
1 parent 3602c87
commit 953b218
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/Sources/gRPC/Call.swift b/Sources/gRPC/Call.swift
@@ -219,26 +219,28 @@ public class Call {
       }
       operations = [
         .sendInitialMetadata(metadata.copy()),
+        .sendMessage(ByteBuffer(data:message)),
+        .sendCloseFromClient,
         .receiveInitialMetadata,
+        .receiveMessage,
         .receiveStatusOnClient,
-        .sendMessage(ByteBuffer(data: message)),
-        .sendCloseFromClient,
-        .receiveMessage
       ]
     case .serverStreaming:
       guard let message = message else {
         throw CallError.invalidMessage
       }
       operations = [
         .sendInitialMetadata(metadata.copy()),
+        .sendMessage(ByteBuffer(data:message)),
+        .sendCloseFromClient,
         .receiveInitialMetadata,
-        .sendMessage(ByteBuffer(data: message)),
-        .sendCloseFromClient
+        .receiveStatusOnClient,
       ]
     case .clientStreaming, .bidiStreaming:
       operations = [
         .sendInitialMetadata(metadata.copy()),
-        .receiveInitialMetadata
+        .receiveInitialMetadata,
+        .receiveStatusOnClient,
       ]
     }
     try perform(OperationGroup(call: self,