parrallelize

dust-tt · Nov 22, 2023 · de15b38 · de15b38
1 parent dc9f91d
commit de15b38
Showing 1 changed file with 38 additions and 29 deletions.
diff --git a/connectors/src/connectors/google_drive/index.ts b/connectors/src/connectors/google_drive/index.ts
@@ -61,35 +61,44 @@ export async function createGoogleDriveConnector(
         const driveClient = await getDriveClient(nangoConnectionId);
 
         // Sanity checks to confirm we have sufficient permissions
-        const sanityCheckAbout = await driveClient.about.get({ fields: "*" });
-        if (sanityCheckAbout.status !== 200) {
-          throw new Error(
-            `Could not get google drive info. Error message: ${
-              sanityCheckAbout.statusText || "unknown"
-            }`
-          );
-        }
-        const sanityCheckFilesGet = await driveClient.files.get({
-          fileId: "root",
-        });
-        if (sanityCheckFilesGet.status !== 200) {
-          throw new Error(
-            `Could not call google drive files get. Error message: ${
-              sanityCheckFilesGet.statusText || "unknown"
-            }`
-          );
-        }
-        const sanityCheckFilesList = await driveClient.drives.list({
-          pageSize: 10,
-          fields: "nextPageToken, drives(id, name)",
-        });
-        if (sanityCheckFilesList.status !== 200) {
-          throw new Error(
-            `Could not call google drive files list. Error message: ${
-              sanityCheckFilesList.statusText || "unknown"
-            }`
-          );
-        }
+        const getAbout = () => driveClient.about.get({ fields: "*" });
+        const getFiles = () => driveClient.files.get({ fileId: "root" });
+        const listFiles = () =>
+          driveClient.drives.list({
+            pageSize: 10,
+            fields: "nextPageToken, drives(id, name)",
+          });
+        Promise.all([getAbout(), getFiles(), listFiles()])
+          .then(
+            ([sanityCheckAbout, sanityCheckFilesGet, sanityCheckFilesList]) => {
+              if (sanityCheckAbout.status !== 200) {
+                throw new Error(
+                  `Could not get google drive info. Error message: ${
+                    sanityCheckAbout.statusText || "unknown"
+                  }`
+                );
+              }
+              if (sanityCheckFilesGet.status !== 200) {
+                throw new Error(
+                  `Could not call google drive files get. Error message: ${
+                    sanityCheckFilesGet.statusText || "unknown"
+                  }`
+                );
+              }
+              if (sanityCheckFilesList.status !== 200) {
+                throw new Error(
+                  `Could not call google drive files list. Error message: ${
+                    sanityCheckFilesList.statusText || "unknown"
+                  }`
+                );
+              }
+            }
+          )
+          .catch(() => {
+            throw new Error(
+              "Error trying to check sufficient permissions from Google."
+            );
+          });
 
         const connector = await Connector.create(
           {