feat: Tracing Integration

HazAT · HazAT · commit 5007b5c4b684 · 2019-04-25T09:19:05.000+02:00
diff --git a/packages/hub/src/spancontext.ts b/packages/hub/src/spancontext.ts
@@ -0,0 +1,18 @@
+import { SpanContext as SpanContextInterface } from '@sentry/types';
+import { uuid4 } from '@sentry/utils';
+
+export class SpanContext implements SpanContextInterface {
+  public constructor(
+    public traceId: string = uuid4(),
+    public spanId: string = uuid4().substring(16),
+    public recorded: boolean = false,
+    public parent?: SpanContext,
+  ) {}
+
+  /**
+   * @inheritDoc
+   */
+  public toTraceparent(): string {
+    return `00-${this.traceId}-${this.spanId}-${this.recorded ? '01' : '00'}`;
+  }
+}
diff --git a/packages/integrations/src/index.ts b/packages/integrations/src/index.ts
@@ -6,5 +6,6 @@ export { Ember } from './ember';
 export { ExtraErrorData } from './extraerrordata';
 export { ReportingObserver } from './reportingobserver';
 export { RewriteFrames } from './rewriteframes';
+export { Tracing } from './tracing';
 export { Transaction } from './transaction';
 export { Vue } from './vue';
diff --git a/packages/integrations/src/tracing.ts b/packages/integrations/src/tracing.ts
@@ -0,0 +1,99 @@
+import { Integration } from '@sentry/types';
+import { fill, getGlobalObject, supportsNativeFetch, uuid4 } from '@sentry/utils';
+
+/**
+ * Session Tracking Integration
+ */
+export class Tracing implements Integration {
+  /**
+   * @inheritDoc
+   */
+  public name: string = Tracing.id;
+
+  /**
+   * @inheritDoc
+   */
+  public static id: string = 'Tracing';
+
+  /**
+   * Constructor for OpenTracingIntegration
+   *
+   * @param traceId Optional TraceId that should be set into the integration.
+   */
+  public constructor(private readonly _traceId: string = uuid4()) {}
+
+  /**
+   * @inheritDoc
+   */
+  public setupOnce(): void {
+    this._traceXHR();
+    this._traceFetch();
+  }
+
+  /**
+   * JSDoc
+   */
+  private _traceXHR(): void {
+    if (!('XMLHttpRequest' in global)) {
+      return;
+    }
+
+    const traceId = this._traceId;
+
+    const xhrproto = XMLHttpRequest.prototype;
+    fill(
+      xhrproto,
+      'send',
+      originalSend =>
+        function(this: XMLHttpRequest, ...args: any[]): void {
+          this.setRequestHeader('sentry-trace', traceId);
+          // tslint:disable-next-line: no-unsafe-any
+          return originalSend.apply(this, args);
+        },
+    );
+  }
+
+  /**
+   * JSDoc
+   */
+  private _traceFetch(): void {
+    if (!supportsNativeFetch()) {
+      return;
+    }
+
+    const traceId = this._traceId;
+
+    // tslint:disable: only-arrow-functions
+    fill(getGlobalObject<Window>(), 'fetch', function(originalFetch: () => void): () => void {
+      return function(...args: any[]): void {
+        const options = args[1] as { [key: string]: any };
+        if (options) {
+          if (options.headers) {
+            options.headers = {
+              ...options.headers,
+              'sentry-trace': traceId,
+            };
+          } else {
+            options.headers = {
+              'sentry-trace': traceId,
+            };
+          }
+        }
+        // tslint:disable-next-line: no-unsafe-any
+        return originalFetch.apply(global, args);
+      };
+    });
+    // tslint:enable: only-arrow-functions
+
+    // fill(
+    //   getGlobalObject<Window>(),
+    //   'fetch',
+    //   originalFetch =>
+    //     function(...args: any[]): void {
+    //       console.log(args);
+    //       // tslint:disable-next-line: no-unsafe-any
+    //       return originalFetch.apply(this, args);
+    //     },
+    // );
+  }
+}
diff --git a/packages/types/src/event.ts b/packages/types/src/event.ts
@@ -3,7 +3,6 @@ import { Exception } from './exception';
 import { Request } from './request';
 import { SdkInfo } from './sdkinfo';
 import { Severity } from './severity';
-import { Span } from './span';
 import { Stacktrace } from './stacktrace';
 import { User } from './user';
 
@@ -33,7 +32,6 @@ export interface Event {
   tags?: { [key: string]: string };
   extra?: { [key: string]: any };
   user?: User;
-  spans?: Span[];
   type?: EventType;
 }
 
diff --git a/packages/types/src/index.ts b/packages/types/src/index.ts
@@ -16,7 +16,7 @@ export { Response } from './response';
 export { Scope } from './scope';
 export { SdkInfo } from './sdkinfo';
 export { Severity } from './severity';
-export { Span } from './span';
+export { SpanContext } from './spancontext';
 export { StackFrame } from './stackframe';
 export { Stacktrace } from './stacktrace';
 export { Status } from './status';
diff --git a/packages/types/src/scope.ts b/packages/types/src/scope.ts
@@ -1,6 +1,7 @@
 import { Breadcrumb } from './breadcrumb';
 import { EventProcessor } from './eventprocessor';
 import { Severity } from './severity';
+import { SpanContext } from './spancontext';
 import { User } from './user';
 
 /**
@@ -63,6 +64,12 @@ export interface Scope {
    */
   setContext(name: string, context: { [key: string]: any } | null): this;
 
+  /**
+   * Sets the SpanContext on the scope.
+   * @param spanContext SpanContext
+   */
+  setSpanContext(spanContext: SpanContext | null): this;
+
   /** Clears the current scope and resets its properties. */
   clear(): this;
 
diff --git a/packages/types/src/span.ts b/packages/types/src/span.ts
diff --git a/packages/types/src/spancontext.ts b/packages/types/src/spancontext.ts
@@ -0,0 +1,9 @@
+/** JSDoc */
+export interface SpanContext {
+  traceId: string;
+  spanId: string;
+  recorded: boolean;
+  parent?: SpanContext;
+  /** JSDoc */
+  toTraceparent(): string;
+}
