122 lines
4 KiB
TypeScript
122 lines
4 KiB
TypeScript
/*@internal*/
|
|
/** Performance measurements for the compiler. */
|
|
namespace ts.performance {
|
|
declare const onProfilerEvent: { (markName: string): void; profiler: boolean; };
|
|
|
|
// NOTE: cannot use ts.noop as core.ts loads after this
|
|
const profilerEvent: (markName: string) => void = typeof onProfilerEvent === "function" && onProfilerEvent.profiler === true ? onProfilerEvent : () => { /*empty*/ };
|
|
|
|
let enabled = false;
|
|
let profilerStart = 0;
|
|
let counts: Map<number>;
|
|
let marks: Map<number>;
|
|
let measures: Map<number>;
|
|
|
|
export interface Timer {
|
|
enter(): void;
|
|
exit(): void;
|
|
}
|
|
|
|
export function createTimerIf(condition: boolean, measureName: string, startMarkName: string, endMarkName: string) {
|
|
return condition ? createTimer(measureName, startMarkName, endMarkName) : nullTimer;
|
|
}
|
|
|
|
export function createTimer(measureName: string, startMarkName: string, endMarkName: string): Timer {
|
|
let enterCount = 0;
|
|
return {
|
|
enter,
|
|
exit
|
|
};
|
|
|
|
function enter() {
|
|
if (++enterCount === 1) {
|
|
mark(startMarkName);
|
|
}
|
|
}
|
|
|
|
function exit() {
|
|
if (--enterCount === 0) {
|
|
mark(endMarkName);
|
|
measure(measureName, startMarkName, endMarkName);
|
|
}
|
|
else if (enterCount < 0) {
|
|
Debug.fail("enter/exit count does not match.");
|
|
}
|
|
}
|
|
}
|
|
|
|
export const nullTimer: Timer = { enter: noop, exit: noop };
|
|
|
|
/**
|
|
* Marks a performance event.
|
|
*
|
|
* @param markName The name of the mark.
|
|
*/
|
|
export function mark(markName: string) {
|
|
if (enabled) {
|
|
marks.set(markName, timestamp());
|
|
counts.set(markName, (counts.get(markName) || 0) + 1);
|
|
profilerEvent(markName);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Adds a performance measurement with the specified name.
|
|
*
|
|
* @param measureName The name of the performance measurement.
|
|
* @param startMarkName The name of the starting mark. If not supplied, the point at which the
|
|
* profiler was enabled is used.
|
|
* @param endMarkName The name of the ending mark. If not supplied, the current timestamp is
|
|
* used.
|
|
*/
|
|
export function measure(measureName: string, startMarkName?: string, endMarkName?: string) {
|
|
if (enabled) {
|
|
const end = endMarkName && marks.get(endMarkName) || timestamp();
|
|
const start = startMarkName && marks.get(startMarkName) || profilerStart;
|
|
measures.set(measureName, (measures.get(measureName) || 0) + (end - start));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the number of times a marker was encountered.
|
|
*
|
|
* @param markName The name of the mark.
|
|
*/
|
|
export function getCount(markName: string) {
|
|
return counts && counts.get(markName) || 0;
|
|
}
|
|
|
|
/**
|
|
* Gets the total duration of all measurements with the supplied name.
|
|
*
|
|
* @param measureName The name of the measure whose durations should be accumulated.
|
|
*/
|
|
export function getDuration(measureName: string) {
|
|
return measures && measures.get(measureName) || 0;
|
|
}
|
|
|
|
/**
|
|
* Iterate over each measure, performing some action
|
|
*
|
|
* @param cb The action to perform for each measure
|
|
*/
|
|
export function forEachMeasure(cb: (measureName: string, duration: number) => void) {
|
|
measures.forEach((measure, key) => {
|
|
cb(key, measure);
|
|
});
|
|
}
|
|
|
|
/** Enables (and resets) performance measurements for the compiler. */
|
|
export function enable() {
|
|
counts = createMap<number>();
|
|
marks = createMap<number>();
|
|
measures = createMap<number>();
|
|
enabled = true;
|
|
profilerStart = timestamp();
|
|
}
|
|
|
|
/** Disables performance measurements for the compiler. */
|
|
export function disable() {
|
|
enabled = false;
|
|
}
|
|
}
|