123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.ProgressController = void 0;
- var _errors = require("./errors");
- var _utils = require("../utils");
- var _manualPromise = require("../utils/manualPromise");
- /**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- class ProgressController {
- constructor(metadata, sdkObject) {
- this._forceAbortPromise = new _manualPromise.ManualPromise();
- // Cleanups to be run only in the case of abort.
- this._cleanups = [];
- this._logName = 'api';
- this._state = 'before';
- this._deadline = 0;
- this._timeout = 0;
- this.metadata = void 0;
- this.instrumentation = void 0;
- this.sdkObject = void 0;
- this.metadata = metadata;
- this.sdkObject = sdkObject;
- this.instrumentation = sdkObject.instrumentation;
- this._forceAbortPromise.catch(e => null); // Prevent unhandled promise rejection.
- }
- setLogName(logName) {
- this._logName = logName;
- }
- abort(error) {
- this._forceAbortPromise.reject(error);
- }
- async run(task, timeout) {
- var _this$sdkObject$attri;
- if (timeout) {
- this._timeout = timeout;
- this._deadline = timeout ? (0, _utils.monotonicTime)() + timeout : 0;
- }
- (0, _utils.assert)(this._state === 'before');
- this._state = 'running';
- (_this$sdkObject$attri = this.sdkObject.attribution.context) === null || _this$sdkObject$attri === void 0 ? void 0 : _this$sdkObject$attri._activeProgressControllers.add(this);
- const progress = {
- log: message => {
- progress.logEntry({
- message
- });
- },
- logEntry: entry => {
- if ('message' in entry) {
- const message = entry.message;
- if (this._state === 'running') this.metadata.log.push(message);
- // Note: we might be sending logs after progress has finished, for example browser logs.
- this.instrumentation.onCallLog(this.sdkObject, this.metadata, this._logName, message);
- }
- },
- timeUntilDeadline: () => this._deadline ? this._deadline - (0, _utils.monotonicTime)() : 2147483647,
- // 2^31-1 safe setTimeout in Node.
- isRunning: () => this._state === 'running',
- cleanupWhenAborted: cleanup => {
- if (this._state === 'running') this._cleanups.push(cleanup);else runCleanup(cleanup);
- },
- throwIfAborted: () => {
- if (this._state === 'aborted') throw new AbortedError();
- },
- beforeInputAction: async element => {
- await this.instrumentation.onBeforeInputAction(this.sdkObject, this.metadata, element);
- },
- metadata: this.metadata
- };
- const timeoutError = new _errors.TimeoutError(`Timeout ${this._timeout}ms exceeded.`);
- const timer = setTimeout(() => this._forceAbortPromise.reject(timeoutError), progress.timeUntilDeadline());
- try {
- const promise = task(progress);
- const result = await Promise.race([promise, this._forceAbortPromise]);
- this._state = 'finished';
- return result;
- } catch (e) {
- this._state = 'aborted';
- await Promise.all(this._cleanups.splice(0).map(runCleanup));
- throw e;
- } finally {
- var _this$sdkObject$attri2;
- (_this$sdkObject$attri2 = this.sdkObject.attribution.context) === null || _this$sdkObject$attri2 === void 0 ? void 0 : _this$sdkObject$attri2._activeProgressControllers.delete(this);
- clearTimeout(timer);
- }
- }
- }
- exports.ProgressController = ProgressController;
- async function runCleanup(cleanup) {
- try {
- await cleanup();
- } catch (e) {}
- }
- class AbortedError extends Error {}
|