Fix ordering issue

grpc · Nov 12, 2024 · e110eb3 · e110eb3
1 parent 71a7450
commit e110eb3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Sources/GRPCNIOTransportCore/Client/GRPCClientStreamHandler.swift b/Sources/GRPCNIOTransportCore/Client/GRPCClientStreamHandler.swift
@@ -266,6 +266,8 @@ extension GRPCClientStreamHandler {
           // machine) then return.
           if self.requestFinished { return }
 
+          self.requestFinished = true
+
           // Write an empty data frame with the EOS flag set, to signal the RPC
           // request is now finished.
           context.write(
@@ -280,7 +282,6 @@ extension GRPCClientStreamHandler {
             promise: nil
           )
           context.flush()
-          self.requestFinished = true
           break loop
 
         case .awaitMoreMessages: