feat(v9/core): add MCP server instrumentation (#17196)

backport of #16817

Co-authored-by: Miguel Betegón <[email protected]>

Author: AbhiPrasad

dev-packages/e2e-tests/test-applications/node-express/src/app.ts

@@ -25,6 +25,8 @@ import { mcpRouter } from './mcp';
 const app = express();
 const port = 3030;
 
+app.use(express.json());
+
 app.use(mcpRouter);
 
 app.get('/crash-in-with-monitor/:id', async (req, res) => {

dev-packages/e2e-tests/test-applications/node-express/src/mcp.ts

@@ -55,7 +55,7 @@ mcpRouter.post('/messages', async (req, res) => {
   const sessionId = req.query.sessionId;
   const transport = transports[sessionId as string];
   if (transport) {
-    await transport.handlePostMessage(req, res);
+    await transport.handlePostMessage(req, res, req.body);
   } else {
     res.status(400).send('No transport found for sessionId');
   }

dev-packages/e2e-tests/test-applications/node-express/tests/errors.test.ts

@@ -13,7 +13,7 @@ test('Sends correct error event', async ({ baseURL }) => {
   expect(errorEvent.exception?.values).toHaveLength(1);
   expect(errorEvent.exception?.values?.[0]?.value).toBe('This is an exception with id 123');
 
-  expect(errorEvent.request).toEqual({
+  expect(errorEvent.request).toMatchObject({
     method: 'GET',
     cookies: {},
     headers: expect.any(Object),
@@ -43,11 +43,11 @@ test('Should record caught exceptions with local variable', async ({ baseURL })
 
 test('To not crash app from withMonitor', async ({ baseURL }) => {
   const doRequest = async (id: number) => {
-    const response = await fetch(`${baseURL}/crash-in-with-monitor/${id}`)
+    const response = await fetch(`${baseURL}/crash-in-with-monitor/${id}`);
     return response.json();
-  }
-  const [response1, response2] = await Promise.all([doRequest(1), doRequest(2)])
-  expect(response1.message).toBe('This is an exception withMonitor: 1')
-  expect(response2.message).toBe('This is an exception withMonitor: 2')
-  expect(response1.pid).toBe(response2.pid) //Just to double-check, TBS
+  };
+  const [response1, response2] = await Promise.all([doRequest(1), doRequest(2)]);
+  expect(response1.message).toBe('This is an exception withMonitor: 1');
+  expect(response2.message).toBe('This is an exception withMonitor: 2');
+  expect(response1.pid).toBe(response2.pid); //Just to double-check, TBS
 });

dev-packages/e2e-tests/test-applications/node-express/tests/mcp.test.ts

@@ -18,7 +18,7 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
       return transactionEvent.transaction === 'POST /messages';
     });
     const toolTransactionPromise = waitForTransaction('node-express', transactionEvent => {
-      return transactionEvent.transaction === 'mcp-server/tool:echo';
+      return transactionEvent.transaction === 'tools/call echo';
     });
 
     const toolResult = await client.callTool({
@@ -39,10 +39,12 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
 
     const postTransaction = await postTransactionPromise;
     expect(postTransaction).toBeDefined();
+    expect(postTransaction.contexts?.trace?.op).toEqual('http.server');
 
     const toolTransaction = await toolTransactionPromise;
     expect(toolTransaction).toBeDefined();
-
+    expect(toolTransaction.contexts?.trace?.op).toEqual('mcp.server');
+    expect(toolTransaction.contexts?.trace?.data?.['mcp.method.name']).toEqual('tools/call');
     // TODO: When https://github.com/modelcontextprotocol/typescript-sdk/pull/358 is released check for trace id equality between the post transaction and the handler transaction
   });
 
@@ -51,7 +53,7 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
       return transactionEvent.transaction === 'POST /messages';
     });
     const resourceTransactionPromise = waitForTransaction('node-express', transactionEvent => {
-      return transactionEvent.transaction === 'mcp-server/resource:echo';
+      return transactionEvent.transaction === 'resources/read echo://foobar';
     });
 
     const resourceResult = await client.readResource({
@@ -64,10 +66,12 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
 
     const postTransaction = await postTransactionPromise;
     expect(postTransaction).toBeDefined();
+    expect(postTransaction.contexts?.trace?.op).toEqual('http.server');
 
     const resourceTransaction = await resourceTransactionPromise;
     expect(resourceTransaction).toBeDefined();
-
+    expect(resourceTransaction.contexts?.trace?.op).toEqual('mcp.server');
+    expect(resourceTransaction.contexts?.trace?.data?.['mcp.method.name']).toEqual('resources/read');
     // TODO: When https://github.com/modelcontextprotocol/typescript-sdk/pull/358 is released check for trace id equality between the post transaction and the handler transaction
   });
 
@@ -76,7 +80,7 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
       return transactionEvent.transaction === 'POST /messages';
     });
     const promptTransactionPromise = waitForTransaction('node-express', transactionEvent => {
-      return transactionEvent.transaction === 'mcp-server/prompt:echo';
+      return transactionEvent.transaction === 'prompts/get echo';
     });
 
     const promptResult = await client.getPrompt({
@@ -100,10 +104,12 @@ test('Should record transactions for mcp handlers', async ({ baseURL }) => {
 
     const postTransaction = await postTransactionPromise;
     expect(postTransaction).toBeDefined();
+    expect(postTransaction.contexts?.trace?.op).toEqual('http.server');
 
     const promptTransaction = await promptTransactionPromise;
     expect(promptTransaction).toBeDefined();
-
+    expect(promptTransaction.contexts?.trace?.op).toEqual('mcp.server');
+    expect(promptTransaction.contexts?.trace?.data?.['mcp.method.name']).toEqual('prompts/get');
     // TODO: When https://github.com/modelcontextprotocol/typescript-sdk/pull/358 is released check for trace id equality between the post transaction and the handler transaction
   });
 });

packages/core/src/index.ts

@@ -118,7 +118,7 @@ export { featureFlagsIntegration, type FeatureFlagsIntegration } from './integra
 export { profiler } from './profiling';
 export { instrumentFetchRequest } from './fetch';
 export { trpcMiddleware } from './trpc';
-export { wrapMcpServerWithSentry } from './mcp-server';
+export { wrapMcpServerWithSentry } from './integrations/mcp-server';
 export { captureFeedback } from './feedback';
 export type { ReportDialogOptions } from './report-dialog';
 export { _INTERNAL_captureLog, _INTERNAL_flushLogsBuffer, _INTERNAL_captureSerializedLog } from './logs/exports';