perf: optimize GraphQL route and add domain caching

- Wrap data-fetching functions (getPage, getSiteInfo, getFullSiteSetup) with React.cache() to deduplicate SSR calls
- Add in-memory domain cache with 60s TTL (domain-cache.ts)
- Use getCachedDomain in verify-domain route and GraphQL route handler
- Parallelize domain lookup, session check, and body parsing with Promise.all in GraphQL route
- Store plain objects in domain cache, hydrate fresh Mongoose docs per request to prevent cross-request mutation

Author: Rajat Saxena

apps/web/app/api/graph/route.ts

@@ -3,9 +3,9 @@ import schema from "@/graphql";
 import { graphql } from "graphql";
 import { getAddress } from "@/lib/utils";
 import User from "@models/User";
-import DomainModel, { Domain } from "@models/Domain";
 import { auth } from "@/auth";
 import { als } from "@/async-local-storage";
+import { getCachedDomain } from "@/lib/domain-cache";
 
 async function updateLastActive(user: any) {
     const dateNow = new Date();
@@ -20,26 +20,25 @@ async function updateLastActive(user: any) {
 }
 
 export async function POST(req: NextRequest) {
-    const domain = await DomainModel.findOne<Domain>({
-        name: req.headers.get("domain"),
-    });
+    const [domain, session, body] = await Promise.all([
+        getCachedDomain(req.headers.get("domain")!),
+        auth.api.getSession({ headers: req.headers }),
+        req.json(),
+    ]);
+
     if (!domain) {
         return Response.json({ message: "Domain not found" }, { status: 404 });
     }
 
+    if (!body.hasOwnProperty("query")) {
+        return Response.json({ error: "Query is missing" }, { status: 400 });
+    }
+
     const map = new Map();
     map.set("domain", req.headers.get("domain"));
     map.set("domainId", req.headers.get("domainId"));
     als.enterWith(map);
 
-    const session = await auth.api.getSession({
-        headers: req.headers,
-    });
-    const body = await req.json();
-    if (!body.hasOwnProperty("query")) {
-        return Response.json({ error: "Query is missing" }, { status: 400 });
-    }
-
     let user;
     if (session) {
         user = await User.findOne({
@@ -53,11 +52,6 @@ export async function POST(req: NextRequest) {
         }
     }
 
-    // const body = await req.json();
-    // if (!body.hasOwnProperty("query")) {
-    //     return Response.json({ error: "Query is missing" }, { status: 400 });
-    // }
-
     let query, variables;
     if (typeof body.query === "string") {
         query = body.query;

apps/web/graphql/pages/logic.ts

@@ -1,4 +1,5 @@
 import { responses } from "../../config/strings";
+import DomainModel from "@models/Domain";
 import { checkIfAuthenticated } from "../../lib/graphql";
 import GQLContext from "../../models/GQLContext";
 import PageModel, { Page } from "../../models/Page";
@@ -280,15 +281,19 @@ export const publish = async (
     }
     page.socialImage = page.draftSocialImage;
 
-    ctx.subdomain.typefaces = ctx.subdomain.draftTypefaces;
-    ctx.subdomain.sharedWidgets = ctx.subdomain.draftSharedWidgets;
-    // ctx.subdomain.draftSharedWidgets = {};
-
     if (ctx.subdomain.themeId) {
         await publishTheme(ctx.subdomain.themeId, ctx);
     }
 
-    await (ctx.subdomain as any).save();
+    await DomainModel.findOneAndUpdate(
+        { _id: ctx.subdomain._id },
+        {
+            $set: {
+                typefaces: ctx.subdomain.draftTypefaces,
+                sharedWidgets: ctx.subdomain.draftSharedWidgets,
+            },
+        },
+    );
     for (const mediaId of mediaToDelete) {
         await deleteMedia(mediaId);
     }

apps/web/graphql/themes/logic.ts

@@ -1,4 +1,5 @@
 import { checkIfAuthenticated } from "../../lib/graphql";
+import DomainModel from "@models/Domain";
 import { responses } from "../../config/strings";
 import constants from "../../config/constants";
 import GQLContext from "../../models/GQLContext";
@@ -140,8 +141,10 @@ export const updateDraftTheme = async (
     }
 
     await theme.save();
-    ctx.subdomain.lastEditedThemeId = theme.themeId;
-    await (ctx.subdomain as any).save();
+    await DomainModel.findOneAndUpdate(
+        { _id: ctx.subdomain._id },
+        { $set: { lastEditedThemeId: theme.themeId } },
+    );
 
     return formatTheme(theme);
 };
@@ -216,9 +219,10 @@ export const switchTheme = async (themeId: string, ctx: GQLContext) => {
         theme = await publishTheme(themeId, ctx);
     }
 
-    ctx.subdomain.themeId = themeId;
-    ctx.subdomain.lastEditedThemeId = themeId;
-    await (ctx.subdomain as any).save();
+    await DomainModel.findOneAndUpdate(
+        { _id: ctx.subdomain._id },
+        { $set: { themeId, lastEditedThemeId: themeId } },
+    );
 
     return formatTheme(theme);
 };

apps/web/lib/domain-cache.ts

@@ -2,7 +2,7 @@ import DomainModel, { Domain } from "../models/Domain";
 
 const domainCache = new Map<
     string,
-    { data: Domain | null; expiresAt: number }
+    { data: Record<string, any>; expiresAt: number }
 >();
 const TTL = 60_000; // 60 seconds
 
@@ -35,11 +35,16 @@ export async function getCachedDomain(
 ): Promise<Domain | null> {
     const cached = domainCache.get(hostName);
     if (cached && cached.expiresAt > Date.now()) {
-        return cached.data;
+        return DomainModel.hydrate(cached.data) as unknown as Domain;
     }
 
     const domain = await getDomain(hostName);
-    domainCache.set(hostName, { data: domain, expiresAt: Date.now() + TTL });
+    if (domain) {
+        domainCache.set(hostName, {
+            data: (domain as any).toObject(),
+            expiresAt: Date.now() + TTL,
+        });
+    }
     return domain;
 }