@nxtedition/lib 23.6.21 → 23.7.1

package/http.js

@@ -155,6 +155,78 @@ function noop() {}
 
 const pendingSet = (globalThis._nxt_lib_http_pending ??= new Set())
 
+export async function upgradeMiddleware(ctx, next) {
+  const { req, socket, logger } = ctx
+  const startTime = performance.now()
+
+  let aborted = false
+
+  req.on('error', noop)
+  socket.on('error', (err) => {
+    // NOTE: Special case where the client becomes unreachable.
+    if (err.message.startsWith('read ')) {
+      aborted = true
+    }
+  })
+
+  const isHealthcheck = req.url === '/healthcheck' || req.url === '/_up'
+  const reqLogger = isHealthcheck ? null : logger.child({ req })
+
+  pendingSet.add(ctx)
+  try {
+    if (req.method === 'GET' || req.method === 'HEAD' || req.method === 'OPTIONS') {
+      req.resume() // Dump the body if there is one.
+    }
+
+    reqLogger?.debug('stream started')
+
+    const thenable = next()
+
+    if (thenable?.then) {
+      await thenable
+    }
+
+    assert(socket.destroyed || socket.writableEnded, 'stream not completed')
+
+    const elapsedTime = performance.now() - startTime
+
+    if (isHealthcheck) {
+      // Do nothing...
+    } else if (socket.errored) {
+      reqLogger?.error({ err: socket.errored, req, socket, elapsedTime }, 'stream error')
+    } else if (!socket.writableEnded) {
+      reqLogger?.debug({ req, socket, elapsedTime }, 'stream aborted')
+    } else if (socket.statusCode >= 500) {
+      reqLogger?.error({ req, socket, elapsedTime }, 'stream error')
+    } else if (socket.statusCode >= 400) {
+      reqLogger?.warn({ req, socket, elapsedTime }, 'stream failed')
+    } else {
+      reqLogger?.debug({ req, socket, elapsedTime }, 'stream completed')
+    }
+  } catch (err) {
+    ctx[kAbortController]?.abort(err)
+
+    const statusCode = err.statusCode || err.$metadata?.httpStatusCode || 500
+    const elapsedTime = performance.now() - startTime
+
+    if (req.aborted || aborted || (!socket.errored && socket.closed) || err.name === 'AbortError') {
+      reqLogger?.debug({ socket, err, elapsedTime }, 'stream aborted')
+    } else if (statusCode < 500) {
+      reqLogger?.warn({ socket, err, elapsedTime }, 'stream failed')
+    } else {
+      reqLogger?.error({ socket, err, elapsedTime }, 'stream error')
+    }
+    socket.destroy(err)
+  } finally {
+    pendingSet.delete(ctx)
+
+    if (!socket.writableEnded && !socket.destroyed) {
+      socket.destroy()
+      reqLogger?.warn('socket destroyed')
+    }
+  }
+}
+
 export async function requestMiddleware(ctx, next) {
   const { req, res, logger } = ctx
   const startTime = performance.now()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/lib",
-  "version": "23.6.21",
+  "version": "23.7.1",
   "license": "MIT",
   "author": "Robert Nagy <robert.nagy@boffins.se>",
   "type": "module",