Use res.headersSent when available

expressjs · Sep 27, 2017 · ddf4f8c · ddf4f8c
1 parent 7348b5c
commit ddf4f8c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/HISTORY.md b/HISTORY.md
@@ -1,6 +1,7 @@
 unreleased
 ==========
 
+  * Use `res.headersSent` when available
   * deps: basic-auth@~2.0.0
      - Use `safe-buffer` for improved Buffer API
   * deps: debug@2.6.9

diff --git a/index.js b/index.js
@@ -182,7 +182,7 @@ morgan.format('tiny', ':method :url :status :res[content-length] - :response-tim
 
 morgan.format('dev', function developmentFormatLine (tokens, req, res) {
   // get the status code if response written
-  var status = res._header
+  var status = headersSent(res)
     ? res.statusCode
     : undefined
 
@@ -261,7 +261,7 @@ morgan.token('date', function getDateToken (req, res, format) {
  */
 
 morgan.token('status', function getStatusToken (req, res) {
-  return res._header
+  return headersSent(res)
     ? String(res.statusCode)
     : undefined
 })
@@ -328,7 +328,7 @@ morgan.token('req', function getRequestToken (req, res, field) {
  */
 
 morgan.token('res', function getResponseHeader (req, res, field) {
-  if (!res._header) {
+  if (!headersSent(res)) {
     return undefined
   }
 
@@ -469,6 +469,20 @@ function getip (req) {
     undefined
 }
 
+/**
+ * Determine if the response headers have been sent.
+ *
+ * @param {object} res
+ * @returns {boolean}
+ * @private
+ */
+
+function headersSent (res) {
+  return typeof res.headersSent !== 'boolean'
+    ? Boolean(res._header)
+    : res.headersSent
+}
+
 /**
  * Pad number to two digits.
  *