fix: don't prod check paused connectors (#4582)

Co-authored-by: Henry Fontanier <[email protected]>
dust-tt · May 26, 2024 · 5074184 · 5074184
1 parent 0f67914
commit 5074184
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/front/production_checks/checks/check_active_workflows_for_connectors.ts b/front/production_checks/checks/check_active_workflows_for_connectors.ts
@@ -1,7 +1,7 @@
+import type { ConnectorProvider } from "@dust-tt/types";
 import {
-  makeConfluenceSyncWorkflowId,
-  type ConnectorProvider,
   getIntercomSyncWorkflowId,
+  makeConfluenceSyncWorkflowId,
 } from "@dust-tt/types";
 import type { Client, WorkflowHandle } from "@temporalio/client";
 import { QueryTypes } from "sequelize";
@@ -14,6 +14,7 @@ interface ConnectorBlob {
   id: number;
   dataSourceName: string;
   workspaceId: string;
+  pausedAt: Date | null;
 }
 
 interface ProviderCheck {
@@ -35,7 +36,7 @@ const providersToCheck: Partial<Record<ConnectorProvider, ProviderCheck>> = {
 
 async function listAllConnectorsForProvider(provider: ConnectorProvider) {
   const connectors: ConnectorBlob[] = await connectorsReplica.query(
-    `SELECT id, "dataSourceName", "workspaceId" FROM connectors WHERE "type" = :provider and  "errorType" IS NULL`,
+    `SELECT id, "dataSourceName", "workspaceId", "pausedAt" FROM connectors WHERE "type" = :provider and  "errorType" IS NULL`,
     {
       type: QueryTypes.SELECT,
       replacements: {
@@ -83,6 +84,9 @@ export const checkActiveWorkflows: CheckFunction = async (
 
     const missingActiveWorkflows: any[] = [];
     for (const connector of connectors) {
+      if (connector.pausedAt) {
+        continue;
+      }
       heartbeat();
 
       const isActive = await areTemporalWorkflowsRunning(

diff --git a/front/production_checks/checks/check_notion_active_workflows.ts b/front/production_checks/checks/check_notion_active_workflows.ts
@@ -10,13 +10,14 @@ interface NotionConnector {
   id: number;
   dataSourceName: string;
   workspaceId: string;
+  pausedAt: Date | null;
 }
 
 async function listAllNotionConnectors() {
   const connectorsReplica = getConnectorReplicaDbConnection();
 
   const notionConnectors: NotionConnector[] = await connectorsReplica.query(
-    `SELECT id, "dataSourceName", "workspaceId" FROM connectors WHERE "type" = 'notion' and  "errorType" IS NULL`,
+    `SELECT id, "dataSourceName", "workspaceId", "pausedAt" FROM connectors WHERE "type" = 'notion' and  "errorType" IS NULL`,
     {
       type: QueryTypes.SELECT,
     }
@@ -63,6 +64,9 @@ export const checkNotionActiveWorkflows: CheckFunction = async (
 
   const missingActiveWorkflows: any[] = [];
   for (const notionConnector of notionConnectors) {
+    if (notionConnector.pausedAt) {
+      continue;
+    }
     heartbeat();
 
     const isActive = await areTemporalWorkflowsRunning(client, notionConnector);