index.js

'use strict'

const fp = require('fastify-plugin')
const { memo, create } = require('asyncforge')

const app = memo('fastify.app')
const request = memo('fastify.request')
const reply = memo('fastify.reply')
const logger = memo('fastify.logger')

const fastifyAsyncForge = fp(function (fastify, opts, next) {
  fastify.decorate('runInAsyncScope', function (fn) {
    const store = create()
    return store.run(() => {
      app.set(this)
      logger.set(this.log)
      return fn()
    })
  })

  fastify.decorate('enterWith', function () {
    const store = create()
    store.enterWith()
    app.set(this)
    logger.set(this.log)
  })

  fastify.addHook('onRequest', function (req, res, next) {
    const store = create()

    // We use callbacks because we cannot propagate through async/await
    store.run(() => {
      app.set(this)
      request.set(req)
      reply.set(res)
      logger.set(req.log)
      next()
    })
  })

  create(() => {
    app.set(fastify)
    logger.set(fastify.log)
    next()
  })
})

module.exports = fastifyAsyncForge
module.exports.app = app
module.exports.request = request
module.exports.reply = reply
module.exports.logger = logger