From c62d41623b361d793586d930f47b5e44829e29f2 Mon Sep 17 00:00:00 2001
From: Michael Lumish <mlumish@google.com>
Date: Tue, 3 Jan 2023 09:53:00 -0800
Subject: [PATCH] grpc-js: Discard buffer tracker entry when RetryingCall ends

---
 packages/grpc-js/src/retrying-call.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/packages/grpc-js/src/retrying-call.ts b/packages/grpc-js/src/retrying-call.ts
index f9bda9eb..8daf5ba7 100644
--- a/packages/grpc-js/src/retrying-call.ts
+++ b/packages/grpc-js/src/retrying-call.ts
@@ -202,6 +202,15 @@ export class RetryingCall implements Call {
 
   private reportStatus(statusObject: StatusObject) {
     this.trace('ended with status: code=' + statusObject.code + ' details="' + statusObject.details + '"');
+    this.bufferTracker.freeAll(this.callNumber);
+    for (let i = 0; i < this.writeBuffer.length; i++) {
+      if (this.writeBuffer[i].entryType === 'MESSAGE') {
+        this.writeBuffer[i] = {
+          entryType: 'FREED',
+          allocated: false
+        };
+      }
+    }
     process.nextTick(() => {
       this.listener?.onReceiveStatus(statusObject);
     });