mirror of
https://github.com/microsoft/playwright.git
synced 2025-06-26 21:40:17 +00:00
88 lines
2.4 KiB
TypeScript
88 lines
2.4 KiB
TypeScript
/**
|
|
* 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.
|
|
*/
|
|
|
|
import type { RawStack } from './stackTrace';
|
|
|
|
export type ZoneType = 'apiZone' | 'expectZone';
|
|
|
|
class ZoneManager {
|
|
lastZoneId = 0;
|
|
readonly _zones = new Map<number, Zone<any>>();
|
|
|
|
run<T, R>(type: ZoneType, data: T, func: (data: T) => R | Promise<R>): R | Promise<R> {
|
|
return new Zone<T>(this, ++this.lastZoneId, type, data).run(func);
|
|
}
|
|
|
|
zoneData<T>(type: ZoneType, rawStack: RawStack): T | null {
|
|
for (const line of rawStack) {
|
|
const index = line.indexOf('__PWZONE__[');
|
|
if (index !== -1) {
|
|
const zoneId = + line.substring(index + '__PWZONE__['.length, line.indexOf(']', index));
|
|
const zone = this._zones.get(zoneId);
|
|
if (zone && zone.type === type)
|
|
return zone.data;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
|
|
class Zone<T> {
|
|
private _manager: ZoneManager;
|
|
readonly id: number;
|
|
readonly type: ZoneType;
|
|
data: T;
|
|
readonly wallTime: number;
|
|
|
|
constructor(manager: ZoneManager, id: number, type: ZoneType, data: T) {
|
|
this._manager = manager;
|
|
this.id = id;
|
|
this.type = type;
|
|
this.data = data;
|
|
this.wallTime = Date.now();
|
|
}
|
|
|
|
run<R>(func: (data: T) => R | Promise<R>): R | Promise<R> {
|
|
this._manager._zones.set(this.id, this);
|
|
Object.defineProperty(func, 'name', { value: `__PWZONE__[${this.id}]` });
|
|
return runWithFinally(() => func(this.data), () => {
|
|
this._manager._zones.delete(this.id);
|
|
});
|
|
}
|
|
}
|
|
|
|
export function runWithFinally<R>(func: () => R | Promise<R>, finallyFunc: Function): R | Promise<R> {
|
|
try {
|
|
const result = func();
|
|
if (result instanceof Promise) {
|
|
return result.then(r => {
|
|
finallyFunc();
|
|
return r;
|
|
}).catch(e => {
|
|
finallyFunc();
|
|
throw e;
|
|
});
|
|
}
|
|
finallyFunc();
|
|
return result;
|
|
} catch (e) {
|
|
finallyFunc();
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
export const zones = new ZoneManager();
|