feat(cli): add keepAliveTimeout option

swordev · Mar 18, 2024 · afe2a9c · afe2a9c
1 parent 480e4dc
commit afe2a9c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/odd-monkeys-fail.md b/.changeset/odd-monkeys-fail.md
@@ -0,0 +1,5 @@
+---
+"@datatruck/cli": patch
+---
+
+Add `keepAliveTimeout` option
diff --git a/packages/cli/src/utils/datatruck/repository-server.ts b/packages/cli/src/utils/datatruck/repository-server.ts
@@ -18,6 +18,7 @@ export type DatatruckRepositoryServerOptions = {
     address?: string;
   };
   trustProxy?: true | { remoteAddressHeader: string };
+  keepAliveTimeout?: number;
   allowlist?: {
     /**
      * @default true
@@ -109,7 +110,7 @@ export function createDatatruckRepositoryServer(
   } = {},
 ) {
   const counter = new Counter();
-  return createServer(async (req, res) => {
+  const server = createServer(async (req, res) => {
     const url = req.url || "";
     if (url === "/" || url === "/favicon.ico") return res.end();
     const id = counter.next();
@@ -177,4 +178,7 @@ export function createDatatruckRepositoryServer(
       if (!res.writableEnded) res.end();
     }
   });
+  if (typeof inOptions.keepAliveTimeout === "number")
+    server.keepAliveTimeout = inOptions.keepAliveTimeout;
+  return server;
 }