Use CDN sample rate (wip)

Author: danieljackins

packages/browser/src/browser/index.ts

@@ -122,6 +122,13 @@ export interface CDNSettings {
         version: number
       }
     | {}
+
+  /**
+   * Settings for auto instrumentation
+   */
+  autoInstrumentationSettings?: {
+    sampleRate: number
+  }
 }
 
 export interface AnalyticsBrowserSettings {

packages/signals/signals/src/core/client/index.ts

@@ -10,13 +10,15 @@ export class SignalsIngestSettings {
   shouldDisableSignalRedaction: () => boolean
   signalIngestion: () => boolean
   writeKey?: string
+  sampleRate: number
   constructor(settings: SignalsIngestSettingsConfig) {
     this.flushAt = settings.flushAt ?? 5
     this.apiHost = settings.apiHost ?? 'signals.segment.io/v1'
     this.flushInterval = settings.flushInterval ?? 2000
     this.shouldDisableSignalRedaction =
       settings.shouldDisableSignalRedaction ?? (() => false)
     this.signalIngestion = settings.signalIngestion ?? (() => false)
+    this.sampleRate = settings.sampleRate ?? 0
   }
 }
 
@@ -26,6 +28,7 @@ export interface SignalsIngestSettingsConfig {
   flushInterval?: number
   shouldDisableSignalRedaction?: () => boolean
   signalIngestion?: () => boolean
+  sampleRate?: number
 }
 /**
  * This currently just uses the Segment analytics-next library to send signals.
@@ -72,11 +75,23 @@ export class SignalsIngestClient {
     return analytics
   }
 
+  private shouldIngestSignals(): boolean {
+    let shouldIngest = false
+    if (Math.random() > this.settings.sampleRate) {
+      // currently a no-op, but will be used once signals are sent outside of debug mode
+      shouldIngest = false
+    }
+    if (this.settings.signalIngestion()) {
+      shouldIngest = true
+    }
+    return shouldIngest
+  }
+
   private sendTrackCall(signal: Signal) {
     if (!this.analytics) {
       throw new Error('Please initialize before calling this method.')
     }
-    if (!this.settings.signalIngestion) {
+    if (!this.shouldIngestSignals()) {
       return
     }
     const disableRedaction = this.settings.shouldDisableSignalRedaction()

packages/signals/signals/src/core/signals/settings.ts

@@ -15,6 +15,7 @@ export type SignalsSettingsConfig = Pick<
   | 'flushInterval'
   | 'disableSignalsRedaction'
   | 'signalsIngestion'
+  | 'sampleRate'
   | 'networkSignalsAllowList'
   | 'networkSignalsDisallowList'
   | 'networkSignalsAllowSameDomain'
@@ -58,6 +59,7 @@ export class SignalGlobalSettings {
       flushInterval: settings.flushInterval,
       shouldDisableSignalRedaction: this.redaction.getDisableSignalRedaction,
       signalIngestion: this.ingestion.getSignalIngestion,
+      sampleRate: settings.sampleRate,
     }
     this.sandbox = {
       functionHost: settings.functionHost,
@@ -73,6 +75,7 @@ export class SignalGlobalSettings {
   public update({
     edgeFnDownloadURL,
     disallowListURLs,
+    sampleRate,
   }: {
     /**
      * The URL to download the edge function from
@@ -82,13 +85,18 @@ export class SignalGlobalSettings {
      * Add new URLs to the disallow list
      */
     disallowListURLs: (string | undefined)[]
+    /**
+     * The sample rate pulled from CDN settings
+     */
+    sampleRate: number
   }): void {
     edgeFnDownloadURL && (this.sandbox.edgeFnDownloadURL = edgeFnDownloadURL)
     this.network.networkSignalsFilterList.disallowed.addURLLike(
       ...disallowListURLs.filter(<T>(val: T): val is NonNullable<T> =>
         Boolean(val)
       )
     )
+    this.ingestClient.sampleRate = sampleRate
   }
 }
 

packages/signals/signals/src/core/signals/signals.ts

@@ -91,6 +91,9 @@ export class Signals implements ISignals {
         analyticsService.instance.settings.apiHost,
         analyticsService.instance.settings.cdnURL,
       ],
+      sampleRate:
+        analyticsService.instance.settings.cdnSettings
+          .autoInstrumentationSettings?.sampleRate || 0,
     })
 
     const sandbox = new Sandbox(

packages/signals/signals/src/types/analytics-api.ts

@@ -8,9 +8,14 @@ export type EdgeFnCDNSettings = {
   downloadURL: string
 }
 
+export type AutoInstrumentationCDNSettings = {
+  sampleRate: number
+}
+
 export interface CDNSettings {
   integrations: CDNSettingsIntegrations
   edgeFunction?: EdgeFnCDNSettings | { [key: string]: never }
+  autoInstrumentationSettings?: AutoInstrumentationCDNSettings
 }
 
 export interface SegmentEventStub {

packages/signals/signals/src/types/settings.ts

@@ -28,6 +28,11 @@ export interface SignalsPluginSettingsConfig {
    */
   signalsIngestion?: boolean
 
+  /**
+   * Percentage of sessions that should be ingested
+   */
+  sampleRate?: number
+
   /**
    * Override signals API host
    * @default signals.segment.io/v1