half close right after write

Author: serkanerip

packages/grpc-js/src/retrying-call.ts

@@ -211,6 +211,7 @@ export class RetryingCall implements Call, DeadlineInfoProvider {
   private nextRetryBackoffSec = 0;
   private startTime: Date;
   private maxAttempts: number;
+  private halfClosed: boolean = false;
   constructor(
     private readonly channel: InternalChannel,
     private readonly callConfig: CallConfig,
@@ -789,8 +790,11 @@ export class RetryingCall implements Call, DeadlineInfoProvider {
           );
           break;
         case 'HALF_CLOSE':
-          childCall.nextMessageToSend += 1;
-          childCall.call.halfClose();
+          if (this.halfClosed !== true) {
+            childCall.nextMessageToSend += 1;
+            childCall.call.halfClose();
+            this.halfClosed = true;
+          }
           break;
         case 'FREED':
           // Should not be possible
@@ -868,12 +872,13 @@ export class RetryingCall implements Call, DeadlineInfoProvider {
       allocated: false,
     });
     for (const call of this.underlyingCalls) {
-      if (
-        call?.state === 'ACTIVE' &&
-        call.nextMessageToSend === halfCloseIndex
-      ) {
-        call.nextMessageToSend += 1;
-        call.call.halfClose();
+      if (call?.state === 'ACTIVE' && this.halfClosed !== true) {
+        // || call.nextMessageToSend === halfCloseIndex - 1 added to not wait for write callback
+        if (call.nextMessageToSend === halfCloseIndex || call.nextMessageToSend === halfCloseIndex - 1) {
+          call.nextMessageToSend += 1;
+          call.call.halfClose();
+          this.halfClosed = true;
+        }
       }
     }
   }