o11y 250.5.0 → 250.6.0

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

@@ -896,12 +896,13 @@ const stopReason = {
     discarded
 };
 class ActivityImpl {
-    constructor(_name, _rootId, _onStopped, _onError, timeout, id, _isSampled) {
+    constructor(_name, _rootId, _onStopped, _onError, timeout, id, _isSampled, _startPerfTime, _noRerooting) {
         this._name = _name;
         this._rootId = _rootId;
         this._onStopped = _onStopped;
         this._onError = _onError;
         this._isSampled = _isSampled;
+        this._noRerooting = _noRerooting;
         this._errorCount = 0;
         this._timerOverridden = TimerOverride.none;
         const { tsNow, perfNow } = utility.time();
@@ -923,6 +924,7 @@ class ActivityImpl {
         }
         this._startTimestamp = tsNow;
         this._startPerfTime = perfNow;
+        this._overrideTimer(_startPerfTime);
         if (timeout > 0) {
             this._timer = setTimeout(() => {
                 this._stopReason = this._stopReason || stopReason.timedout;
@@ -969,8 +971,9 @@ class ActivityImpl {
                 }
                 if (isStopTimeDefined &&
                     (perfStop < 0 ||
-                        perfStop < (isStartTimeDefined ? perfStart : this._startPerfTime))) {
-                    throw new Error('perfStopOverride, if defined, must be >= 0 and >= startTime');
+                        perfStop < (isStartTimeDefined ? perfStart : this._startPerfTime) ||
+                        perfStop > utility.time().perfNow + 100)) {
+                    throw new Error('perfStopOverride, if defined, must be >= 0 and >= startTime and <= now');
                 }
             }
             this._stopInternal(userSchemaOrText, userData, options);
@@ -1057,7 +1060,8 @@ class ActivityImpl {
             stopPerfTime: this._stopPerfTime,
             errorCount: this._errorCount,
             isSampled: this.getIsSampled(),
-            timerOverridden: this._timerOverridden
+            timerOverridden: this._timerOverridden,
+            noRerooting: this._noRerooting
         };
     }
     getTraceHeaders(options) {
@@ -1588,8 +1592,10 @@ class InstrumentationImpl {
             utility.requireArgument(name, 'name');
             utility.requireArgumentIfDefined(options, 'options', 'object');
             let context;
+            let noRerooting = (options === null || options === void 0 ? void 0 : options.perfStartOverride) !== undefined;
             const rootId = (_a = options === null || options === void 0 ? void 0 : options.instrumentationContext) === null || _a === void 0 ? void 0 : _a.rootId;
             if (rootId) {
+                noRerooting = true;
                 context = {
                     rootId,
                     isRootActivitySampled: options.instrumentationContext.isRootActivitySampled
@@ -1598,7 +1604,7 @@ class InstrumentationImpl {
             else {
                 context = this._nextGen.getDefaultInstrumentationContext();
             }
-            const activity = new ActivityImpl(name, context.rootId, this._onActivityStoppedCallback, this._onActivityErrorCallback, maxActivityDurationMsecs, undefined, context.isRootActivitySampled);
+            const activity = new ActivityImpl(name, context.rootId, this._onActivityStoppedCallback, this._onActivityErrorCallback, maxActivityDurationMsecs, undefined, context.isRootActivitySampled, options === null || options === void 0 ? void 0 : options.perfStartOverride, noRerooting);
             return activity;
         }, activityNoOp);
     }
@@ -1612,13 +1618,22 @@ class InstrumentationImpl {
             userPayload,
             errorCount: detail.errorCount,
             isSampled: detail.isSampled,
-            timerOverridden: detail.timerOverridden
+            timerOverridden: detail.timerOverridden,
+            isRerooted: detail.isRerooted
         };
     }
     _handleActivityStop(detail) {
         if (detail.stopReason === discarded) {
             return;
         }
+        if (!detail.isRoot && !detail.noRerooting) {
+            const rahe = this._nextGen.getRootActivityHistoryEntryByTime(detail.startPerfTime);
+            if ((!rahe && detail.rootId !== undefined) || (rahe && detail.rootId !== rahe.id)) {
+                detail.rootId = rahe === null || rahe === void 0 ? void 0 : rahe.id;
+                detail.isSampled = (rahe === null || rahe === void 0 ? void 0 : rahe.isSampled) || false;
+                detail.isRerooted = true;
+            }
+        }
         const activityData = this._getActivityData(detail);
         this._logActivity(activityData, detail.startTimestamp, detail.rootId);
     }
@@ -1742,9 +1757,17 @@ class InstrumentationImpl {
         utility.requireArgument(listener, 'listener', 'function');
         this._nextGen.registerForLogPrompt(listener);
     }
+    _getAsaOptions(options) {
+        return options
+            ? Object.assign({
+                instrumentationContext: options === null || options === void 0 ? void 0 : options.instrumentationContext
+            }, options.startOptions)
+            : undefined;
+    }
     activity(name, execute, options) {
         var _a, _b, _c, _d;
-        const act = this.startActivity(name, options);
+        const asaOptions = this._getAsaOptions(options);
+        const act = this.startActivity(name, asaOptions);
         try {
             return execute(act);
         }
@@ -1753,12 +1776,13 @@ class InstrumentationImpl {
             throw err;
         }
         finally {
-            act.stop((_c = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _c === void 0 ? void 0 : _c.schema, (_d = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _d === void 0 ? void 0 : _d.payload);
+            act.stop((_c = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _c === void 0 ? void 0 : _c.schema, (_d = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _d === void 0 ? void 0 : _d.payload, options === null || options === void 0 ? void 0 : options.stopOptions);
         }
     }
     async activityAsync(name, execute, options) {
         var _a, _b, _c, _d;
-        const act = this.startActivity(name, options);
+        const asaOptions = this._getAsaOptions(options);
+        const act = this.startActivity(name, asaOptions);
         try {
             return await execute(act);
         }
@@ -1767,15 +1791,66 @@ class InstrumentationImpl {
             throw err;
         }
         finally {
-            act.stop((_c = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _c === void 0 ? void 0 : _c.schema, (_d = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _d === void 0 ? void 0 : _d.payload);
+            act.stop((_c = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _c === void 0 ? void 0 : _c.schema, (_d = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _d === void 0 ? void 0 : _d.payload, options === null || options === void 0 ? void 0 : options.stopOptions);
         }
     }
 }
 InstrumentationImpl.defaultAppName = 'APP_NOT_REGISTERED';
 
+const GRACE = 0.1;
+const DEFAULT_LIMIT = 20;
+class RootActivityHistoryImpl {
+    constructor(limit) {
+        this._historyMap = new Map();
+        this._historyList = new Array();
+        if (utility.requireArgumentIfDefined(limit, 'limit', 'number')) {
+            limit = Math.ceil(limit);
+            if (limit <= 0) {
+                throw new Error('Limit must be a positive number');
+            }
+        }
+        else {
+            limit = DEFAULT_LIMIT;
+        }
+        this._graceCount = Math.max(1, Math.floor(limit * GRACE));
+        this._realLimit = 1 + limit + this._graceCount;
+    }
+    add(entry) {
+        if (this._historyMap.has(entry.id)) {
+            throw new Error(`RootActivityHistoryImpl.add: entry already exists for rootId: ${entry.id}`);
+        }
+        this._historyMap.set(entry.id, entry);
+        this._historyList.push(entry);
+        if (this._historyList.length === this._realLimit) {
+            const removed = this._historyList.splice(1, this._graceCount);
+            removed.forEach((e) => this._historyMap.delete(e.id));
+        }
+    }
+    get count() {
+        return this._historyList.length;
+    }
+    findById(rootId) {
+        return this._historyMap.get(rootId);
+    }
+    findByTime(atTime) {
+        let closest;
+        let closestDiff = Number.MAX_VALUE;
+        for (const entry of this._historyList) {
+            const startDelta = atTime - entry.startPerfTime;
+            const isDuringRoot = startDelta >= 0 &&
+                (entry.stopPerfTime === undefined || atTime < entry.stopPerfTime);
+            if (isDuringRoot && startDelta < closestDiff) {
+                closest = entry;
+                closestDiff = startDelta;
+            }
+        }
+        return closest;
+    }
+}
+
 class RootActivityImpl extends ActivityImpl {
-    constructor(name, onStopped, onError, rootId, _isSampled) {
-        super(name, undefined, onStopped, onError, undefined, rootId || utility.generateUniqueId(32), _isSampled);
+    constructor(name, onStopped, onError, rootId, isSampled, startPerfTime) {
+        super(name, undefined, onStopped, onError, undefined, rootId || utility.generateUniqueId(32), isSampled, startPerfTime);
     }
     get preRootId() {
         return this._preRootId;
@@ -1828,13 +1903,21 @@ class AppInstrumentationImpl extends InstrumentationImpl {
     constructor(_nextGen, _name, _allowMulti) {
         super(_nextGen, _name);
         this._allowMulti = _allowMulti;
+        this._rootActivityHistory = new RootActivityHistoryImpl();
         this._onRootActivityStoppedCallback = this._handleRootActivityStop.bind(this);
     }
-    startRootActivity(name, rootId, isSampled) {
+    startRootActivity(name, rootId, isSampled, options) {
         return this._safety.tryCatchReturn(() => {
             utility.requireArgument(name, 'name');
-            const newRootActivity = new RootActivityImpl(name, this._onRootActivityStoppedCallback, this._onActivityErrorCallback, rootId, isSampled);
+            const newRootActivity = new RootActivityImpl(name, this._onRootActivityStoppedCallback, this._onActivityErrorCallback, rootId, isSampled, options === null || options === void 0 ? void 0 : options.perfStartOverride);
             if (!this._allowMulti) {
+                const entry = {
+                    id: newRootActivity.getId(),
+                    name,
+                    startPerfTime: newRootActivity.getStartPerfTime(),
+                    isSampled: newRootActivity.getIsSampled()
+                };
+                this._rootActivityHistory.add(entry);
                 if (this._singleRootActivity && !this._singleRootActivity.isStopped) {
                     const preRootId = this._singleRootActivity.getId();
                     this._singleRootActivity.terminate();
@@ -1846,6 +1929,11 @@ class AppInstrumentationImpl extends InstrumentationImpl {
         }, rootActivityNoOp);
     }
     _handleRootActivityStop(detail) {
+        const historyEntry = this._rootActivityHistory.findById(detail.id);
+        if (historyEntry) {
+            historyEntry.stopPerfTime = detail.stopPerfTime;
+            historyEntry.startPerfTime = detail.startPerfTime;
+        }
         if (detail.stopReason === discarded) {
             return;
         }
@@ -1870,6 +1958,9 @@ class AppInstrumentationImpl extends InstrumentationImpl {
     _initMetrics() {
         return new MetricsImpl(this.name, () => this.name);
     }
+    getRootActivityHistoryEntry(atTime) {
+        return this._rootActivityHistory.findByTime(atTime);
+    }
 }
 
 const pathLimit = 5;
@@ -2235,6 +2326,10 @@ class NextgenImpl {
         var _a;
         return (_a = this._appInstr) === null || _a === void 0 ? void 0 : _a.name;
     }
+    getRootActivityHistoryEntryByTime(atTime) {
+        var _a;
+        return (_a = this._appInstr) === null || _a === void 0 ? void 0 : _a.getRootActivityHistoryEntry(atTime);
+    }
     addLog(loggerName, schema, data, timestamp, rootId) {
         var _a, _b, _c;
         payloadUtility.checkInputs(schema, data);
@@ -2535,7 +2630,7 @@ class ConsoleCollector {
     }
 }
 
-const version = '250.5.0';
+const version = '250.6.0';
 
 const idleDetector = new IdleDetectorImpl({
     logThreshold: 300,