{"id":22826,"date":"2021-03-04T08:00:00","date_gmt":"2021-03-04T08:00:00","guid":{"rendered":"https:\/\/codeandpepper.com\/?p=22826"},"modified":"2023-10-11T14:17:19","modified_gmt":"2023-10-11T14:17:19","slug":"asynclocalstorage-in-node-js","status":"publish","type":"post","link":"https:\/\/codeandpepper.com\/asynclocalstorage-in-node-js\/","title":{"rendered":"Simple Context Passing With AsyncLocalStorage in Node.js"},"content":{"rendered":"\n

Node.js is one of the most popular programming languages used on the web nowadays. It\u2019s single core design has lots of positives, but sometimes can be a little challenging – especially when it comes to passing the context between asynchronous invocations. It seems not to be such a problem anymore, though. Developers can use AsyncLocalStorage in Node.js<\/strong> to overcome long lasting problems. Let\u2019s dive in to check how to do that.<\/p>\n\n\n\n\n\n\n\n

\"Context<\/figure><\/div>\n\n\n\n

What problems can AsyncLocalStorage in Node.js solve<\/h2>\n\n\n\n

I bet that most of you have written a custom HTTP request logger to be sure what\u2019s happening behind the scenes for each server invocation. The code roughly could look like this.<\/p>\n\n\n\n

const express = require(\"express\");\n\nconst { v4: uuid } = require(\"uuid\");\n\nconst indexRouter = require(\".\/routes\/index\");\n\nconst app = express();\n\nfunction requestLogger(req, ...args) {\n  console.log(req.id, args);\n}\n\napp.use(express.json());\napp.use(express.urlencoded({ extended: false }));\n\napp.use((req, res, next) => {\n  req.id = uuid();\n  requestLogger(req, \"request started\");\n  next();\n});\n\napp.use(\"\/\", indexRouter);\n\napp.use((req, res, next) => {\n  requestLogger(req, \"request ended\");\n  next();\n});\n\nmodule.exports = app;<\/code><\/pre>\n\n\n\n
Let\u2019s take a look at the requestLogger function: the first parameter includes a request object, while the second one is the message that shall be logged. By looking at this piece of code, it is easy to identify the problem – it is needed to pass the \u201creq\u201d variable (lets call it the request context) to the logger. This kind of a pattern has been known in the Node.js community for years, but obviously that does not mean it should be the standard. It\u2019s worth pointing out that it’s a relatively simple example – oftentimes you may need to log for details of SQL query or details of cloud-related operations (for example by uploading to AWS S3). The situation can get complex really fast and it\u2019s hard to code just one logging method for each use-case. But the solution is on the horizon.<\/p>\n\n\n\n
AsyncLocalStorage in Node.js to the rescue!<\/h2>\n\n\n\n
AsyncLocalStorage<\/strong> is a class which creates an asynchronous state within callbacks and promise chains. It allows us to persist context through a lifetime of async operations, for example within a web request, RabbitMQ event or WebSocket message.<\/p>\n\n\n\n
Before I start explaining how to do it – make sure you are running Node.js in versions 12.17.0 or 13.10.0 at least (14+ supports AsyncLocalStorage by default).<\/strong><\/p>\n\n\n\n
Let\u2019s try to refactor the previous example so it uses this new feature. Also, to show the power of AsyncLocalStorage in Node.js<\/strong>, we will make our logger inform us about the path of request and time delta between the request start and log call. <\/p>\n\n\n\n
const express = require(\"express\");\n\nconst { AsyncLocalStorage } = require(\"async_hooks\");\nconst { v4: uuid } = require(\"uuid\");\n\nconst indexRouter = require(\".\/routes\/index\");\n\nconst app = express();\nconst context = new AsyncLocalStorage();\n\nfunction requestLogger(...args) {\n  const store = context.getStore();\n  const id = store.get(\"id\");\n  const timeStart = store.get(\"timeStart\");\n  const { originalUrl } = store.get(\"request\");\n  console.log(`${id}, ${originalUrl}, ${+new Date() - timeStart}ms`, args);\n}\n\napp.use(express.json());\napp.use(express.urlencoded({ extended: false }));\n\napp.use((req, res, next) => {\n  const store = new Map();\n  context.run(store, () => {\n    store.set(\"id\", uuid());\n    store.set(\"timeStart\", +new Date());\n    store.set(\"request\", req);\n    requestLogger(\"request started\");\n    next();\n  });\n});\n\napp.use(\"\/\", indexRouter);\n\napp.use((req, res, next) => {\n  requestLogger(\"request ended\");\n  next();\n});\n\nmodule.exports = app;<\/code><\/pre>\n\n\n\n
First, it\u2019s necessary to import and create the AsyncLocalStorage context. Then let\u2019s take a look at our middleware on line 22 – first Map is created. It\u2019s going to keep our context. This way we can store more than one variable inside our context with flexible data structure. By calling context.run with our store as an argument, a shared context instance is being created. Inside the callback, we supply the store with needed details about our context – here request ID is being created, request object saved and the timestamp of invocation is saved. <\/p>\n\n\n\n
Now, let\u2019s move back to the requestLogger function. First of all, we can remove all context-related arguments. Later, we have to grab our store from the AsyncLocalStorage instance. The latter is just a simple implementation of the business logic. As you can see, the solution is robust and easy to read. Moreover, we can store as many context-related details as we want. <\/p>\n\n\n\n
The versatility of AsyncLocalStorage in Node.js <\/h2>\n\n\n\n
Example shown in this article is really easy – but this is just an explanation. You can easily implement sophisticated monitoring tools, loggers with lots\u2019 of features, much better error handling, single SQL transaction per HTTP request and more. <\/p>\n\n\n\n
But there are also some tips you have to remember about:<\/p>\n\n\n\n