From e430d57d33f0fff5a9ef9681032f34a3ea7c9b63 Mon Sep 17 00:00:00 2001
From: Jiachi Liu <inbox@huozhi.im>
Date: Tue, 14 May 2024 17:17:47 +0200
Subject: [PATCH] use array to filer the injected metadata

---
 packages/next/src/export/index.ts             |  2 +-
 .../next/src/server/app-render/app-render.tsx | 18 ++++---
 .../make-get-server-inserted-html.tsx         | 31 ++++++++----
 packages/next/src/server/app-render/types.ts  |  2 +-
 packages/next/src/server/base-server.ts       |  3 +-
 packages/next/src/server/config-schema.ts     |  2 +-
 packages/next/src/server/config-shared.ts     |  6 +--
 .../client-trace-metadata.test.ts             | 47 ++++++++-----------
 .../client-trace-metadata/next.config.js      | 10 ++++
 9 files changed, 69 insertions(+), 52 deletions(-)
 create mode 100644 test/e2e/opentelemetry/client-trace-metadata/next.config.js

diff --git a/packages/next/src/export/index.ts b/packages/next/src/export/index.ts
index b667808c6c8ec..b52db8cd6d3dd 100644
--- a/packages/next/src/export/index.ts
+++ b/packages/next/src/export/index.ts
@@ -420,8 +420,8 @@ export async function exportAppImpl(
     deploymentId: nextConfig.deploymentId,
     experimental: {
       isAppPPREnabled: checkIsAppPPREnabled(nextConfig.experimental.ppr),
+      clientTraceMetadata: nextConfig.experimental.clientTraceMetadata,
       swrDelta: nextConfig.experimental.swrDelta,
-      clientTraceMetadata: nextConfig.experimental.clientTraceMetadata === true,
     },
   }
 
diff --git a/packages/next/src/server/app-render/app-render.tsx b/packages/next/src/server/app-render/app-render.tsx
index b5a0a1478edf8..23facb23c4902 100644
--- a/packages/next/src/server/app-render/app-render.tsx
+++ b/packages/next/src/server/app-render/app-render.tsx
@@ -76,7 +76,10 @@ import { appendMutableCookies } from '../web/spec-extension/adapters/request-coo
 import { createServerInsertedHTML } from './server-inserted-html'
 import { getRequiredScripts } from './required-scripts'
 import { addPathPrefix } from '../../shared/lib/router/utils/add-path-prefix'
-import { makeGetServerInsertedHTML } from './make-get-server-inserted-html'
+import {
+  getTracedMetadata,
+  makeGetServerInsertedHTML,
+} from './make-get-server-inserted-html'
 import { walkTreeWithFlightRouterState } from './walk-tree-with-flight-router-state'
 import { createComponentTree } from './create-component-tree'
 import { getAssetQueryString } from './get-asset-query-string'
@@ -912,6 +915,11 @@ async function renderToHTMLOrFlightImpl(
       tree,
       formState,
     }: RenderToStreamOptions): Promise<RenderToStreamResult> => {
+      const tracingMetadata = getTracedMetadata(
+        getTracer().getTracePropagationData(),
+        renderOpts.experimental.clientTraceMetadata
+      )
+
       const polyfills: JSX.IntrinsicElements['script'][] =
         buildManifest.polyfillFiles
           .filter(
@@ -995,9 +1003,7 @@ async function renderToHTMLOrFlightImpl(
         renderServerInsertedHTML,
         serverCapturedErrors: allCapturedErrors,
         basePath: renderOpts.basePath,
-        traceData: renderOpts.experimental.clientTraceMetadata
-          ? getTracer().getTracePropagationData()
-          : [],
+        tracingMetadata: tracingMetadata,
       })
 
       const renderer = createStaticRenderer({
@@ -1322,9 +1328,7 @@ async function renderToHTMLOrFlightImpl(
                 renderServerInsertedHTML,
                 serverCapturedErrors: [],
                 basePath: renderOpts.basePath,
-                traceData: renderOpts.experimental.clientTraceMetadata
-                  ? getTracer().getTracePropagationData()
-                  : [],
+                tracingMetadata: tracingMetadata,
               }),
               serverInsertedHTMLToHead: true,
               validateRootLayout,
diff --git a/packages/next/src/server/app-render/make-get-server-inserted-html.tsx b/packages/next/src/server/app-render/make-get-server-inserted-html.tsx
index 8571cf949b3c9..624b1df8d7ac7 100644
--- a/packages/next/src/server/app-render/make-get-server-inserted-html.tsx
+++ b/packages/next/src/server/app-render/make-get-server-inserted-html.tsx
@@ -11,16 +11,24 @@ import { RedirectStatusCode } from '../../client/components/redirect-status-code
 import { addPathPrefix } from '../../shared/lib/router/utils/add-path-prefix'
 import type { ClientTraceDataEntry } from '../lib/trace/tracer'
 
+export function getTracedMetadata(
+  traceData: ClientTraceDataEntry[],
+  clientTraceMetadata: string[] | undefined
+): ClientTraceDataEntry[] | undefined {
+  if (!clientTraceMetadata) return undefined
+  return traceData.filter(({ key }) => clientTraceMetadata.includes(key))
+}
+
 export function makeGetServerInsertedHTML({
   polyfills,
   renderServerInsertedHTML,
   serverCapturedErrors,
-  traceData,
+  tracingMetadata,
   basePath,
 }: {
   polyfills: JSX.IntrinsicElements['script'][]
   renderServerInsertedHTML: () => React.ReactNode
-  traceData: ClientTraceDataEntry[]
+  tracingMetadata: ClientTraceDataEntry[] | undefined
   serverCapturedErrors: Error[]
   basePath: string
 }) {
@@ -84,14 +92,17 @@ export function makeGetServerInsertedHTML({
               return <script key={polyfill.src} {...polyfill} />
             })
         }
-        {traceData.map(({ key, value }) => {
-          // The meta-tag name is prefixed with "_next-trace-data-" for mainly two reasons:
-          // - Avoid collisions with meta tags created by the user
-          // - Allow clients that want to propagate _all_ trace data to explicitly query for meta tags prefixed with "_next-trace-data-"
-          return (
-            <meta key={key} name={`_next-trace-data-${key}`} content={value} />
-          )
-        })}
+        {tracingMetadata
+          ? tracingMetadata.map(({ key, value }) => {
+              return (
+                <meta
+                  key={`next-trace-data-${key}:${value}`}
+                  name={key}
+                  content={value}
+                />
+              )
+            })
+          : null}
         {errorMetaTags}
       </>,
       {
diff --git a/packages/next/src/server/app-render/types.ts b/packages/next/src/server/app-render/types.ts
index 91564f8de6081..7709994149763 100644
--- a/packages/next/src/server/app-render/types.ts
+++ b/packages/next/src/server/app-render/types.ts
@@ -170,7 +170,7 @@ export interface RenderOptsPartial {
      */
     isRoutePPREnabled?: boolean
     swrDelta: SwrDelta | undefined
-    clientTraceMetadata: boolean
+    clientTraceMetadata: string[] | undefined
   }
   postponed?: string
   /**
diff --git a/packages/next/src/server/base-server.ts b/packages/next/src/server/base-server.ts
index 88ec35fc504a4..3f66e33d2ea52 100644
--- a/packages/next/src/server/base-server.ts
+++ b/packages/next/src/server/base-server.ts
@@ -556,8 +556,7 @@ export default abstract class Server<
       experimental: {
         isAppPPREnabled,
         swrDelta: this.nextConfig.experimental.swrDelta,
-        clientTraceMetadata:
-          this.nextConfig.experimental.clientTraceMetadata === true,
+        clientTraceMetadata: this.nextConfig.experimental.clientTraceMetadata,
       },
     }
 
diff --git a/packages/next/src/server/config-schema.ts b/packages/next/src/server/config-schema.ts
index dfdac28456479..4c00c7cce8386 100644
--- a/packages/next/src/server/config-schema.ts
+++ b/packages/next/src/server/config-schema.ts
@@ -390,7 +390,7 @@ export const configSchema: zod.ZodType<NextConfig> = z.lazy(() =>
         optimizePackageImports: z.array(z.string()).optional(),
         optimizeServerReact: z.boolean().optional(),
         instrumentationHook: z.boolean().optional(),
-        clientTraceMetadata: z.boolean().optional(),
+        clientTraceMetadata: z.array(z.string()).optional(),
         turbotrace: z
           .object({
             logLevel: z
diff --git a/packages/next/src/server/config-shared.ts b/packages/next/src/server/config-shared.ts
index 8726dee79dc89..418c486e453a3 100644
--- a/packages/next/src/server/config-shared.ts
+++ b/packages/next/src/server/config-shared.ts
@@ -366,9 +366,9 @@ export interface ExperimentalConfig {
   instrumentationHook?: boolean
 
   /**
-   * Enables the propagation of trace data through meta tags to the client.
+   * The array of the meta tags to the client injected by tracing propagation data.
    */
-  clientTraceMetadata?: boolean
+  clientTraceMetadata?: string[]
 
   /**
    * Using this feature will enable the `react@experimental` for the `app` directory.
@@ -924,7 +924,7 @@ export const defaultConfig: NextConfig = {
     turbotrace: undefined,
     typedRoutes: false,
     instrumentationHook: false,
-    clientTraceMetadata: false,
+    clientTraceMetadata: undefined,
     bundlePagesExternals: false,
     parallelServerCompiles: false,
     parallelServerBuildTraces: false,
diff --git a/test/e2e/opentelemetry/client-trace-metadata/client-trace-metadata.test.ts b/test/e2e/opentelemetry/client-trace-metadata/client-trace-metadata.test.ts
index 03ea3bfddff1f..b5b39d6ca6a2c 100644
--- a/test/e2e/opentelemetry/client-trace-metadata/client-trace-metadata.test.ts
+++ b/test/e2e/opentelemetry/client-trace-metadata/client-trace-metadata.test.ts
@@ -1,41 +1,34 @@
 import { nextTestSetup } from 'e2e-utils'
 
-describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
+describe('clientTraceMetadata (ppr=%p)', () => {
   const { next, isNextDev } = nextTestSetup({
     files: __dirname,
     dependencies: require('./package.json').dependencies,
-    nextConfig: {
-      experimental: {
-        instrumentationHook: true,
-        clientTraceMetadata: true,
-        ppr: pprEnabled,
-      },
-    },
   })
 
   it('should inject propagation data for a dynamically server-side-rendered page', async () => {
     const $ = await next.render$('/dynamic-page')
     const headHtml = $.html('head')
     expect(headHtml).toContain(
-      '<meta name="_next-trace-data-my-test-key-1" content="my-test-value-1">'
+      '<meta name="my-test-key-1" content="my-test-value-1">'
     )
     expect($.html('head')).toContain(
-      '<meta name="_next-trace-data-my-test-key-2" content="my-test-value-2">'
+      '<meta name="my-test-key-2" content="my-test-value-2">'
     )
     expect($.html('head')).toMatch(
-      /<meta name="_next-trace-data-my-parent-span-id" content="[a-f0-9]{16}">/
+      /<meta name="my-parent-span-id" content="[a-f0-9]{16}">/
     )
   })
 
   it('hard loading a dynamic page twice should yield different dynamic trace data', async () => {
     const browser1 = await next.browser('/dynamic-page')
     const firstLoadSpanIdContent = await browser1
-      .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+      .elementByCss('meta[name="my-parent-span-id"]')
       .getAttribute('content')
 
     const browser2 = await next.browser('/dynamic-page')
     const secondLoadSpanIdContent = await browser2
-      .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+      .elementByCss('meta[name="my-parent-span-id"]')
       .getAttribute('content')
 
     expect(firstLoadSpanIdContent).toMatch(/[a-f0-9]{16}/)
@@ -49,13 +42,13 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         const $ = await next.render$('/static-page')
         const headHtml = $.html('head')
         expect(headHtml).toContain(
-          '<meta name="_next-trace-data-my-test-key-1" content="my-test-value-1">'
+          '<meta name="my-test-key-1" content="my-test-value-1">'
         )
         expect($.html('head')).toContain(
-          '<meta name="_next-trace-data-my-test-key-2" content="my-test-value-2">'
+          '<meta name="my-test-key-2" content="my-test-value-2">'
         )
         expect($.html('head')).toMatch(
-          /<meta name="_next-trace-data-my-parent-span-id" content="[a-f0-9]{16}">/
+          /<meta name="my-parent-span-id" content="[a-f0-9]{16}">/
         )
       })
 
@@ -63,7 +56,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         const browser = await next.browser('/static-page')
 
         const initialSpanIdTagContent = await browser
-          .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+          .elementByCss('meta[name="my-parent-span-id"]')
           .getAttribute('content')
 
         // We are in dev mode so the static page should contain propagation data
@@ -73,7 +66,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#dynamic-page-header')
 
         const updatedSpanIdTagContent = await browser
-          .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+          .elementByCss('meta[name="my-parent-span-id"]')
           .getAttribute('content')
 
         expect(initialSpanIdTagContent).toBe(updatedSpanIdTagContent)
@@ -83,7 +76,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         const browser = await next.browser('/static-page')
 
         const initialSpanIdTagContent = await browser
-          .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+          .elementByCss('meta[name="my-parent-span-id"]')
           .getAttribute('content')
 
         // We are in dev mode so the static page should contain propagation data
@@ -93,7 +86,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#static-page-2-header')
 
         const updatedSpanIdTagContent = await browser
-          .elementByCss('meta[name="_next-trace-data-my-parent-span-id"]')
+          .elementByCss('meta[name="my-parent-span-id"]')
           .getAttribute('content')
 
         expect(initialSpanIdTagContent).toBe(updatedSpanIdTagContent)
@@ -105,13 +98,13 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         const $ = await next.render$('/static-page')
         const headHtml = $.html('head')
         expect(headHtml).not.toContain(
-          '<meta name="_next-trace-data-my-test-key-1" content="my-test-value-1">'
+          '<meta name="my-test-key-1" content="my-test-value-1">'
         )
         expect($.html('head')).not.toContain(
-          '<meta name="_next-trace-data-my-test-key-2" content="my-test-value-2">'
+          '<meta name="my-test-key-2" content="my-test-value-2">'
         )
         expect($.html('head')).not.toMatch(
-          /<meta name="_next-trace-data-my-parent-span-id" content="[a-f0-9]{16}">/
+          /<meta name="my-parent-span-id" content="[a-f0-9]{16}">/
         )
       })
 
@@ -121,7 +114,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#static-page-header')
 
         const initialSpanIdTag = await browser.eval(
-          'document.querySelector(\'meta[name="_next-trace-data-my-parent-span-id"]\')'
+          'document.querySelector(\'meta[name="my-parent-span-id"]\')'
         )
 
         // We are in prod mode so we are not expecting propagation data to be present for a static page
@@ -131,7 +124,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#dynamic-page-header')
 
         const updatedSpanIdTag = await browser.eval(
-          'document.querySelector(\'meta[name="_next-trace-data-my-parent-span-id"]\')'
+          'document.querySelector(\'meta[name="my-parent-span-id"]\')'
         )
 
         // After the navigation to the dynamic page, there should still be no meta tag with propagation data
@@ -144,7 +137,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#static-page-header')
 
         const initialSpanIdTag = await browser.eval(
-          'document.querySelector(\'meta[name="_next-trace-data-my-parent-span-id"]\')'
+          'document.querySelector(\'meta[name="my-parent-span-id"]\')'
         )
 
         // We are in prod mode so we are not expecting propagation data to be present for a static page
@@ -154,7 +147,7 @@ describe.each([false, true])('clientTraceMetadata (ppr=%p)', (pprEnabled) => {
         await browser.elementByCss('#static-page-2-header')
 
         const updatedSpanIdTag = await browser.eval(
-          'document.querySelector(\'meta[name="_next-trace-data-my-parent-span-id"]\')'
+          'document.querySelector(\'meta[name="my-parent-span-id"]\')'
         )
 
         // After the navigation to the dynamic page, there should still be no meta tag with propagation data
diff --git a/test/e2e/opentelemetry/client-trace-metadata/next.config.js b/test/e2e/opentelemetry/client-trace-metadata/next.config.js
new file mode 100644
index 0000000000000..529a5ed29d0cf
--- /dev/null
+++ b/test/e2e/opentelemetry/client-trace-metadata/next.config.js
@@ -0,0 +1,10 @@
+module.exports = {
+  experimental: {
+    instrumentationHook: true,
+    clientTraceMetadata: [
+      'my-parent-span-id',
+      'my-test-key-1',
+      'my-test-key-2',
+    ],
+  },
+}