o11y 240.7.0 → 240.8.0

package/dist/modules/o11y/client/client.js

@@ -220,6 +220,14 @@ class Tracing {
 Tracing._global = utility.getGlobal();
 Tracing._isNetworkInstrumentationEnabled = false;
 
+var TimerOverride;
+(function (TimerOverride) {
+    TimerOverride[TimerOverride["none"] = 0] = "none";
+    TimerOverride[TimerOverride["start"] = 1] = "start";
+    TimerOverride[TimerOverride["stop"] = 2] = "stop";
+    TimerOverride[TimerOverride["both"] = 3] = "both";
+})(TimerOverride || (TimerOverride = {}));
+
 const timedout = 'timedout';
 const terminated = 'terminated';
 const discarded = 'discarded';
@@ -236,6 +244,7 @@ class ActivityImpl {
         this._onError = _onError;
         this._isSampled = _isSampled;
         this._errorCount = 0;
+        this._timerOverridden = TimerOverride.none;
         const { tsNow, perfNow } = utility.time();
         ActivityImpl._count += 1;
         this._id = id || utility.generateUniqueId();
@@ -284,10 +293,51 @@ class ActivityImpl {
             this._onError(error, this._getDetail(userSchemaOrText, userData));
         });
     }
-    stop(userSchemaOrText, userData) {
-        this._safety.tryCatch(() => this._stopInternal(userSchemaOrText, userData));
+    stop(userSchemaOrText, userData, options) {
+        this._safety.tryCatch(() => {
+            utility.requireArgumentIfDefined(userSchemaOrText, 'userSchemaOrText', [
+                'object',
+                'string'
+            ]);
+            utility.requireArgumentIfDefined(userData, 'userData', 'object');
+            if (options) {
+                const perfStart = options.perfStartOverride;
+                const isStartTimeDefined = utility.requireArgumentIfDefined(perfStart, 'options.perfStartOverride', 'number');
+                const perfStop = options.perfStopOverride;
+                const isStopTimeDefined = utility.requireArgumentIfDefined(perfStop, 'options.perfStopOverride', 'number');
+                if (isStartTimeDefined && perfStart < 0) {
+                    throw new Error('perfStartOverride, if defined, must be >= 0');
+                }
+                if (isStopTimeDefined &&
+                    (perfStop < 0 ||
+                        perfStop < (isStartTimeDefined ? perfStart : this._startPerfTime))) {
+                    throw new Error('perfStopOverride, if defined, must be >= 0 and >= startTime');
+                }
+            }
+            this._stopInternal(userSchemaOrText, userData, options);
+        });
     }
-    _stopInternal(userSchemaOrText, userData) {
+    _overrideTimer(perfStartTime, perfStopTime) {
+        if (perfStartTime !== undefined) {
+            this._startTimestamp = Math.round(this._startTimestamp - (this._startPerfTime - perfStartTime));
+            this._startPerfTime = perfStartTime;
+            this._setTimerOverride(TimerOverride.start);
+        }
+        if (perfStopTime !== undefined) {
+            this._stopPerfTime = perfStopTime;
+            this._setTimerOverride(TimerOverride.stop);
+        }
+    }
+    _setTimerOverride(value) {
+        if (this._timerOverridden === TimerOverride.both ||
+            this._timerOverridden === value ||
+            value === TimerOverride.none) {
+            return;
+        }
+        this._timerOverridden =
+            this._timerOverridden === TimerOverride.none ? value : TimerOverride.both;
+    }
+    _stopInternal(userSchemaOrText, userData, options) {
         const stopPerfTime = utility.perfNow();
         if (this._usePerf) {
             try {
@@ -306,7 +356,14 @@ class ActivityImpl {
         if (this.isStopped) {
             return;
         }
-        this._stopPerfTime = stopPerfTime;
+        if (options) {
+            this._overrideTimer(options.perfStartOverride, options.perfStopOverride);
+        }
+        this._stopPerfTime =
+            this._timerOverridden === TimerOverride.none ||
+                this._timerOverridden === TimerOverride.start
+                ? stopPerfTime
+                : this._stopPerfTime;
         const activityDetail = this._getDetail(userSchemaOrText, userData);
         this._onStopped(activityDetail);
     }
@@ -340,7 +397,8 @@ class ActivityImpl {
             startPerfTime: this._startPerfTime,
             stopPerfTime: this._stopPerfTime,
             errorCount: this._errorCount,
-            isSampled: this.getIsSampled()
+            isSampled: this.getIsSampled(),
+            timerOverridden: this._timerOverridden
         };
     }
     getTraceHeaders(options) {