feat: disable subscriptions

uselagoon · Jul 4, 2024 · 88664a7 · 88664a7
1 parent 2be4445
commit 88664a7
Show file tree

Hide file tree

Showing 5 changed files with 126 additions and 106 deletions.
diff --git a/src/lib/ApiConnection.js b/src/lib/ApiConnection.js
@@ -16,6 +16,8 @@ import ErrorPage from 'pages/_error.js';
 
 const { serverRuntimeConfig, publicRuntimeConfig } = getConfig();
 
+const disableSubscriptions = publicRuntimeConfig.DISABLE_SUBSCRIPTIONS?.toLowerCase() === 'true';
+
 const ApiConnection = ({ children }) => (
   <AuthContext.Consumer>
     {auth => {
@@ -34,13 +36,13 @@ const ApiConnection = ({ children }) => (
         const wsLink = new WebSocketLink({
           uri: publicRuntimeConfig.GRAPHQL_API.replace(/https/, 'wss').replace(/http/, 'ws'),
           options: {
+            lazy: disableSubscriptions,
             reconnect: true,
             connectionParams: {
               authToken: auth.apiToken,
             },
           },
         });
-
         return ApolloLink.split(
           ({ query }) => {
             const { kind, operation } = getMainDefinition(query);

diff --git a/src/next.config.js b/src/next.config.js
@@ -32,6 +32,7 @@ module.exports = {
     LAGOON_UI_VIEW_ENV_VARIABLES: process.env.LAGOON_UI_VIEW_ENV_VARIABLES,
     PLUGIN_SCRIPTS: pluginRegistry,
     WEBHOOK_URL: process.env.WEBHOOK_URL,
+    DISABLE_SUBSCRIPTIONS: process.env.DISABLE_SUBSCRIPTIONS,
   },
   distDir: '../build',
   webpack(config, options) {

diff --git a/src/pages/backups.js b/src/pages/backups.js
@@ -27,6 +27,8 @@ const { publicRuntimeConfig } = getConfig();
 const envLimit = parseInt(publicRuntimeConfig.LAGOON_UI_BACKUPS_LIMIT, 10);
 const customMessage = publicRuntimeConfig.LAGOON_UI_BACKUPS_LIMIT_MESSAGE;
 
+const disableSubscriptions = publicRuntimeConfig.DISABLE_SUBSCRIPTIONS?.toLowerCase() === 'true';
+
 /**
  * Displays the backups page, given the name of an openshift project.
  */
@@ -117,50 +119,52 @@ export const PageBackups = ({ router }) => {
     );
   }
 
-  subscribeToMore({
-    document: BackupsSubscription,
-    variables: { environment: environment.id },
-    updateQuery: (prevStore, { subscriptionData }) => {
-      if (!subscriptionData.data) return prevStore;
-      const prevBackups = prevStore.environment.backups;
-      const incomingBackup = subscriptionData.data.backupChanged;
-      const existingIndex = prevBackups.findIndex(prevBackup => prevBackup.id === incomingBackup.id);
-      let newBackups;
-
-      // New backup.
-      if (existingIndex === -1) {
-        // Don't add new deleted backups.
-        if (incomingBackup.deleted !== '0000-00-00 00:00:00') {
-          return prevStore;
+  if (!disableSubscriptions) {
+    subscribeToMore({
+      document: BackupsSubscription,
+      variables: { environment: environment.id },
+      updateQuery: (prevStore, { subscriptionData }) => {
+        if (!subscriptionData.data) return prevStore;
+        const prevBackups = prevStore.environment.backups;
+        const incomingBackup = subscriptionData.data.backupChanged;
+        const existingIndex = prevBackups.findIndex(prevBackup => prevBackup.id === incomingBackup.id);
+        let newBackups;
+
+        // New backup.
+        if (existingIndex === -1) {
+          // Don't add new deleted backups.
+          if (incomingBackup.deleted !== '0000-00-00 00:00:00') {
+            return prevStore;
+          }
+
+          newBackups = [incomingBackup, ...prevBackups];
         }
-
-        newBackups = [incomingBackup, ...prevBackups];
-      }
-      // Existing backup.
-      else {
-        // Updated backup
-        if (incomingBackup.deleted === '0000-00-00 00:00:00') {
-          newBackups = Object.assign([...prevBackups], {
-            [existingIndex]: incomingBackup,
-          });
-        }
-        // Deleted backup
+        // Existing backup.
         else {
-          newBackups = R.remove(existingIndex, 1, prevBackups);
+          // Updated backup
+          if (incomingBackup.deleted === '0000-00-00 00:00:00') {
+            newBackups = Object.assign([...prevBackups], {
+              [existingIndex]: incomingBackup,
+            });
+          }
+          // Deleted backup
+          else {
+            newBackups = R.remove(existingIndex, 1, prevBackups);
+          }
         }
-      }
 
-      const newStore = {
-        ...prevStore,
-        environment: {
-          ...prevStore.environment,
-          backups: newBackups,
-        },
-      };
-
-      return newStore;
-    },
-  });
+        const newStore = {
+          ...prevStore,
+          environment: {
+            ...prevStore.environment,
+            backups: newBackups,
+          },
+        };
+
+        return newStore;
+      },
+    });
+  }
 
   return (
     <>

diff --git a/src/pages/deployments.js b/src/pages/deployments.js
@@ -28,6 +28,8 @@ const { publicRuntimeConfig } = getConfig();
 const envLimit = parseInt(publicRuntimeConfig.LAGOON_UI_DEPLOYMENTS_LIMIT, 10);
 const customMessage = publicRuntimeConfig.LAGOON_UI_DEPLOYMENTS_LIMIT_MESSAGE;
 
+const disableSubscriptions = publicRuntimeConfig.DISABLE_SUBSCRIPTIONS?.toLowerCase() === 'true';
+
 /**
  * Displays the deployments page, given the openshift project name.
  */
@@ -44,7 +46,10 @@ export const PageDeployments = ({ router }) => {
   });
 
   const handleRefetch = async () =>
-    await refetch({ openshiftProjectName: router.query.openshiftProjectName, limit: resultLimit });
+    await refetch({
+      openshiftProjectName: router.query.openshiftProjectName,
+      limit: resultLimit,
+    });
 
   useEffect(() => {
     let urlResultLimit = envLimit;
@@ -118,39 +123,40 @@ export const PageDeployments = ({ router }) => {
       />
     );
   }
+  if (!disableSubscriptions) {
+    subscribeToMore({
+      document: DeploymentsSubscription,
+      variables: { environment: environment.id },
+      updateQuery: (prevStore, { subscriptionData }) => {
+        if (!subscriptionData.data) return prevStore;
+        const prevDeployments = prevStore.environment.deployments;
+        const incomingDeployment = subscriptionData.data.deploymentChanged;
+        const existingIndex = prevDeployments.findIndex(prevDeployment => prevDeployment.id === incomingDeployment.id);
+        let newDeployments;
+
+        // New deployment.
+        if (existingIndex === -1) {
+          newDeployments = [incomingDeployment, ...prevDeployments];
+        }
+        // Updated deployment
+        else {
+          newDeployments = Object.assign([...prevDeployments], {
+            [existingIndex]: incomingDeployment,
+          });
+        }
 
-  subscribeToMore({
-    document: DeploymentsSubscription,
-    variables: { environment: environment.id },
-    updateQuery: (prevStore, { subscriptionData }) => {
-      if (!subscriptionData.data) return prevStore;
-      const prevDeployments = prevStore.environment.deployments;
-      const incomingDeployment = subscriptionData.data.deploymentChanged;
-      const existingIndex = prevDeployments.findIndex(prevDeployment => prevDeployment.id === incomingDeployment.id);
-      let newDeployments;
-
-      // New deployment.
-      if (existingIndex === -1) {
-        newDeployments = [incomingDeployment, ...prevDeployments];
-      }
-      // Updated deployment
-      else {
-        newDeployments = Object.assign([...prevDeployments], {
-          [existingIndex]: incomingDeployment,
-        });
-      }
-
-      const newStore = {
-        ...prevStore,
-        environment: {
-          ...prevStore.environment,
-          deployments: newDeployments,
-        },
-      };
-
-      return newStore;
-    },
-  });
+        const newStore = {
+          ...prevStore,
+          environment: {
+            ...prevStore.environment,
+            deployments: newDeployments,
+          },
+        };
+
+        return newStore;
+      },
+    });
+  }
 
   return (
     <>

diff --git a/src/pages/tasks.js b/src/pages/tasks.js
@@ -28,6 +28,8 @@ const { publicRuntimeConfig } = getConfig();
 const envLimit = parseInt(publicRuntimeConfig.LAGOON_UI_TASKS_LIMIT, 10);
 const customMessage = publicRuntimeConfig.AGOON_UI_TASKS_LIMIT_MESSAGE;
 
+const disableSubscriptions = publicRuntimeConfig.DISABLE_SUBSCRIPTIONS?.toLowerCase() === 'true';
+
 /**
  * Displays the tasks page, given the openshift project name.
  */
@@ -43,7 +45,10 @@ export const PageTasks = ({ router, renderAddTasks }) => {
   });
 
   const handleRefetch = async () =>
-    await refetch({ openshiftProjectName: router.query.openshiftProjectName, limit: resultLimit });
+    await refetch({
+      openshiftProjectName: router.query.openshiftProjectName,
+      limit: resultLimit,
+    });
 
   useEffect(() => {
     let urlResultLimit = envLimit;
@@ -120,38 +125,40 @@ export const PageTasks = ({ router, renderAddTasks }) => {
     );
   }
 
-  subscribeToMore({
-    document: TasksSubscription,
-    variables: { environment: environment.id },
-    updateQuery: (prevStore, { subscriptionData }) => {
-      if (!subscriptionData.data) return prevStore;
-      const prevTasks = prevStore.environment.tasks;
-      const incomingTask = subscriptionData.data.taskChanged;
-      const existingIndex = prevTasks.findIndex(prevTask => prevTask.id === incomingTask.id);
-      let newTasks;
-
-      // New task.
-      if (existingIndex === -1) {
-        newTasks = [incomingTask, ...prevTasks];
-      }
-      // Updated task
-      else {
-        newTasks = Object.assign([...prevTasks], {
-          [existingIndex]: incomingTask,
-        });
-      }
-
-      const newStore = {
-        ...prevStore,
-        environment: {
-          ...prevStore.environment,
-          tasks: newTasks,
-        },
-      };
+  if (!disableSubscriptions) {
+    subscribeToMore({
+      document: TasksSubscription,
+      variables: { environment: environment.id },
+      updateQuery: (prevStore, { subscriptionData }) => {
+        if (!subscriptionData.data) return prevStore;
+        const prevTasks = prevStore.environment.tasks;
+        const incomingTask = subscriptionData.data.taskChanged;
+        const existingIndex = prevTasks.findIndex(prevTask => prevTask.id === incomingTask.id);
+        let newTasks;
+
+        // New task.
+        if (existingIndex === -1) {
+          newTasks = [incomingTask, ...prevTasks];
+        }
+        // Updated task
+        else {
+          newTasks = Object.assign([...prevTasks], {
+            [existingIndex]: incomingTask,
+          });
+        }
 
-      return newStore;
-    },
-  });
+        const newStore = {
+          ...prevStore,
+          environment: {
+            ...prevStore.environment,
+            tasks: newTasks,
+          },
+        };
+
+        return newStore;
+      },
+    });
+  }
 
   return (
     <>