Add configurable request timeout for NDJSON loader

Adds --request-timeout CLI flag to configure the mssql request timeout,
with a default of 300000ms (5 minutes). This allows large NDJSON files
to be loaded without timing out during bulk insert operations.

Also supports MSSQL_REQUEST_TIMEOUT environment variable.

Author: johngrimes

src/load.ts

@@ -23,6 +23,7 @@ function getDryRunDatabaseConfig(commandOptions: Record<string, unknown>): {
   password: string;
   database: string;
   trustServerCertificate: boolean | undefined;
+  requestTimeout: number | undefined;
 } {
   return {
     host: (commandOptions.host as string | undefined) ?? "localhost",
@@ -33,6 +34,7 @@ function getDryRunDatabaseConfig(commandOptions: Record<string, unknown>): {
     trustServerCertificate: commandOptions.trustServerCertificate as
       | boolean
       | undefined,
+    requestTimeout: commandOptions.requestTimeout as number | undefined,
   };
 }
 
@@ -58,6 +60,7 @@ function buildLoaderOptions(
         trustServerCertificate: commandOptions.trustServerCertificate as
           | boolean
           | undefined,
+        requestTimeout: commandOptions.requestTimeout as number | undefined,
       });
 
   return {
@@ -119,6 +122,12 @@ export function createLoadCommand(): Command {
     .option("--password <password>", "Database password")
     .option("--database <database>", "Database name")
     .option("--trust-server-certificate", "Trust server certificate", false)
+    .option(
+      "--request-timeout <ms>",
+      "Request timeout in milliseconds",
+      parseInt,
+      300000,
+    )
     .option(
       "--pattern <pattern>",
       "File naming pattern (default: {ResourceType}.ndjson)",

src/loader/connection.ts

@@ -23,6 +23,7 @@ export async function createConnectionPool(
     user: config.user,
     password: config.password,
     database: config.database,
+    requestTimeout: config.requestTimeout,
     options: {
       trustServerCertificate: config.trustServerCertificate ?? false,
       // Enable multiple active result sets for parallel operations.
@@ -103,12 +104,19 @@ export function getDatabaseConfigFromEnv(
     overrides?.trustServerCertificate ??
     process.env.MSSQL_TRUST_SERVER_CERTIFICATE === "true";
 
+  const requestTimeout =
+    overrides?.requestTimeout ??
+    (process.env.MSSQL_REQUEST_TIMEOUT
+      ? parseInt(process.env.MSSQL_REQUEST_TIMEOUT, 10)
+      : undefined);
+
   return {
     host,
     port,
     user,
     password,
     database,
     trustServerCertificate,
+    requestTimeout,
   };
 }

src/loader/types.ts

@@ -20,6 +20,8 @@ export interface DatabaseConfig {
   database: string;
   /** Whether to trust the server certificate. */
   trustServerCertificate?: boolean;
+  /** Request timeout in milliseconds (default: 300000). */
+  requestTimeout?: number;
 }
 
 /**