Merge pull request #2306 from murgatroid99/grpc-js_retry_memory_leak_fix

grpc-js: Discard buffer tracker entry when RetryingCall ends
grpc · Jan 3, 2023 · 7f5cb7d · 7f5cb7d
2 parents c9f8f93 + c62d416
commit 7f5cb7d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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);
     });