perf: allow keep alive for HEAD requests

ronag · ronag · commit 76d03a7238c1 · 2023-01-17T20:04:44.000+01:00
diff --git a/lib/client.js b/lib/client.js
@@ -441,6 +441,7 @@ class Parser {
 
     this.keepAlive = ''
     this.contentLength = ''
+    this.connection = ''
     this.maxResponseSize = client[kMaxResponseSize]
   }
 
@@ -616,6 +617,8 @@ class Parser {
     const key = this.headers[len - 2]
     if (key.length === 10 && key.toString().toLowerCase() === 'keep-alive') {
       this.keepAlive += buf.toString()
+    } else if (key.length === 10 && key.toString().toLowerCase() === 'connection') {
+      this.connection += buf.toString()
     } else if (key.length === 14 && key.toString().toLowerCase() === 'content-length') {
       this.contentLength += buf.toString()
     }
@@ -709,7 +712,11 @@ class Parser {
     assert.strictEqual(this.timeoutType, TIMEOUT_HEADERS)
 
     this.statusCode = statusCode
-    this.shouldKeepAlive = shouldKeepAlive
+    this.shouldKeepAlive = (
+      shouldKeepAlive ||
+      // Override llhttp value which does not allow keepAlive for HEAD.
+      (request.method === 'HEAD' && !socket[kReset] && this.connection.toLowerCase() === 'keep-alive')
+    )
 
     if (this.statusCode >= 200) {
       const bodyTimeout = request.bodyTimeout != null
@@ -739,7 +746,7 @@ class Parser {
     this.headers = []
     this.headersSize = 0
 
-    if (shouldKeepAlive && client[kPipelining]) {
+    if (this.shouldKeepAlive && client[kPipelining]) {
       const keepAliveTimeout = this.keepAlive ? util.parseKeepAliveTimeout(this.keepAlive) : null
 
       if (keepAliveTimeout != null) {
@@ -769,7 +776,6 @@ class Parser {
     }
 
     if (request.method === 'HEAD') {
-      assert(socket[kReset])
       return 1
     }
 
@@ -843,6 +849,7 @@ class Parser {
     this.bytesRead = 0
     this.contentLength = ''
     this.keepAlive = ''
+    this.connection = ''
 
     assert(this.headers.length % 2 === 0)
     this.headers = []
@@ -1376,8 +1383,8 @@ function write (client, request) {
     socket[kReset] = true
   }
 
-  if (reset) {
-    socket[kReset] = true
+  if (reset != null) {
+    socket[kReset] = reset
   }
 
   if (client[kMaxRequests] && socket[kCounter]++ >= client[kMaxRequests]) {
diff --git a/lib/core/request.js b/lib/core/request.js
@@ -144,7 +144,7 @@ class Request {
 
     this.blocking = blocking == null ? false : blocking
 
-    this.reset = reset == null ? false : reset
+    this.reset = reset == null ? null : reset
 
     this.host = null
 
diff --git a/test/client-head-reset-override.js b/test/client-head-reset-override.js
@@ -0,0 +1,62 @@
+'use strict'
+
+const { createServer } = require('http')
+const { test } = require('tap')
+const { Client } = require('..')
+
+test('override HEAD reset', (t) => {
+  const expected = 'testing123'
+  const server = createServer((req, res) => {
+    if (req.method === 'GET') {
+      res.write(expected)
+    }
+    res.end()
+  })
+  t.teardown(server.close.bind(server))
+
+  server.listen(0, () => {
+    const client = new Client(`http://localhost:${server.address().port}`)
+    t.teardown(client.close.bind(client))
+
+    let done
+    client.on('disconnect', () => {
+      if (!done) {
+        t.fail()
+      }
+    })
+
+    client.request({
+      path: '/',
+      method: 'HEAD',
+      reset: false
+    }, (err, res) => {
+      t.error(err)
+      res.body.resume()
+    })
+
+    client.request({
+      path: '/',
+      method: 'HEAD',
+      reset: false
+    }, (err, res) => {
+      t.error(err)
+      res.body.resume()
+    })
+
+    client.request({
+      path: '/',
+      method: 'GET',
+      reset: false
+    }, (err, res) => {
+      t.error(err)
+      let str = ''
+      res.body.on('data', (data) => {
+        str += data
+      }).on('end', () => {
+        t.same(str, expected)
+        done = true
+        t.end()
+      })
+    })
+  })
+})
