fix: Client.stream writableNeedDrain

ronag · ronag · commit cd2e3520831a · 2020-09-25T18:52:07.000+02:00
Fixes: #441 Refs: nodejs/node#35348 Refs: nodejs/node#35341
diff --git a/lib/client-stream.js b/lib/client-stream.js
@@ -109,6 +109,12 @@ class StreamHandler extends AsyncResource {
     })
 
     this.res = res
+
+    const needDrain = res.writableNeedDrain !== undefined
+      ? res.writableNeedDrain
+      : res._writableState && res._writableState.needDrain
+
+    return needDrain
   }
 
   onData (chunk) {
diff --git a/lib/core/client.js b/lib/core/client.js
@@ -535,7 +535,9 @@ class Parser extends HTTPParser {
     }
 
     try {
-      request.onHeaders(statusCode, headers, statusCode < 200 ? null : socket[kResume])
+      if (request.onHeaders(statusCode, headers, statusCode < 200 ? null : socket[kResume]) === false) {
+        socket[kPause]()
+      }
     } catch (err) {
       util.destroy(socket, err)
       return 1
diff --git a/lib/core/request.js b/lib/core/request.js
@@ -117,7 +117,7 @@ class Request {
 
     clearRequestTimeout(this)
 
-    this[kHandler].onHeaders(statusCode, headers, resume)
+    return this[kHandler].onHeaders(statusCode, headers, resume)
   }
 
   onBody (chunk, offset, length) {

Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ class Request {`
`117`	`117`
`118`	`118`	`clearRequestTimeout(this)`
`119`	`119`
`120`		`- this[kHandler].onHeaders(statusCode, headers, resume)`
	`120`	`+ return this[kHandler].onHeaders(statusCode, headers, resume)`
`121`	`121`	`}`
`122`	`122`
`123`	`123`	`onBody (chunk, offset, length) {`