Merge pull request #1304 from tediousjs/arthur/unpipe-payload

fix: unpipe payload streams on cancel or error
tediousjs · Aug 1, 2021 · 3f57180 · 3f57180
2 parents 9865073 + f195526
commit 3f57180
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/connection.ts b/src/connection.ts
@@ -3244,6 +3244,8 @@ class Connection extends EventEmitter {
       request.rst! = [];
 
       const onCancel = () => {
+        payloadStream.unpipe(message);
+
         // set the ignore bit and end the message.
         message.ignore = true;
         message.end();
@@ -3274,6 +3276,8 @@ class Connection extends EventEmitter {
 
       const payloadStream = (Readable as any).from(payload!) as Stream;
       payloadStream.once('error', (error) => {
+        payloadStream.unpipe(message);
+
         // Only set a request error if no error was set yet.
         request.error ??= error;