fix(H2 Client): pause stream if response data is sent before response…

… event
mertcanaltin · Mar 23, 2024 · 9219ee3 · 9219ee3
1 parent cf6a5ea
commit 9219ee3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/dispatcher/client-h2.js b/lib/dispatcher/client-h2.js
@@ -253,6 +253,7 @@ function writeH2 (client, request) {
 
   /** @type {import('node:http2').ClientHttp2Stream} */
   let stream
+  let isStreamResponseStarted = false
 
   const { hostname, port } = client[kUrl]
 
@@ -388,6 +389,7 @@ function writeH2 (client, request) {
   ++session[kOpenStreams]
 
   stream.once('response', headers => {
+    isStreamResponseStarted = true
     const { [HTTP2_HEADER_STATUS]: statusCode, ...realHeaders } = headers
     request.onResponseStarted()
 
@@ -419,7 +421,7 @@ function writeH2 (client, request) {
   })
 
   stream.on('data', (chunk) => {
-    if (request.onData(chunk) === false) {
+    if (!isStreamResponseStarted || request.onData(chunk) === false) {
       stream.pause()
     }
   })