@contrail/util 1.1.15-alpha-2 → 1.1.15-alpha-4

package/lib/performance-util/performance-util.d.ts

@@ -5,13 +5,20 @@ type TimingNode = {
     durationMs: number;
     children: TimingNode[];
 };
+type ActiveSpan = {
+    node: TimingNode;
+    startTime: number;
+};
 type TimingProfile = {
     timingSpans: TimingSpans;
     timingStack: TimingStack;
+    activeSpans: Map<string, ActiveSpan>;
 };
 export declare function getDefaultTimingProfile(): TimingProfile;
 export declare function clearDefaultTimingProfile(): void;
 export declare function withTimingAsync<T>(label: string, fn: () => Promise<T>): Promise<T>;
 export declare function withTimingSync<T>(label: string, fn: () => T): T;
-export declare function displayTimingTree(nodes: TimingSpans, depth?: number): string;
+export declare function displayTimingTree(nodes?: TimingSpans, depth?: number): string;
+export declare function startSpan(spanName: string): void;
+export declare function endSpan(spanName: string): void;
 export {};

package/lib/performance-util/performance-util.js

@@ -14,9 +14,12 @@ exports.clearDefaultTimingProfile = clearDefaultTimingProfile;
 exports.withTimingAsync = withTimingAsync;
 exports.withTimingSync = withTimingSync;
 exports.displayTimingTree = displayTimingTree;
+exports.startSpan = startSpan;
+exports.endSpan = endSpan;
 const DEFAULT_TIMING_PROFILE = {
     timingSpans: [],
     timingStack: [],
+    activeSpans: new Map(),
 };
 function getDefaultTimingProfile() {
     return DEFAULT_TIMING_PROFILE;
@@ -30,6 +33,7 @@ function getTimingSpans() {
 function clearDefaultTimingProfile() {
     DEFAULT_TIMING_PROFILE.timingSpans = [];
     DEFAULT_TIMING_PROFILE.timingStack = [];
+    DEFAULT_TIMING_PROFILE.activeSpans.clear();
 }
 function withTimingAsync(label, fn) {
     return __awaiter(this, void 0, void 0, function* () {
@@ -70,7 +74,7 @@ function withTimingSync(label, fn) {
         getTimingStack().pop();
     }
 }
-function displayTimingTree(nodes, depth = 0) {
+function displayTimingTree(nodes = DEFAULT_TIMING_PROFILE.timingSpans, depth = 0) {
     const lines = [];
     const indent = '  '.repeat(depth);
     const totalWidth = 80;
@@ -93,3 +97,41 @@ function displayTimingTree(nodes, depth = 0) {
     }
     return result;
 }
+function startSpan(spanName) {
+    if (DEFAULT_TIMING_PROFILE.activeSpans.has(spanName)) {
+        throw new Error(`Span "${spanName}" is already active. Please end it before starting a new one with the same name.`);
+    }
+    const startTime = performance.now();
+    const timingNode = { label: spanName, durationMs: 0, children: [] };
+    const parent = getTimingStack().length > 0 ? getTimingStack()[getTimingStack().length - 1] : null;
+    if (parent) {
+        parent.children.push(timingNode);
+    }
+    else {
+        getTimingSpans().push(timingNode);
+    }
+    getTimingStack().push(timingNode);
+    DEFAULT_TIMING_PROFILE.activeSpans.set(spanName, {
+        node: timingNode,
+        startTime: startTime,
+    });
+}
+function endSpan(spanName) {
+    const activeSpan = DEFAULT_TIMING_PROFILE.activeSpans.get(spanName);
+    if (!activeSpan) {
+        throw new Error(`No active span found with name "${spanName}". Make sure to call startSpan() first.`);
+    }
+    const endTime = performance.now();
+    activeSpan.node.durationMs = endTime - activeSpan.startTime;
+    const currentTop = getTimingStack()[getTimingStack().length - 1];
+    if (currentTop === activeSpan.node) {
+        getTimingStack().pop();
+    }
+    else {
+        const stackIndex = getTimingStack().findIndex(node => node === activeSpan.node);
+        if (stackIndex !== -1) {
+            getTimingStack().splice(stackIndex, 1);
+        }
+    }
+    DEFAULT_TIMING_PROFILE.activeSpans.delete(spanName);
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contrail/util",
-  "version": "1.1.15-alpha-2",
+  "version": "1.1.15-alpha-4",
   "description": "General JavaScript utilities",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",