feat: customizable request header name

darora · darora · commit 715306c7b84d · 2022-03-08T18:09:55.000+08:00
diff --git a/src/server/app.ts b/src/server/app.ts
@@ -1,5 +1,5 @@
 import fastify from 'fastify'
-import { PG_META_EXPORT_DOCS, PG_META_PORT } from './constants'
+import { PG_META_EXPORT_DOCS, PG_META_PORT, PG_META_REQ_HEADER } from './constants'
 import routes from './routes'
 import { extractRequestForLogging } from './utils'
 import pino from 'pino'
@@ -17,6 +17,7 @@ const logger = pino({
 const app = fastify({
   logger,
   disableRequestLogging: true,
+  requestIdHeader: PG_META_REQ_HEADER,
 })
 
 app.setErrorHandler((error, request, reply) => {
diff --git a/src/server/constants.ts b/src/server/constants.ts
@@ -14,3 +14,4 @@ export const PG_CONNECTION = `postgres://${PG_META_DB_USER}:${PG_META_DB_PASSWOR
 export const PG_META_EXPORT_DOCS = process.env.PG_META_EXPORT_DOCS === 'true' || false
 
 export const DEFAULT_POOL_CONFIG = { max: 1, connectionTimeoutMillis: PG_CONN_TIMEOUT_SECS * 1000 }
+export const PG_META_REQ_HEADER = process.env.PG_META_REQ_HEADER || 'request-id'

Original file line number	Diff line number	Diff line change
@@ -14,3 +14,4 @@ export const PG_CONNECTION = `postgres://${PG_META_DB_USER}:${PG_META_DB_PASSWOR
`14`	`14`	`export const PG_META_EXPORT_DOCS = process.env.PG_META_EXPORT_DOCS === 'true' \|\| false`
`15`	`15`
`16`	`16`	`export const DEFAULT_POOL_CONFIG = { max: 1, connectionTimeoutMillis: PG_CONN_TIMEOUT_SECS * 1000 }`
	`17`	`+export const PG_META_REQ_HEADER = process.env.PG_META_REQ_HEADER \|\| 'request-id'`