From 691c5937c1bfe6c76aa82ad54a268b834b0eb048 Mon Sep 17 00:00:00 2001
From: msukkari <michael.sukkarieh@mail.mcgill.ca>
Date: Tue, 25 Mar 2025 18:17:39 -0700
Subject: [PATCH] add connection manager upsert timeout env var

---
 packages/backend/src/connectionManager.ts | 3 ++-
 packages/backend/src/env.ts               | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/backend/src/connectionManager.ts b/packages/backend/src/connectionManager.ts
index e6188159..815de3cd 100644
--- a/packages/backend/src/connectionManager.ts
+++ b/packages/backend/src/connectionManager.ts
@@ -8,6 +8,7 @@ import { Redis } from 'ioredis';
 import { RepoData, compileGithubConfig, compileGitlabConfig, compileGiteaConfig, compileGerritConfig } from "./repoCompileUtils.js";
 import { BackendError, BackendException } from "@sourcebot/error";
 import { captureEvent } from "./posthog.js";
+import { env } from "./env.js";
 import * as Sentry from "@sentry/node";
 
 interface IConnectionManager {
@@ -220,7 +221,7 @@ export class ConnectionManager implements IConnectionManager {
             }
             const totalUpsertDuration = performance.now() - totalUpsertStart;
             this.logger.info(`Upserted ${repoData.length} repos in ${totalUpsertDuration}ms`);
-        });
+        }, { timeout: env.CONNECTION_MANAGER_UPSERT_TIMEOUT_MS });
 
         return {
             repoCount: repoData.length,
diff --git a/packages/backend/src/env.ts b/packages/backend/src/env.ts
index 9b5c86ba..ea730073 100644
--- a/packages/backend/src/env.ts
+++ b/packages/backend/src/env.ts
@@ -43,6 +43,8 @@ export const env = createEnv({
 
         DATABASE_URL: z.string().url().default("postgresql://postgres:postgres@localhost:5432/postgres"),
         CONFIG_PATH: z.string().optional(),
+
+        CONNECTION_MANAGER_UPSERT_TIMEOUT_MS: numberSchema.default(10000),
     },
     runtimeEnv: process.env,
     emptyStringAsUndefined: true,