feat: attach sdk info to chunks

getsentry · Jul 31, 2024 · 0a37791 · 0a37791
1 parent 945cdbc
commit 0a37791
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 2 deletions.
diff --git a/packages/profiling-node/src/integration.ts b/packages/profiling-node/src/integration.ts
@@ -231,7 +231,7 @@ class ContinuousProfiler {
     }
 
     DEBUG_BUILD && logger.log(`[Profiling] Profile chunk ${this._chunkData.id} sent to Sentry.`);
-    const chunk = createProfilingChunkEvent(this._client, this._client.getOptions(), profile, {
+    const chunk = createProfilingChunkEvent(this._client, this._client.getOptions(), profile, this._client.getSdkMetadata()?.sdk, {
       chunk_id: this._chunkData.id,
       trace_id: this._chunkData.startTraceID,
       profiler_id: this._profilerId,

diff --git a/packages/profiling-node/src/utils.ts b/packages/profiling-node/src/utils.ts
@@ -194,12 +194,14 @@ function createProfileChunkPayload(
     trace_id,
     profiler_id,
     chunk_id,
+    sdk,
   }: {
     release: string;
     environment: string;
     trace_id: string | undefined;
     chunk_id: string;
     profiler_id: string;
+    sdk: SdkInfo | undefined;
   },
 ): ProfileChunk {
   // Log a warning if the profile has an invalid traceId (should be uuidv4).
@@ -213,6 +215,10 @@ function createProfileChunkPayload(
 
   const profile: ProfileChunk = {
     chunk_id: chunk_id,
+    client_sdk: {
+      name: sdk?.name ?? 'unknown',
+      version: sdk?.version ?? 'unknown',
+    },
     profiler_id: profiler_id,
     platform: 'node',
     version: CONTINUOUS_FORMAT_VERSION,
@@ -235,6 +241,7 @@ export function createProfilingChunkEvent(
   client: Client,
   options: { release?: string; environment?: string },
   profile: RawChunkCpuProfile,
+  sdk: SdkInfo | undefined,
   identifiers: { trace_id: string | undefined; chunk_id: string; profiler_id: string },
 ): ProfileChunk | null {
   if (!isValidProfileChunk(profile)) {
@@ -247,6 +254,7 @@ export function createProfilingChunkEvent(
     trace_id: identifiers.trace_id ?? '',
     chunk_id: identifiers.chunk_id,
     profiler_id: identifiers.profiler_id,
+    sdk
   });
 }
 

diff --git a/packages/profiling-node/test/spanProfileUtils.test.ts b/packages/profiling-node/test/spanProfileUtils.test.ts
@@ -2,7 +2,7 @@ import * as Sentry from '@sentry/node';
 
 import { getMainCarrier } from '@sentry/core';
 import type { NodeClientOptions } from '@sentry/node/build/types/types';
-import type { Transport } from '@sentry/types';
+import type { ProfileChunk, Transport } from '@sentry/types';
 import { GLOBAL_OBJ, createEnvelope, logger } from '@sentry/utils';
 import { CpuProfilerBindings } from '../src/cpu_profiler';
 import { type ProfilingIntegration, _nodeProfilingIntegration } from '../src/integration';
@@ -402,6 +402,51 @@ describe('continuous profiling', () => {
     delete getMainCarrier().__SENTRY__;
   });
 
+  it('attaches sdk metadata to chunks', () => {
+    // @ts-expect-error we just mock the return type and ignore the signature
+    jest.spyOn(CpuProfilerBindings, 'stopProfiling').mockImplementation(() => {
+      return {
+        samples: [
+          {
+            stack_id: 0,
+            thread_id: '0',
+            elapsed_since_start_ns: '10',
+          },
+          {
+            stack_id: 0,
+            thread_id: '0',
+            elapsed_since_start_ns: '10',
+          },
+        ],
+        measurements: {},
+        stacks: [[0]],
+        frames: [],
+        resources: [],
+        profiler_logging_mode: 'lazy',
+      };
+    });
+
+    const [client, transport] = makeContinuousProfilingClient();
+    Sentry.setCurrentClient(client);
+    client.init();
+
+    const transportSpy = jest.spyOn(transport, 'send').mockReturnValue(Promise.resolve({}));
+
+    const integration = client.getIntegrationByName<ProfilingIntegration>('ProfilingIntegration');
+    if (!integration) {
+      throw new Error('Profiling integration not found');
+    }
+    integration._profiler.start();
+    jest.advanceTimersByTime(1000);
+    integration._profiler.stop();
+    jest.advanceTimersByTime(1000);
+
+    const profile = transportSpy.mock.calls?.[0]?.[0]?.[1]?.[0]?.[1] as ProfileChunk;
+    expect(profile.client_sdk.name).toBe('sentry.javascript.node');
+    expect(profile.client_sdk.version).toEqual(expect.stringMatching(/\d+\.\d+\.\d+/));
+  });
+
+
   it('initializes the continuous profiler and binds the sentry client', () => {
     const startProfilingSpy = jest.spyOn(CpuProfilerBindings, 'startProfiling');
 

diff --git a/packages/types/src/profiling.ts b/packages/types/src/profiling.ts
@@ -126,4 +126,8 @@ export interface Profile extends BaseProfile<ThreadCpuProfile> {
 export interface ProfileChunk extends BaseProfile<ContinuousThreadCpuProfile> {
   chunk_id: string;
   profiler_id: string;
+  client_sdk: {
+    name: string;
+    version: string;
+  };
 }