Fix godmode publishall (#1364)

* refactor(site.router): implement background processing for site publishing

* fix(publishing): update success toast message to reflect site count during background publishing

Author: adriangohjw

apps/studio/src/pages/godmode/publishing.tsx

@@ -64,9 +64,9 @@ const GodModePublishingPage: NextPageWithLayout = () => {
 
   const { mutate: publishAllSite, isLoading: isPublishingAllSite } =
     trpc.site.publishAll.useMutation({
-      onSuccess: () => {
+      onSuccess: ({ siteCount }) => {
         toast({
-          title: "All sites published successfully",
+          title: `Starting to publish ${siteCount} sites in the background...`,
           status: "success",
           ...BRIEF_TOAST_SETTINGS,
         })

apps/studio/src/server/modules/site/site.router.ts

@@ -406,19 +406,31 @@ export const siteRouter = router({
       .where("codeBuildId", "is not", null)
       .execute()
 
-    await Promise.all(
-      sites.map((site) =>
-        db.transaction().execute(async (tx) => {
-          await logPublishEvent(tx, {
-            by: byUser,
-            eventType: AuditLogEvent.Publish,
-            delta: { before: null, after: null },
-            metadata: {},
+    // Start background processing to not block the main thread and avoid timeouts
+    void (async () => {
+      // Looping intsead of Promise.all to avoid maxing out Prisma's connection pool
+      for (const site of sites) {
+        try {
+          await db.transaction().execute(async (tx) => {
+            await logPublishEvent(tx, {
+              by: byUser,
+              eventType: AuditLogEvent.Publish,
+              delta: { before: null, after: null },
+              metadata: {},
+              siteId: site.id,
+            })
+            await publishSite(ctx.logger, site.id)
+          })
+        } catch (error) {
+          ctx.logger.error({
+            msg: "Failed to publish site",
             siteId: site.id,
+            error,
           })
-          await publishSite(ctx.logger, site.id)
-        }),
-      ),
-    )
+        }
+      }
+    })()
+
+    return { siteCount: sites.length }
   }),
 })