add caching

Author: s1gr1d

packages/nuxt/src/runtime/utils/route-extraction.ts

@@ -2,6 +2,30 @@ import { logger } from '@sentry/core';
 import type { NuxtSSRContext } from 'nuxt/app';
 import type { NuxtPage } from 'nuxt/schema';
 
+export type NuxtPageSubset = { path: NuxtPage['path']; file: NuxtPage['file'] };
+
+// buildTimePagesData -> Map<file, path>
+const routeMappingCache = new WeakMap<NuxtPageSubset[], Map<string, string>>();
+
+/**
+ * Creates a mapping from module paths to parametrized routes for faster lookup.
+ * The `buildTimePagesData` doesn't change during SSR, as it's generated at build time.
+ */
+function createRouteMapping(buildTimePagesData: NuxtPageSubset[]): Map<string, string> {
+  const mapping = new Map<string, string>();
+
+  for (const routeData of buildTimePagesData) {
+    if (routeData.file && routeData.path) {
+      // Handle Unix and Windows paths
+      const normalizedFile = routeData.file.replace(/\\/g, '/');
+      mapping.set(normalizedFile, routeData.path);
+    }
+  }
+
+  routeMappingCache.set(buildTimePagesData, mapping);
+  return mapping;
+}
+
 /**
  * Extracts route information from the SSR context modules and URL.
  *
@@ -17,12 +41,12 @@ import type { NuxtPage } from 'nuxt/schema';
  *
  * @param buildTimePagesData
  * An array of NuxtPage objects representing the build-time pages data.
- * Example: [{ name: 'some-path', path: '/some/path' }, { name: 'user-userId', path: '/user/:userId()' }]
+ * Example: [{ file: '/a/file/pages/some/path', path: '/some/path' }, { file: '/a/file/pages/user/[userId].vue', path: '/user/:userId()' }]
  */
 export function extractParametrizedRouteFromContext(
   ssrContextModules?: NuxtSSRContext['modules'],
   currentUrl?: NuxtSSRContext['url'],
-  buildTimePagesData: NuxtPage[] = [],
+  buildTimePagesData: NuxtPageSubset[] = [],
 ): null | { parametrizedRoute: string } {
   if (!ssrContextModules || !currentUrl) {
     logger.warn('SSR context modules or URL is not available.');
@@ -35,30 +59,24 @@ export function extractParametrizedRouteFromContext(
 
   const modulesArray = Array.from(ssrContextModules);
 
-  // Find the route data that corresponds to a module in ssrContext.modules
-  const foundRouteData = buildTimePagesData.find(routeData => {
-    if (!routeData.file) return false;
-
-    return modulesArray.some(module => {
-      // Extract the folder name and relative path from the page file
-      // e.g., 'pages/test-param/[param].vue' -> folder: 'pages', path: 'test-param/[param].vue'
-      const filePathParts = module.split('/');
+  const existingRouteMapping = routeMappingCache.get(buildTimePagesData);
+  const routeMapping = existingRouteMapping || createRouteMapping(buildTimePagesData);
 
-      // Exclude root-level files (e.g., 'app.vue')
-      if (filePathParts.length < 2) return false;
+  const modulePagePaths = modulesArray.map(module => {
+    const filePathParts = module.split('/');
+    if (filePathParts.length < 2) return null;
 
-      // Normalize path separators to handle both Unix and Windows paths
-      const normalizedRouteFile = routeData.file?.replace(/\\/g, '/');
-
-      const pagesFolder = filePathParts[0];
-      const pageRelativePath = filePathParts.slice(1).join('/');
-
-      // Check if any module in ssrContext.modules ends with the same folder/relative path structure
-      return normalizedRouteFile?.endsWith(`/${pagesFolder}/${pageRelativePath}`);
-    });
+    const pagesFolder = filePathParts[0];
+    const pageRelativePath = filePathParts.slice(1).join('/');
+    return `/${pagesFolder}/${pageRelativePath}`;
   });
 
-  const parametrizedRoute = foundRouteData?.path ?? null;
+  for (const [pageFile, route] of routeMapping) {
+    // Check if any module of the requested page ends with the same folder/relative path structure as the parametrized filePath from build time.
+    if (modulePagePaths.some(filePath => filePath && pageFile.endsWith(filePath))) {
+      return { parametrizedRoute: route };
+    }
+  }
 
-  return parametrizedRoute === null ? null : { parametrizedRoute };
+  return null;
 }