Iain Sproat
2454af7237
* chore(logging): mixpanel should use logger with request context * log with request when onUserCreated event is triggered
54 lines
1.3 KiB
TypeScript
54 lines
1.3 KiB
TypeScript
import { REQUEST_ID_HEADER } from '@/logging/expressLogging'
|
|
import { asyncRequestContextEnabled } from '@/modules/shared/helpers/envHelper'
|
|
import type express from 'express'
|
|
import { AsyncLocalStorage } from 'node:async_hooks'
|
|
import type { Logger } from 'pino'
|
|
|
|
type StorageType = {
|
|
requestId: string
|
|
dbMetrics: {
|
|
totalDuration: number
|
|
totalCount: number
|
|
}
|
|
}
|
|
|
|
const storage = asyncRequestContextEnabled()
|
|
? new AsyncLocalStorage<StorageType>()
|
|
: undefined
|
|
|
|
export const initiateRequestContextMiddleware: express.RequestHandler = (
|
|
req,
|
|
_res,
|
|
next
|
|
) => {
|
|
const reqId = req.id || req.headers[REQUEST_ID_HEADER] || 'unknown'
|
|
enterNewRequestContext({ reqId: reqId as string })
|
|
next()
|
|
}
|
|
|
|
export const enterNewRequestContext = (params: { reqId: string }) => {
|
|
const { reqId } = params
|
|
const store: StorageType = {
|
|
requestId: reqId,
|
|
dbMetrics: {
|
|
totalCount: 0,
|
|
totalDuration: 0
|
|
}
|
|
}
|
|
storage?.enterWith(store)
|
|
}
|
|
|
|
export const getRequestContext = () => storage?.getStore()
|
|
|
|
export const maybeLoggerWithContext = ({ logger }: { logger?: Logger }) => {
|
|
const reqCtx = getRequestContext()
|
|
return logger?.child({
|
|
...(reqCtx
|
|
? {
|
|
req: { id: reqCtx.requestId },
|
|
dbMetrics: reqCtx.dbMetrics
|
|
}
|
|
: {})
|
|
})
|
|
}
|