Fixed backfill script timing out before DB connection

2026-05-22 09:17:51 -04:00
parent f8b1811b74
commit 46f8962742
1 changed files with 17 additions and 0 deletions
@@ -10,8 +10,23 @@ const {
    DB_USERNAME,
    DB_PASSWORD,
    DB_DATABASE,
+    DB_CONNECT_TIMEOUT_MS,
+    DB_SOCKET_TIMEOUT_MS,
 } = process.env;

+function parseTimeout(value, fallback) {
+    if (value === undefined || value === null || value === "") {
+        return fallback;
+    }
+
+    const parsed = Number(value);
+    if (!Number.isFinite(parsed) || parsed <= 0) {
+        throw new Error(`Invalid timeout value: ${value}`);
+    }
+
+    return parsed;
+}
+
 function parseArgs(argv) {
    const args = {
        dryRun: false,
@@ -95,6 +110,8 @@ function key(memberId, courseId) {
        password: DB_PASSWORD,
        database: DB_DATABASE,
        multipleStatements: false,
+        connectTimeout: parseTimeout(DB_CONNECT_TIMEOUT_MS, 10000),
+        socketTimeout: parseTimeout(DB_SOCKET_TIMEOUT_MS, 60000),
    });

    try {