chore: telemery span filtering

Author: mshanemc

packages/salesforcedx-vscode-services/src/observability/spanTransformProcessor.ts

@@ -10,7 +10,7 @@ import * as Effect from 'effect/Effect';
 import * as SubscriptionRef from 'effect/SubscriptionRef';
 import * as os from 'node:os';
 import { env, UIKind, version, workspace } from 'vscode';
-import {  getDefaultOrgRef } from '../core/defaultOrgRef';
+import { getDefaultOrgRef } from '../core/defaultOrgRef';
 
 /** Custom span processor that transforms spans before they're exported */
 export class SpanTransformProcessor extends BatchSpanProcessor {

packages/salesforcedx-vscode-services/src/observability/spanUtils.ts

@@ -10,6 +10,9 @@ import { ReadableSpan } from '@opentelemetry/sdk-trace-base';
 /** Check if a span is a top-level span (has no parent) */
 export const isTopLevelSpan = (span: ReadableSpan): boolean => span.parentSpanContext === undefined;
 
+/** Check if a span has a command attribute */
+export const isCommandSpan = (span: ReadableSpan): boolean => span.attributes['command'] !== undefined;
+
 /** Convert span attributes to string key-value pairs, filtering out undefined/null values */
 export const convertAttributes = (attributes: Attributes): Attributes =>
   Object.fromEntries(

packages/salesforcedx-vscode-services/src/observability/spansNode.ts

@@ -7,15 +7,27 @@
 import type { SdkLayerConfig } from './sdkLayerConfig';
 import { AzureMonitorTraceExporter } from '@azure/monitor-opentelemetry-exporter';
 import { NodeSdk } from '@effect/opentelemetry';
+import type { ExportResult } from '@opentelemetry/core';
 import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
-import { ConsoleSpanExporter } from '@opentelemetry/sdk-trace-base';
+import { ConsoleSpanExporter, type ReadableSpan } from '@opentelemetry/sdk-trace-base';
 import { Global } from '@salesforce/core/global';
 import { join } from 'node:path';
 import { DEFAULT_AI_CONNECTION_STRING, isTelemetryExtensionConfigurationEnabled } from './appInsights';
 import { FileSpanExporterNode } from './fileSpanExporterNode';
 import { getConsoleTracesEnabled, getFileTracesEnabled, getLocalTracesEnabled } from './localTracing';
 import { O11ySpanExporter } from './o11ySpanExporter';
 import { SpanTransformProcessor } from './spanTransformProcessor';
+import { isCommandSpan, isTopLevelSpan } from './spanUtils';
+
+class FilteredAzureMonitorTraceExporter extends AzureMonitorTraceExporter {
+  public override async export(
+    spans: ReadableSpan[],
+    resultCallback: (result: ExportResult) => void
+  ): Promise<void> {
+    const filteredSpans = spans.filter(span => isTopLevelSpan(span) || isCommandSpan(span));
+    return super.export(filteredSpans, resultCallback);
+  }
+}
 
 export const NodeSdkLayerFor = ({ extensionName, extensionVersion, o11yEndpoint, productFeatureId }: SdkLayerConfig) =>
   NodeSdk.layer(() => ({
@@ -33,7 +45,7 @@ export const NodeSdkLayerFor = ({ extensionName, extensionVersion, o11yEndpoint,
       ...(isTelemetryExtensionConfigurationEnabled()
         ? [
             new SpanTransformProcessor(
-              new AzureMonitorTraceExporter({
+              new FilteredAzureMonitorTraceExporter({
                 connectionString: DEFAULT_AI_CONNECTION_STRING,
                 storageDirectory: join(Global.SF_DIR, 'vscode-extensions-telemetry')
               }),