http: cleanup ClientRequest oncreate

PR-URL: #36862 Reviewed-By: Rich Trott <rtrott@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Daniele Belardi <dwon.dnl@gmail.com>
nodejs · May 1, 2021 · e7bc37909cd76a62e5e9e012fe6d9d837af3c26f · e7bc379
1 parent e1c62dd
commit e7bc37909cd76a62e5e9e012fe6d9d837af3c26f
Showing with 17 additions and 22 deletions.

+17 −22 lib/_http_client.js
diff --git a/lib/_http_client.js b/lib/_http_client.js
@@ -36,6 +36,7 @@ const {
 const net = require('net');
 const url = require('url');
 const assert = require('internal/assert');
+const { once } = require('internal/util');
 const {
   _checkIsHttpToken: checkIsHttpToken,
   debug,
@@ -236,8 +237,6 @@ function ClientRequest(input, options, cb) {
   this.host = host;
   this.protocol = protocol;
 
-  let called = false;
-
   if (this.agent) {
     // If there is an agent we should default to Connection:keep-alive,
     // but only if the Agent will actually reuse the connection!
@@ -301,18 +300,6 @@ function ClientRequest(input, options, cb) {
                       options.headers);
   }
 
-  const oncreate = (err, socket) => {
-    if (called)
-      return;
-    called = true;
-    if (err) {
-      process.nextTick(() => this.emit('error', err));
-      return;
-    }
-    this.onSocket(socket);
-    this._deferToConnect(null, null, () => this._flush());
-  };
-
   // initiate connection
   if (this.agent) {
     this.agent.addRequest(this, options);
@@ -321,20 +308,27 @@ function ClientRequest(input, options, cb) {
     this._last = true;
     this.shouldKeepAlive = false;
     if (typeof options.createConnection === 'function') {
-      const newSocket = options.createConnection(options, oncreate);
-      if (newSocket && !called) {
-        called = true;
-        this.onSocket(newSocket);
-      } else {
-        return;
+      const oncreate = once((err, socket) => {
+        if (err) {
+          process.nextTick(() => this.emit('error', err));
+        } else {
+          this.onSocket(socket);
+        }
+      });
+
+      try {
+        const newSocket = options.createConnection(options, oncreate);
+        if (newSocket) {
+          oncreate(null, newSocket);
+        }
+      } catch (err) {
+        oncreate(err);
       }
     } else {
       debug('CLIENT use net.createConnection', options);
       this.onSocket(net.createConnection(options));
     }
   }
-
-  this._deferToConnect(null, null, () => this._flush());
 }
 ObjectSetPrototypeOf(ClientRequest.prototype, OutgoingMessage.prototype);
 ObjectSetPrototypeOf(ClientRequest, OutgoingMessage);
@@ -827,6 +821,7 @@ function onSocketNT(req, socket, err) {
     _destroy(req, null, err);
   } else {
     tickOnSocket(req, socket);
+    req._flush();
   }
 }