refactor: allow ErrorConstructor type for customError (#225)

fastify · Jun 30, 2024 · f63e62d · f63e62d
1 parent 4d311d6
commit f63e62d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/types/index.d.ts b/types/index.d.ts
@@ -35,7 +35,7 @@ declare namespace fastifyUnderPressure {
     pressureHandler?: (request: FastifyRequest, reply: FastifyReply, type: string, value: number | undefined) => Promise<void> | void;
     sampleInterval?: number;
     exposeStatusRoute?: boolean | string | { routeOpts: object; routeSchemaOpts?: object; routeResponseSchemaOpts?: object; url?: string };
-    customError?: Error;
+    customError?: Error | (new () => Error);
   }
 
   export const TYPE_EVENT_LOOP_DELAY = 'eventLoopDelay'

diff --git a/types/index.test-d.ts b/types/index.test-d.ts
@@ -95,6 +95,17 @@ const server = fastify();
   server.register(fastifyUnderPressure, {
     customError: new Error('custom error message')
   });
+
+  class CustomError extends Error {
+    constructor () {
+      super('Custom error message')
+      Error.captureStackTrace(this, CustomError)
+    }
+  }
+
+  server.register(fastifyUnderPressure, {
+    customError: CustomError
+  });
 };
 
 expectType<'eventLoopDelay'>(fastifyUnderPressure.TYPE_EVENT_LOOP_DELAY)