http: cleanup cork logic

Author: ronag

lib/_http_outgoing.js

@@ -90,6 +90,7 @@ const kBytesWritten = Symbol('kBytesWritten');
 const kErrored = Symbol('errored');
 const kHighWaterMark = Symbol('kHighWaterMark');
 const kRejectNonStandardBodyWrites = Symbol('kRejectNonStandardBodyWrites');
+const kSocket = Symbol('kSocket');
 
 const nop = () => {};
 
@@ -211,8 +212,8 @@ ObjectDefineProperty(OutgoingMessage.prototype, 'writableHighWaterMark', {
 ObjectDefineProperty(OutgoingMessage.prototype, 'writableCorked', {
   __proto__: null,
   get() {
-    return this[kCorked];
-  },
+    return this[kSocket] ? this[kSocket].writableCorked : this[kCorked];
+  }
 });
 
 ObjectDefineProperty(OutgoingMessage.prototype, '_headers', {
@@ -260,6 +261,20 @@ ObjectDefineProperty(OutgoingMessage.prototype, 'socket', {
   },
 });
 
+ObjectDefineProperty(OutgoingMessage.prototype, 'socket', {
+  __proto__: null,
+  get: function() {
+    return this[kSocket];
+  },
+  set: function(val) {
+    for (let n = 0; n < this[kCorked]; n++) {
+      val?.cork();
+    }
+    this[kCorked] = 0;
+    this[kSocket] = val;
+  }
+});
+
 ObjectDefineProperty(OutgoingMessage.prototype, '_headerNames', {
   __proto__: null,
   get: internalUtil.deprecate(function() {
@@ -1000,9 +1015,9 @@ function write_(msg, chunk, encoding, callback, fromEnd) {
     }
   }
 
-  if (!fromEnd && msg.socket && !msg.socket.writableCorked) {
-    msg.socket.cork();
-    process.nextTick(connectionCorkNT, msg.socket);
+  if (!fromEnd && !msg.writableCorked) {
+    msg.cork();
+    process.nextTick(connectionUncorkNT, msg);
   }
 
   let ret;
@@ -1027,8 +1042,8 @@ function write_(msg, chunk, encoding, callback, fromEnd) {
 }
 
 
-function connectionCorkNT(conn) {
-  conn.uncork();
+function connectionUncorkNT(msg) {
+  msg.uncork();
 }
 
 OutgoingMessage.prototype.addTrailers = function addTrailers(headers) {
@@ -1143,9 +1158,9 @@ OutgoingMessage.prototype.end = function end(chunk, encoding, callback) {
     // Fully uncork connection on end().
     this[kSocket]._writableState.corked = 1;
     this[kSocket].uncork();
+  } else {
+    this[kCorked] = 0;
   }
-  this[kCorked] = 1;
-  this.uncork();
 
   this.finished = true;