@memlab/core 1.1.17 → 1.1.19

package/dist/index.d.ts

@@ -7,11 +7,11 @@
  * @format
  * @oncall web_perf_infra
  */
-/** @internal */
-export declare function registerPackage(): Promise<void>;
 export * from './lib/Types';
 export * from './lib/NodeHeap';
 /** @internal */
+export declare function registerPackage(): Promise<void>;
+/** @internal */
 export { default as config } from './lib/Config';
 /** @internal */
 export * from './lib/InternalValueSetter';
@@ -36,6 +36,8 @@ export { default as analysis } from './lib/HeapAnalyzer';
 /** @internal */
 export { default as constant } from './lib/Constant';
 /** @internal */
+export { default as memoryBarChart } from './lib/charts/MemoryBarChart';
+/** @internal */
 export { default as modes } from './modes/RunningModes';
 /** @internal */
 export { default as ProcessManager } from './lib/ProcessManager';

package/dist/index.js

@@ -35,9 +35,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TraceFinder = exports.MultiIterationSeqClustering = exports.SequentialClustering = exports.EvaluationMetric = exports.NormalizedTrace = exports.leakClusterLogger = exports.ProcessManager = exports.modes = exports.constant = exports.analysis = exports.browserInfo = exports.serializer = exports.fileManager = exports.utils = exports.BaseOption = exports.info = exports.config = exports.registerPackage = void 0;
+exports.TraceFinder = exports.MultiIterationSeqClustering = exports.SequentialClustering = exports.EvaluationMetric = exports.NormalizedTrace = exports.leakClusterLogger = exports.ProcessManager = exports.modes = exports.memoryBarChart = exports.constant = exports.analysis = exports.browserInfo = exports.serializer = exports.fileManager = exports.utils = exports.BaseOption = exports.info = exports.config = exports.registerPackage = void 0;
 const path_1 = __importDefault(require("path"));
 const PackageInfoLoader_1 = require("./lib/PackageInfoLoader");
+__exportStar(require("./lib/Types"), exports);
+__exportStar(require("./lib/NodeHeap"), exports);
 /** @internal */
 function registerPackage() {
     return __awaiter(this, void 0, void 0, function* () {
@@ -45,8 +47,6 @@ function registerPackage() {
     });
 }
 exports.registerPackage = registerPackage;
-__exportStar(require("./lib/Types"), exports);
-__exportStar(require("./lib/NodeHeap"), exports);
 /** @internal */
 var Config_1 = require("./lib/Config");
 Object.defineProperty(exports, "config", { enumerable: true, get: function () { return __importDefault(Config_1).default; } });
@@ -81,6 +81,9 @@ Object.defineProperty(exports, "analysis", { enumerable: true, get: function ()
 var Constant_1 = require("./lib/Constant");
 Object.defineProperty(exports, "constant", { enumerable: true, get: function () { return __importDefault(Constant_1).default; } });
 /** @internal */
+var MemoryBarChart_1 = require("./lib/charts/MemoryBarChart");
+Object.defineProperty(exports, "memoryBarChart", { enumerable: true, get: function () { return __importDefault(MemoryBarChart_1).default; } });
+/** @internal */
 var RunningModes_1 = require("./modes/RunningModes");
 Object.defineProperty(exports, "modes", { enumerable: true, get: function () { return __importDefault(RunningModes_1).default; } });
 /** @internal */

package/dist/lib/Config.d.ts

@@ -8,8 +8,7 @@
  * @oncall web_perf_infra
  */
 import type { LaunchOptions, Permission } from 'puppeteer';
-import type { AnyFunction, AnyValue, IClusterStrategy, IRunningMode, IScenario, Nullable, Optional, QuickExperiment, ILeakFilter, IPackageInfo } from './Types';
-import { IHeapConfig } from '..';
+import type { AnyFunction, AnyValue, FileOption, IClusterStrategy, IRunningMode, IScenario, IHeapConfig, Nullable, Optional, QuickExperiment, ILeakFilter, IPackageInfo } from './Types';
 interface BrowserLaunchArgumentOptions {
     headless?: boolean;
     userDataDir?: string;
@@ -217,6 +216,8 @@ export declare class MemLabConfig {
     set disableWebSecurity(disable: boolean);
     get disableWebSecurity(): boolean;
     get browserBinaryPath(): string;
+    set defaultFileManagerOption(fileOption: FileOption);
+    get defaultFileManagerOption(): FileOption;
     set reportLeaksInTimers(shouldReport: boolean);
     get reportLeaksInTimers(): boolean;
     setDevice(deviceName: string, options?: {

package/dist/lib/Config.js

@@ -8,6 +8,29 @@
  * @format
  * @oncall web_perf_infra
  */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
@@ -17,7 +40,7 @@ const path_1 = __importDefault(require("path"));
 const RunningModes_1 = __importDefault(require("../modes/RunningModes"));
 const Console_1 = __importDefault(require("./Console"));
 const Constant_1 = __importDefault(require("./Constant"));
-const FileManager_1 = __importDefault(require("./FileManager"));
+const FileManager_1 = __importStar(require("./FileManager"));
 const InternalValueSetter_1 = require("./InternalValueSetter");
 const devices = Constant_1.default.isFRL
     ? {}
@@ -429,6 +452,17 @@ class MemLabConfig {
     get browserBinaryPath() {
         return path_1.default.join(this.browserDir, this.browser);
     }
+    // Default input option for file manager.
+    // If no other input option is provided, the file manager
+    // will generate directories and files based on this default option.
+    set defaultFileManagerOption(fileOption) {
+        FileManager_1.FileManager.defaultFileOption = fileOption;
+        // initialize file and directory paths
+        FileManager_1.default.initDirs(this, fileOption);
+    }
+    get defaultFileManagerOption() {
+        return FileManager_1.FileManager.defaultFileOption;
+    }
     set reportLeaksInTimers(shouldReport) {
         if (shouldReport) {
             this.removeFromSet(this.nodeNameBlockList, this._timerNodes);

package/dist/lib/FileManager.d.ts

@@ -8,35 +8,32 @@
  * @oncall web_perf_infra
  */
 import type { MemLabConfig } from './Config';
-import type { AnyValue, Optional } from './Types';
-/** @internal */
-export declare type FileOption = {
-    workDir?: Optional<string>;
-    clear?: boolean;
-    transient?: boolean;
-};
+import type { AnyValue, FileOption } from './Types';
 /** @internal */
 export declare class FileManager {
+    private memlabConfigCache;
     getDefaultWorkDir(): string;
     generateTmpHeapDir(): string;
     private static transientInstanceIdx;
+    static defaultFileOption: FileOption;
     getWorkDir(options?: FileOption): string;
     getChromeBinaryZipFile(): string;
     getChromeBinaryTimeStampFile(): string;
     getChromeBinaryDir(): string;
-    clearUserDataDir(options: FileOption): void;
-    getDataBaseDir(options: FileOption): string;
+    getDataBaseDir(options?: FileOption): string;
     getCodeDataDir(): string;
     getClusterSampleDataDir(): string;
-    getUserDataDir(options: FileOption): string;
-    getCurDataDir(options: FileOption): string;
+    getUserDataDir(options?: FileOption): string;
+    clearUserDataDir(options?: FileOption): void;
+    getCurDataDir(options?: FileOption): string;
     getWebSourceDir(options?: FileOption): string;
     getWebSourceMetaFile(options?: FileOption): string;
     getDebugDataDir(options?: FileOption): string;
     getDebugSourceFile(options?: FileOption): string;
-    getPersistDataDir(options: FileOption): string;
+    getPersistDataDir(options?: FileOption): string;
     getLoggerOutDir(options?: FileOption): string;
     getHeapAnalysisLogDir(options?: FileOption): string;
+    getHeapSaveLogJSONFile(options?: FileOption): string;
     getTraceClustersDir(options?: FileOption): string;
     getTraceJSONDir(options?: FileOption): string;
     getUnclassifiedTraceClusterDir(options?: FileOption): string;
@@ -67,6 +64,7 @@ export declare class FileManager {
     initNewHeapAnalysisLogFile(options?: FileOption): string;
     getAndInitTSCompileIntermediateDir(): string;
     clearDataDirs(options?: FileOption): void;
+    removeSnapshotFiles(options?: FileOption): void;
     emptyDirIfExists(dir: string): void;
     emptyTraceLogDataDir(options?: FileOption): void;
     resetBrowserDir(): void;

package/dist/lib/FileManager.js

@@ -13,7 +13,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.FileManager = void 0;
-const minimist_1 = __importDefault(require("minimist"));
 const fs_extra_1 = __importDefault(require("fs-extra"));
 const os_1 = __importDefault(require("os"));
 const path_1 = __importDefault(require("path"));
@@ -24,7 +23,7 @@ function joinAndProcessDir(options, ...args) {
     const filepath = path_1.default.join(...args);
     if (!fs_extra_1.default.existsSync(filepath)) {
         try {
-            fs_extra_1.default.mkdirSync(filepath);
+            fs_extra_1.default.mkdirpSync(filepath);
         }
         catch (ex) {
             const err = Utils_1.default.getError(ex);
@@ -37,22 +36,26 @@ function joinAndProcessDir(options, ...args) {
 }
 /** @internal */
 class FileManager {
+    constructor() {
+        this.memlabConfigCache = null;
+    }
     getDefaultWorkDir() {
         return path_1.default.join(this.getTmpDir(), 'memlab');
     }
     generateTmpHeapDir() {
         const dirPath = path_1.default.join(this.getTmpDir(), 'memlab-' + Utils_1.default.getUniqueID());
         if (!fs_extra_1.default.existsSync(dirPath)) {
-            fs_extra_1.default.mkdirSync(dirPath);
+            fs_extra_1.default.mkdirpSync(dirPath);
         }
         return dirPath;
     }
-    getWorkDir(options = {}) {
+    getWorkDir(options = FileManager.defaultFileOption) {
+        var _a;
         // workDir options supercedes all the other options
         if (options.workDir) {
             return path_1.default.resolve(options.workDir);
         }
-        // transient options supercedes other the CLI options
+        // transient options supercedes the other CLI options
         if (options.transient) {
             const idx = ++FileManager.transientInstanceIdx;
             const instanceId = `${process.pid}-${Date.now()}-${idx}`;
@@ -60,8 +63,11 @@ class FileManager {
             return path_1.default.resolve(workDir);
         }
         // workDir from the CLI options
-        const argv = (0, minimist_1.default)(process.argv.slice(2));
-        const workDir = argv['work-dir'] || this.getDefaultWorkDir();
+        const workDir = FileManager.defaultFileOption.workDir ||
+            (
+            // in case there is a transcient working directory generated
+            (_a = this.memlabConfigCache) === null || _a === void 0 ? void 0 : _a.workDir) ||
+            this.getDefaultWorkDir();
         return path_1.default.resolve(workDir);
     }
     getChromeBinaryZipFile() {
@@ -73,11 +79,7 @@ class FileManager {
     getChromeBinaryDir() {
         return path_1.default.join(this.getDefaultWorkDir(), 'chrome');
     }
-    clearUserDataDir(options) {
-        const userDataDir = this.getUserDataDir(options);
-        this.rmDir(userDataDir);
-    }
-    getDataBaseDir(options) {
+    getDataBaseDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getWorkDir(options), 'data');
     }
     getCodeDataDir() {
@@ -86,66 +88,74 @@ class FileManager {
     getClusterSampleDataDir() {
         return path_1.default.join(this.getCodeDataDir(), 'cluster');
     }
-    getUserDataDir(options) {
+    getUserDataDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'profile');
     }
-    getCurDataDir(options) {
+    clearUserDataDir(options = FileManager.defaultFileOption) {
+        const userDataDir = this.getUserDataDir(options);
+        this.rmDir(userDataDir);
+    }
+    getCurDataDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'cur');
     }
-    getWebSourceDir(options = {}) {
+    getWebSourceDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getCurDataDir(options), 'code');
     }
-    getWebSourceMetaFile(options = {}) {
+    getWebSourceMetaFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getWebSourceDir(options), 'files.json');
     }
-    getDebugDataDir(options = {}) {
+    getDebugDataDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'debug');
     }
-    getDebugSourceFile(options = {}) {
+    getDebugSourceFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDebugDataDir(options), 'file.js');
     }
-    getPersistDataDir(options) {
+    getPersistDataDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'persist');
     }
-    getLoggerOutDir(options = {}) {
+    getLoggerOutDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'logger');
     }
     // all heap analysis results generated
-    getHeapAnalysisLogDir(options = {}) {
+    getHeapAnalysisLogDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getLoggerOutDir(options), 'heap-analysis');
     }
+    // memlab save-heap result log file
+    getHeapSaveLogJSONFile(options = FileManager.defaultFileOption) {
+        return path_1.default.join(this.getHeapAnalysisLogDir(options), 'save-heap.json');
+    }
     // all trace clusters generated from the current run
-    getTraceClustersDir(options = {}) {
+    getTraceClustersDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getLoggerOutDir(options), 'trace-clusters');
     }
     // stores JSON file (with heap object and reference details for visualization)
     // of all trace clusters (each cluster has a representative trace)
-    getTraceJSONDir(options = {}) {
+    getTraceJSONDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getLoggerOutDir(options), 'trace-jsons');
     }
-    getUnclassifiedTraceClusterDir(options = {}) {
+    getUnclassifiedTraceClusterDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getLoggerOutDir(options), 'unclassified-clusters');
     }
-    getUniqueTraceClusterDir(options = {}) {
+    getUniqueTraceClusterDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getLoggerOutDir(options), 'unique-trace-clusters');
     }
-    getNewUniqueTraceClusterDir(options = {}) {
+    getNewUniqueTraceClusterDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getUniqueTraceClusterDir(options), 'add');
     }
-    getStaleUniqueTraceClusterDir(options = {}) {
+    getStaleUniqueTraceClusterDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getUniqueTraceClusterDir(options), 'delete');
     }
-    getAllClusterSummaryFile(options = {}) {
+    getAllClusterSummaryFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getUniqueTraceClusterDir(options), 'unique-clusters-summary.txt');
     }
-    getExistingUniqueTraceClusterDir(options = {}) {
+    getExistingUniqueTraceClusterDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getUniqueTraceClusterDir(options), 'existing');
     }
     getAllFilesInDir(dir) {
         const files = fs_extra_1.default.readdirSync(dir);
         return files.map((file) => path_1.default.join(dir, file));
     }
-    getDataOutDir(options = {}) {
+    getDataOutDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataBaseDir(options), 'out');
     }
     getCoreProjectBaseDir() {
@@ -157,19 +167,19 @@ class FileManager {
     getDocDir() {
         return path_1.default.join(this.getMonoRepoDir(), 'website', 'docs');
     }
-    getReportOutDir(options = {}) {
+    getReportOutDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getPersistDataDir(options), 'reports');
     }
-    getPreviewReportDir(options = {}) {
+    getPreviewReportDir(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getPersistDataDir(options), 'report-preview');
     }
-    getLeakSummaryFile(options = {}) {
+    getLeakSummaryFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getDataOutDir(options), 'leaks.txt');
     }
-    getRunMetaFile(options = {}) {
+    getRunMetaFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getCurDataDir(options), 'run-meta.json');
     }
-    getSnapshotSequenceMetaFile(options = {}) {
+    getSnapshotSequenceMetaFile(options = FileManager.defaultFileOption) {
         return path_1.default.join(this.getCurDataDir(options), 'snap-seq.json');
     }
     getInputDataDir() {
@@ -192,6 +202,7 @@ class FileManager {
         }
         return ret;
     }
+    // system default tmp dir
     getTmpDir() {
         return os_1.default.tmpdir();
     }
@@ -211,7 +222,7 @@ class FileManager {
         return { controlWorkDir, testWorkDir };
     }
     // create a unique log file created for heap analysis output
-    initNewHeapAnalysisLogFile(options = {}) {
+    initNewHeapAnalysisLogFile(options = FileManager.defaultFileOption) {
         const dir = this.getHeapAnalysisLogDir(options);
         const file = path_1.default.join(dir, `analysis-${Utils_1.default.getUniqueID()}-out.log`);
         if (!fs_extra_1.default.existsSync(file)) {
@@ -222,10 +233,10 @@ class FileManager {
     getAndInitTSCompileIntermediateDir() {
         const dir = path_1.default.join(this.getTmpDir(), 'memlab-code');
         this.rmDir(dir);
-        fs_extra_1.default.mkdirSync(dir);
+        fs_extra_1.default.mkdirpSync(dir);
         return dir;
     }
-    clearDataDirs(options = {}) {
+    clearDataDirs(options = FileManager.defaultFileOption) {
         const curDataDir = this.getCurDataDir(options);
         if (!fs_extra_1.default.existsSync(curDataDir)) {
             return;
@@ -244,12 +255,29 @@ class FileManager {
             }
         }
     }
+    removeSnapshotFiles(options = FileManager.defaultFileOption) {
+        const curDataDir = this.getCurDataDir(options);
+        if (!fs_extra_1.default.existsSync(curDataDir)) {
+            return;
+        }
+        const dataSuffix = ['.heapsnapshot'];
+        const files = fs_extra_1.default.readdirSync(curDataDir);
+        for (const file of files) {
+            inner: for (const suffix of dataSuffix) {
+                if (file.endsWith(suffix)) {
+                    const filepath = path_1.default.join(curDataDir, file);
+                    fs_extra_1.default.unlinkSync(filepath);
+                    break inner;
+                }
+            }
+        }
+    }
     emptyDirIfExists(dir) {
         if (this.isDirectory(dir)) {
             fs_extra_1.default.emptyDirSync(dir);
         }
     }
-    emptyTraceLogDataDir(options = {}) {
+    emptyTraceLogDataDir(options = FileManager.defaultFileOption) {
         // all leak trace clusters
         this.emptyDirIfExists(this.getTraceClustersDir(options));
         // all JSON files for trace visualization
@@ -290,7 +318,7 @@ class FileManager {
             this.iterateAllFiles(filepath, callback);
         });
     }
-    rmWorkDir(options = {}) {
+    rmWorkDir(options = FileManager.defaultFileOption) {
         try {
             this.rmDir(this.getWorkDir(options));
         }
@@ -303,11 +331,16 @@ class FileManager {
         const internalDir = Constant_1.default.internalDir;
         return filePath.includes(`${sep}${internalDir}${sep}`);
     }
-    initDirs(config, options = {}) {
+    initDirs(config, options = FileManager.defaultFileOption) {
+        // cache the last processed memlab config instance
+        // the instance should be a singleton
+        this.memlabConfigCache = config;
         config.monoRepoDir = Constant_1.default.monoRepoDir;
         // make sure getWorkDir is called first before
         // any other get file or get dir calls
         const workDir = this.getWorkDir(options);
+        // remember the current working directory
+        // especially if this is a transcient working directory
         config.workDir = joinAndProcessDir(options, workDir);
         options = Object.assign(Object.assign({}, options), { workDir });
         config.dataBaseDir = joinAndProcessDir(options, this.getDataBaseDir(options));
@@ -356,4 +389,5 @@ class FileManager {
 }
 exports.FileManager = FileManager;
 FileManager.transientInstanceIdx = 0;
+FileManager.defaultFileOption = {};
 exports.default = new FileManager();

package/dist/lib/HeapAnalyzer.d.ts

@@ -7,17 +7,25 @@
  * @format
  * @oncall web_perf_infra
  */
-import type { E2EStepInfo, HeapNodeIdSet, IHeapSnapshot, IMemoryAnalystOptions, IMemoryAnalystSnapshotDiff, IOveralHeapInfo, LeakTracePathItem, Optional, IOveralLeakInfo, TraceCluster, ISerializedInfo } from './Types';
+import type { E2EStepInfo, HeapNodeIdSet, IHeapSnapshot, IMemoryAnalystSnapshotDiff, IOveralHeapInfo, LeakTracePathItem, Optional, IOveralLeakInfo, TraceCluster, ISerializedInfo, DiffLeakOptions } from './Types';
 import TraceFinder from '../paths/TraceFinder';
+declare type DiffSnapshotsOptions = {
+    loadAllSnapshots?: boolean;
+    workDir?: string;
+};
+declare type WorkDirOptions = {
+    workDir?: string;
+};
 declare class MemoryAnalyst {
     checkLeak(): Promise<ISerializedInfo[]>;
+    diffLeakByWorkDir(options: DiffLeakOptions): Promise<ISerializedInfo[]>;
+    diffMemoryLeakTraces(options: DiffLeakOptions): Promise<ISerializedInfo[]>;
     detectMemoryLeaks(): Promise<ISerializedInfo[]>;
-    visualizeMemoryUsage(options?: IMemoryAnalystOptions): void;
     focus(options?: {
         file?: string;
     }): Promise<void>;
     shouldLoadCompleteSnapshot(tabsOrder: E2EStepInfo[], tab: E2EStepInfo): boolean;
-    diffSnapshots(loadAll?: boolean): Promise<IMemoryAnalystSnapshotDiff>;
+    diffSnapshots(options?: DiffSnapshotsOptions): Promise<IMemoryAnalystSnapshotDiff>;
     preparePathFinder(snapshot: IHeapSnapshot): TraceFinder;
     private dumpPageInteractionSummary;
     private dumpLeakSummaryToConsole;
@@ -29,12 +37,11 @@ declare class MemoryAnalyst {
     printHeapInfo(leakInfo: IOveralHeapInfo): void;
     private printHeapAndLeakInfo;
     private logLeakTraceSummary;
-    searchLeakedTraces(leakedNodeIds: HeapNodeIdSet, snapshot: IHeapSnapshot): Promise<{
-        paths: LeakTracePathItem[];
-    }>;
+    filterLeakPaths(leakedNodeIds: HeapNodeIdSet, snapshot: IHeapSnapshot, options?: WorkDirOptions): LeakTracePathItem[];
+    findLeakTraces(leakedNodeIds: HeapNodeIdSet, snapshot: IHeapSnapshot, options?: WorkDirOptions): Promise<LeakTracePathItem[]>;
     /**
      * Given a set of heap object ids, cluster them based on the similarity
-     * of their retainer traces and return a
+     * of their retainer traces
      * @param leakedNodeIds
      * @param snapshot
      * @returns
@@ -43,7 +50,7 @@ declare class MemoryAnalyst {
     serializeClusterUpdate(clusters: TraceCluster[], options?: {
         reclusterOnly?: boolean;
     }): Promise<void>;
-    dumpPathByNodeId(leakedIdSet: HeapNodeIdSet, snapshot: IHeapSnapshot, nodeIdsInSnapshots: Array<HeapNodeIdSet>, id: number, pathLoaderFile: string, summaryFile: string): void;
+    dumpPathByNodeId(leakedIdSet: HeapNodeIdSet, snapshot: IHeapSnapshot, nodeIdsInSnapshots: Array<HeapNodeIdSet>, id: number, pathLoaderFile: string, summaryFile: string, options?: WorkDirOptions): void;
 }
 declare const _default: MemoryAnalyst;
 export default _default;

package/dist/lib/HeapAnalyzer.js

@@ -22,77 +22,83 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const fs_1 = __importDefault(require("fs"));
-const babar_1 = __importDefault(require("babar"));
-const LeakClusterLogger_1 = __importDefault(require("../logger/LeakClusterLogger"));
-const LeakTraceDetailsLogger_1 = __importDefault(require("../logger/LeakTraceDetailsLogger"));
-const TraceFinder_1 = __importDefault(require("../paths/TraceFinder"));
-const TraceBucket_1 = __importDefault(require("../trace-cluster/TraceBucket"));
 const Config_1 = __importDefault(require("./Config"));
 const Console_1 = __importDefault(require("./Console"));
 const Serializer_1 = __importDefault(require("./Serializer"));
 const Utils_1 = __importDefault(require("./Utils"));
+const FileManager_1 = __importDefault(require("./FileManager"));
+const MemoryBarChart_1 = __importDefault(require("./charts/MemoryBarChart"));
+const LeakClusterLogger_1 = __importDefault(require("../logger/LeakClusterLogger"));
+const LeakTraceDetailsLogger_1 = __importDefault(require("../logger/LeakTraceDetailsLogger"));
+const TraceFinder_1 = __importDefault(require("../paths/TraceFinder"));
+const TraceBucket_1 = __importDefault(require("../trace-cluster/TraceBucket"));
 const LeakObjectFilter_1 = require("./leak-filters/LeakObjectFilter");
 const MLTraceSimilarityStrategy_1 = __importDefault(require("../trace-cluster/strategies/MLTraceSimilarityStrategy"));
 class MemoryAnalyst {
     checkLeak() {
         return __awaiter(this, void 0, void 0, function* () {
-            this.visualizeMemoryUsage();
+            MemoryBarChart_1.default.plotMemoryBarChart();
             Utils_1.default.checkSnapshots();
             return yield this.detectMemoryLeaks();
         });
     }
+    diffLeakByWorkDir(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const controlSnapshotDir = FileManager_1.default.getCurDataDir({
+                workDir: options.controlWorkDir,
+            });
+            const treatmentSnapshotDir = FileManager_1.default.getCurDataDir({
+                workDir: options.treatmentWorkDir,
+            });
+            // check control working dir
+            Utils_1.default.checkSnapshots({ snapshotDir: controlSnapshotDir });
+            // check treatment working dir
+            Utils_1.default.checkSnapshots({ snapshotDir: treatmentSnapshotDir });
+            // display control and treatment memory
+            MemoryBarChart_1.default.plotMemoryBarChart(options);
+            return this.diffMemoryLeakTraces(options);
+        });
+    }
+    // find all unique pattern of leaks
+    diffMemoryLeakTraces(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            Config_1.default.dumpNodeInfo = false;
+            // diff snapshots and get control raw paths
+            let snapshotDiff = yield this.diffSnapshots({
+                loadAllSnapshots: true,
+                workDir: options.controlWorkDir,
+            });
+            const controlLeakPaths = this.filterLeakPaths(snapshotDiff.leakedHeapNodeIdSet, snapshotDiff.snapshot, { workDir: options.controlWorkDir });
+            const controlSnapshot = snapshotDiff.snapshot;
+            // diff snapshots and get treatment raw paths
+            snapshotDiff = yield this.diffSnapshots({
+                loadAllSnapshots: true,
+                workDir: options.treatmentWorkDir,
+            });
+            const treatmentLeakPaths = this.filterLeakPaths(snapshotDiff.leakedHeapNodeIdSet, snapshotDiff.snapshot, { workDir: options.controlWorkDir });
+            const treatmentSnapshot = snapshotDiff.snapshot;
+            Console_1.default.topLevel(`${controlLeakPaths.length} traces from control group`);
+            Console_1.default.topLevel(`${treatmentLeakPaths.length} traces from treatment group`);
+            const result = TraceBucket_1.default.clusterControlTreatmentPaths(controlLeakPaths, controlSnapshot, treatmentLeakPaths, treatmentSnapshot, Utils_1.default.aggregateDominatorMetrics, {
+                strategy: Config_1.default.isMLClustering
+                    ? new MLTraceSimilarityStrategy_1.default()
+                    : undefined,
+            });
+            Console_1.default.midLevel(`MemLab found ${result.treatmentOnlyClusters.length} new leak(s) in the treatment group`);
+            yield this.serializeClusterUpdate(result.treatmentOnlyClusters);
+            // TODO (lgong): log leak traces
+            return [];
+        });
+    }
     // find all unique pattern of leaks
     detectMemoryLeaks() {
         return __awaiter(this, void 0, void 0, function* () {
-            const snapshotDiff = yield this.diffSnapshots(true);
+            const snapshotDiff = yield this.diffSnapshots({ loadAllSnapshots: true });
             Config_1.default.dumpNodeInfo = false;
-            const { paths } = yield this.searchLeakedTraces(snapshotDiff.leakedHeapNodeIdSet, snapshotDiff.snapshot);
+            const paths = yield this.findLeakTraces(snapshotDiff.leakedHeapNodeIdSet, snapshotDiff.snapshot);
             return LeakTraceDetailsLogger_1.default.logTraces(snapshotDiff.leakedHeapNodeIdSet, snapshotDiff.snapshot, snapshotDiff.listOfLeakedHeapNodeIdSet, paths, Config_1.default.traceJsonOutDir);
         });
     }
-    visualizeMemoryUsage(options = {}) {
-        if (Config_1.default.useExternalSnapshot || options.snapshotDir) {
-            return;
-        }
-        const tabsOrder = Utils_1.default.loadTabsOrder();
-        // if memory usage data is incomplete, skip the visualization
-        for (const tab of tabsOrder) {
-            if (!(tab.JSHeapUsedSize > 0)) {
-                if (Config_1.default.verbose) {
-                    Console_1.default.error('Memory usage data incomplete');
-                }
-                return;
-            }
-        }
-        const plotData = tabsOrder.map((tab, idx) => [
-            idx + 1,
-            ((tab.JSHeapUsedSize / 100000) | 0) / 10,
-        ]);
-        // the graph component cannot handle an array with a single element
-        while (plotData.length < 2) {
-            plotData.push([plotData.length + 1, 0]);
-        }
-        // plot visual settings
-        const minY = 1;
-        const maxY = plotData.reduce((m, v) => Math.max(m, v[1]), 0) * 1.15;
-        const yFractions = 1;
-        const yLabelWidth = 1 +
-            Math.max(minY.toFixed(yFractions).length, maxY.toFixed(yFractions).length);
-        const maxWidth = process.stdout.columns - 10;
-        const idealWidth = Math.max(2 * plotData.length + 2 * yLabelWidth, 10);
-        const plotWidth = Math.min(idealWidth, maxWidth);
-        Console_1.default.topLevel('Memory usage across all steps:');
-        Console_1.default.topLevel((0, babar_1.default)(plotData, {
-            color: 'green',
-            width: plotWidth,
-            height: 10,
-            xFractions: 0,
-            yFractions,
-            minY,
-            maxY,
-        }));
-        Console_1.default.topLevel('');
-    }
     focus(options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
             Console_1.default.overwrite(`Generating report for node @${Config_1.default.focusFiberNodeId}`);
@@ -113,7 +119,7 @@ class MemoryAnalyst {
             }
             else {
                 Utils_1.default.checkSnapshots();
-                const snapshotDiff = yield this.diffSnapshots(true);
+                const snapshotDiff = yield this.diffSnapshots({ loadAllSnapshots: true });
                 nodeIdsInSnapshots = snapshotDiff.listOfLeakedHeapNodeIdSet;
                 snapshotLeakedHeapNodeIdSet = snapshotDiff.leakedHeapNodeIdSet;
                 snapshot = snapshotDiff.snapshot;
@@ -130,16 +136,16 @@ class MemoryAnalyst {
         }
         return false;
     }
-    diffSnapshots(loadAll = false) {
+    diffSnapshots(options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
             const nodeIdsInSnapshots = [];
-            const tabsOrder = Utils_1.default.loadTabsOrder();
+            const tabsOrder = Utils_1.default.loadTabsOrder(FileManager_1.default.getSnapshotSequenceMetaFile(options));
             // a set keeping track of node ids generated before the target snapshot
             const baselineIds = new Set();
             let collectBaselineIds = true;
             let targetAllocatedHeapNodeIdSet = null;
             let leakedHeapNodeIdSet = null;
-            const options = { verbose: true };
+            const parseSnapshotOptions = { verbose: true, workDir: options.workDir };
             let snapshot = null;
             for (let i = 0; i < tabsOrder.length; i++) {
                 const tab = tabsOrder[i];
@@ -157,13 +163,13 @@ class MemoryAnalyst {
                     continue;
                 }
                 // in quick mode, there is no need to load all snapshots
-                if (!loadAll && !tab.type) {
+                if (!options.loadAllSnapshots && !tab.type) {
                     continue;
                 }
-                const file = Utils_1.default.getSnapshotFilePath(tab);
+                const file = Utils_1.default.getSnapshotFilePath(tab, options);
                 if (this.shouldLoadCompleteSnapshot(tabsOrder, tab)) {
                     // final snapshot needs to build node index
-                    const opt = Object.assign({ buildNodeIdIndex: true }, options);
+                    const opt = Object.assign(Object.assign({ buildNodeIdIndex: true }, parseSnapshotOptions), { workDir: options.workDir });
                     snapshot = yield Utils_1.default.getSnapshotFromFile(file, opt);
                     // record Ids in the snapshot
                     snapshot.nodes.forEach(node => {
@@ -171,7 +177,7 @@ class MemoryAnalyst {
                     });
                 }
                 else {
-                    idsInSnapshot = yield Utils_1.default.getSnapshotNodeIdsFromFile(file, options);
+                    idsInSnapshot = yield Utils_1.default.getSnapshotNodeIdsFromFile(file, parseSnapshotOptions);
                     nodeIdsInSnapshots.pop();
                     nodeIdsInSnapshots.push(idsInSnapshot);
                 }
@@ -234,10 +240,10 @@ class MemoryAnalyst {
         return finder;
     }
     // summarize the page interaction and dump to the leak text summary file
-    dumpPageInteractionSummary() {
-        const tabsOrder = Utils_1.default.loadTabsOrder();
+    dumpPageInteractionSummary(options = {}) {
+        const tabsOrder = Utils_1.default.loadTabsOrder(FileManager_1.default.getSnapshotSequenceMetaFile(options));
         const tabsOrderStr = Serializer_1.default.summarizeTabsOrder(tabsOrder);
-        fs_1.default.writeFileSync(Config_1.default.exploreResultFile, tabsOrderStr, 'UTF-8');
+        fs_1.default.writeFileSync(FileManager_1.default.getLeakSummaryFile(options), tabsOrderStr, 'UTF-8');
     }
     // summarize the leak and print the info in console
     dumpLeakSummaryToConsole(leakedNodeIds, snapshot) {
@@ -318,9 +324,9 @@ class MemoryAnalyst {
             Console_1.default.topLevel(`· ${name}: ${value}`);
         });
     }
-    printHeapAndLeakInfo(leakedNodeIds, snapshot) {
+    printHeapAndLeakInfo(leakedNodeIds, snapshot, options = {}) {
         // write page interaction summary to the leaks text file
-        this.dumpPageInteractionSummary();
+        this.dumpPageInteractionSummary(options);
         // dump leak summry to console
         this.dumpLeakSummaryToConsole(leakedNodeIds, snapshot);
         // get aggregated leak info
@@ -329,42 +335,46 @@ class MemoryAnalyst {
             this.printHeapInfo(heapInfo);
         }
     }
-    logLeakTraceSummary(trace, nodeIdInPaths, snapshot) {
+    logLeakTraceSummary(trace, nodeIdInPaths, snapshot, options = {}) {
         if (!Config_1.default.isFullRun) {
             return;
         }
         // convert the path to a string
         const pathStr = Serializer_1.default.summarizePath(trace, nodeIdInPaths, snapshot);
-        fs_1.default.appendFileSync(Config_1.default.exploreResultFile, `\n\n${pathStr}\n\n`, 'UTF-8');
+        fs_1.default.appendFileSync(FileManager_1.default.getLeakSummaryFile(options), `\n\n${pathStr}\n\n`, 'UTF-8');
+    }
+    filterLeakPaths(leakedNodeIds, snapshot, options = {}) {
+        const finder = this.preparePathFinder(snapshot);
+        this.printHeapAndLeakInfo(leakedNodeIds, snapshot, options);
+        // get all leaked objects
+        this.filterLeakedObjects(leakedNodeIds, snapshot);
+        const nodeIdInPaths = new Set();
+        const paths = [];
+        let numOfLeakedObjects = 0;
+        let i = 0;
+        // analysis for each node
+        Utils_1.default.applyToNodes(leakedNodeIds, snapshot, node => {
+            if (!Config_1.default.isContinuousTest && ++i % 11 === 0) {
+                Console_1.default.overwrite(`progress: ${i} / ${leakedNodeIds.size} @${node.id}`);
+            }
+            // BFS search for path from the leaked node to GC roots
+            const p = finder.getPathToGCRoots(snapshot, node);
+            if (!p || !Utils_1.default.isInterestingPath(p)) {
+                return;
+            }
+            ++numOfLeakedObjects;
+            paths.push(p);
+            this.logLeakTraceSummary(p, nodeIdInPaths, snapshot, options);
+        }, { reverse: true });
+        if (Config_1.default.verbose) {
+            Console_1.default.midLevel(`${numOfLeakedObjects} leaked objects`);
+        }
+        return paths;
     }
     // find unique paths of leaked nodes
-    searchLeakedTraces(leakedNodeIds, snapshot) {
+    findLeakTraces(leakedNodeIds, snapshot, options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
-            const finder = this.preparePathFinder(snapshot);
-            this.printHeapAndLeakInfo(leakedNodeIds, snapshot);
-            // get all leaked objects
-            this.filterLeakedObjects(leakedNodeIds, snapshot);
-            const nodeIdInPaths = new Set();
-            const paths = [];
-            let numOfLeakedObjects = 0;
-            let i = 0;
-            // analysis for each node
-            Utils_1.default.applyToNodes(leakedNodeIds, snapshot, node => {
-                if (!Config_1.default.isContinuousTest && ++i % 11 === 0) {
-                    Console_1.default.overwrite(`progress: ${i} / ${leakedNodeIds.size} @${node.id}`);
-                }
-                // BFS search for path from the leaked node to GC roots
-                const p = finder.getPathToGCRoots(snapshot, node);
-                if (!p || !Utils_1.default.isInterestingPath(p)) {
-                    return;
-                }
-                ++numOfLeakedObjects;
-                paths.push(p);
-                this.logLeakTraceSummary(p, nodeIdInPaths, snapshot);
-            }, { reverse: true });
-            if (Config_1.default.verbose) {
-                Console_1.default.midLevel(`${numOfLeakedObjects} leaked objects`);
-            }
+            const paths = this.filterLeakPaths(leakedNodeIds, snapshot, options);
             // cluster traces from the current run
             const clusters = TraceBucket_1.default.clusterPaths(paths, snapshot, Utils_1.default.aggregateDominatorMetrics, {
                 strategy: Config_1.default.isMLClustering
@@ -378,14 +388,12 @@ class MemoryAnalyst {
                 const clustersUnclassified = TraceBucket_1.default.generateUnClassifiedClusters(paths, snapshot, Utils_1.default.aggregateDominatorMetrics);
                 LeakClusterLogger_1.default.logUnclassifiedClusters(clustersUnclassified);
             }
-            return {
-                paths: clusters.map(c => c.path),
-            };
+            return clusters.map(c => c.path);
         });
     }
     /**
      * Given a set of heap object ids, cluster them based on the similarity
-     * of their retainer traces and return a
+     * of their retainer traces
      * @param leakedNodeIds
      * @param snapshot
      * @returns
@@ -429,7 +437,7 @@ class MemoryAnalyst {
             }
         });
     }
-    dumpPathByNodeId(leakedIdSet, snapshot, nodeIdsInSnapshots, id, pathLoaderFile, summaryFile) {
+    dumpPathByNodeId(leakedIdSet, snapshot, nodeIdsInSnapshots, id, pathLoaderFile, summaryFile, options = {}) {
         Console_1.default.overwrite('start analysis...');
         const finder = this.preparePathFinder(snapshot);
         const nodeIdInPaths = new Set();
@@ -443,7 +451,7 @@ class MemoryAnalyst {
                 return;
             }
             LeakTraceDetailsLogger_1.default.logTrace(leakedIdSet, snapshot, nodeIdsInSnapshots, path, pathLoaderFile);
-            const tabsOrder = Utils_1.default.loadTabsOrder();
+            const tabsOrder = Utils_1.default.loadTabsOrder(FileManager_1.default.getSnapshotSequenceMetaFile(options));
             const interactionSummary = Serializer_1.default.summarizeTabsOrder(tabsOrder);
             let pathSummary = Serializer_1.default.summarizePath(path, nodeIdInPaths, snapshot, { color: true });
             Console_1.default.topLevel(pathSummary);

package/dist/lib/Types.d.ts

@@ -43,6 +43,12 @@ export declare type HaltOrThrowOptions = {
     printCallback?: () => void;
 };
 /** @internal */
+export declare type FileOption = {
+    workDir?: Optional<string>;
+    clear?: boolean;
+    transient?: boolean;
+};
+/** @internal */
 export declare type CLIOptions = {
     cliArgs: ParsedArgs;
     configFromOptions?: AnyRecord;
@@ -795,6 +801,15 @@ export declare type TraceClusterMetaInfo = {
     meta_data: string;
 };
 /** @internal */
+export declare type ControlTreatmentClusterResult = {
+    controlOnlyClusters: TraceCluster[];
+    treatmentOnlyClusters: TraceCluster[];
+    hybridClusters: Array<{
+        control: TraceCluster;
+        treatment: TraceCluster;
+    }>;
+};
+/** @internal */
 export interface E2EInteraction {
     kind: string;
     timeout?: number;
@@ -1835,6 +1850,17 @@ export interface IOveralLeakInfo extends Partial<IOveralHeapInfo> {
     leakedAlternateFiberNodeSize: number;
 }
 /** @internal */
+export declare type DiffLeakOptions = {
+    controlWorkDir: string;
+    treatmentWorkDir: string;
+};
+/** @internal */
+export declare type PlotMemoryOptions = {
+    controlWorkDir?: string;
+    treatmentWorkDir?: string;
+    workDir?: string;
+} & IMemoryAnalystOptions;
+/** @internal */
 export interface IMemoryAnalystOptions {
     snapshotDir?: string;
     minSnapshots?: number;

package/dist/lib/Utils.d.ts

@@ -95,7 +95,9 @@ declare function checkSnapshots(options?: {
 export declare function resolveSnapshotFilePath(snapshotFile: Nullable<string>): string;
 declare function getSnapshotDirForAnalysis(): string;
 declare function getSingleSnapshotFileForAnalysis(): string;
-declare function getSnapshotFilePath(tab: E2EStepInfo): string;
+declare function getSnapshotFilePath(tab: E2EStepInfo, options?: {
+    workDir?: string;
+}): string;
 declare function equalOrMatch(v1: any, v2: any): boolean;
 declare function getSnapshotFilePathWithTabType(type: string | RegExp): Nullable<string>;
 declare function isMeaningfulNode(node: IHeapNode): boolean;
@@ -122,7 +124,7 @@ declare function getError(maybeError: unknown): Error;
 declare function isNodeDominatedByDeletionsArray(node: IHeapNode): boolean;
 declare function getUniqueID(): string;
 declare function getClosureSourceUrl(node: IHeapNode): Nullable<string>;
-export declare function runShell(command: string, options?: ShellOptions): string;
+export declare function runShell(command: string, options?: ShellOptions): Nullable<string>;
 export declare function getRetainedSize(node: IHeapNode): number;
 export declare function aggregateDominatorMetrics(ids: HeapNodeIdSet, snapshot: IHeapSnapshot, checkNodeCb: (node: IHeapNode) => boolean, nodeMetricsCb: (node: IHeapNode) => number): number;
 declare const _default: {

package/dist/lib/Utils.js

@@ -1346,13 +1346,17 @@ function getSingleSnapshotFileForAnalysis() {
     }
     return resolveSnapshotFilePath(path);
 }
-function getSnapshotFilePath(tab) {
+function getSnapshotFilePath(tab, options = {}) {
+    const fileName = `s${tab.idx}.heapsnapshot`;
+    if (options.workDir) {
+        return path_1.default.join(FileManager_1.default.getCurDataDir(options), fileName);
+    }
     if (!Config_1.default.useExternalSnapshot) {
-        return path_1.default.join(Config_1.default.curDataDir, `s${tab.idx}.heapsnapshot`);
+        return path_1.default.join(Config_1.default.curDataDir, fileName);
     }
     // if we are loading snapshot from external snapshot dir
     if (Config_1.default.externalSnapshotDir) {
-        return path_1.default.join(Config_1.default.externalSnapshotDir, `s${tab.idx}.heapsnapshot`);
+        return path_1.default.join(Config_1.default.externalSnapshotDir, fileName);
     }
     return Config_1.default.externalSnapshotFilePaths[tab.idx - 1];
 }
@@ -1764,7 +1768,7 @@ function runShell(command, options = {}) {
     if (process_1.default.platform !== 'win32') {
         execOptions.shell = '/bin/bash';
     }
-    let ret = '';
+    let ret = null;
     try {
         ret = child_process_1.default.execSync(command, execOptions);
     }

package/dist/lib/charts/MemoryBarChart.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import type { PlotMemoryOptions } from '../Types';
+declare class MemoryBarChart {
+    plotMemoryBarChart(options?: PlotMemoryOptions): void;
+    private loadPlotDataFromTabsOrder;
+    private loadPlotDataFromWorkDir;
+    private loadPlotData;
+    private mergePlotData;
+}
+declare const _default: MemoryBarChart;
+export default _default;
+//# sourceMappingURL=MemoryBarChart.d.ts.map

package/dist/lib/charts/MemoryBarChart.js

@@ -0,0 +1,110 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+'use strict';
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const babar_1 = __importDefault(require("babar"));
+const Config_1 = __importDefault(require("../Config"));
+const Console_1 = __importDefault(require("../Console"));
+const Utils_1 = __importDefault(require("../Utils"));
+const FileManager_1 = __importDefault(require("../FileManager"));
+class MemoryBarChart {
+    plotMemoryBarChart(options = {}) {
+        if (Config_1.default.useExternalSnapshot || options.snapshotDir) {
+            return;
+        }
+        let plotData;
+        try {
+            plotData = this.loadPlotData(options);
+        }
+        catch (ex) {
+            Console_1.default.warning(`plot data not load correctly: ${Utils_1.default.getError(ex).message}`);
+            return;
+        }
+        // normalize plot data
+        const minY = 1;
+        const maxY = plotData.reduce((m, v) => Math.max(m, v[1]), 0) * 1.15;
+        const yFractions = 1;
+        const yLabelWidth = 1 +
+            Math.max(minY.toFixed(yFractions).length, maxY.toFixed(yFractions).length);
+        const maxWidth = process.stdout.columns - 10;
+        const idealWidth = Math.max(2 * plotData.length + 2 * yLabelWidth, 10);
+        const plotWidth = Math.min(idealWidth, maxWidth);
+        Console_1.default.topLevel('Memory usage across all steps:');
+        Console_1.default.topLevel((0, babar_1.default)(plotData, {
+            color: 'green',
+            width: plotWidth,
+            height: 10,
+            xFractions: 0,
+            yFractions,
+            minY,
+            maxY,
+        }));
+        Console_1.default.topLevel('');
+    }
+    loadPlotDataFromTabsOrder(tabsOrder) {
+        for (const tab of tabsOrder) {
+            if (!(tab.JSHeapUsedSize > 0)) {
+                if (Config_1.default.verbose) {
+                    Console_1.default.error('Memory usage data incomplete');
+                }
+                return [];
+            }
+        }
+        const plotData = tabsOrder.map((tab, idx) => [
+            idx + 1,
+            ((tab.JSHeapUsedSize / 100000) | 0) / 10,
+        ]);
+        // the graph component cannot handle an array with a single element
+        while (plotData.length < 2) {
+            plotData.push([plotData.length + 1, 0]);
+        }
+        return plotData;
+    }
+    loadPlotDataFromWorkDir(options = {}) {
+        const tabsOrder = Utils_1.default.loadTabsOrder(FileManager_1.default.getSnapshotSequenceMetaFile(options));
+        return this.loadPlotDataFromTabsOrder(tabsOrder);
+    }
+    loadPlotData(options = {}) {
+        // plot data for a single run
+        if (!options.controlWorkDir && !options.treatmentWorkDir) {
+            return this.loadPlotDataFromWorkDir(options);
+        }
+        // plot data for control and test run
+        const controlPlotData = this.loadPlotDataFromWorkDir({
+            workDir: options.controlWorkDir,
+        });
+        const testPlotData = this.loadPlotDataFromWorkDir({
+            workDir: options.treatmentWorkDir,
+        });
+        // merge plot data
+        return this.mergePlotData([controlPlotData, testPlotData]);
+    }
+    mergePlotData(plotDataArray) {
+        const plotData = [];
+        let xIndex = 1; // starts from 1
+        for (let i = 0; i < plotDataArray.length; ++i) {
+            const data = plotDataArray[i];
+            for (const [, yValue] of data) {
+                plotData.push([xIndex++, yValue]);
+            }
+            // push blank separators
+            if (i < plotDataArray.length - 1) {
+                for (let k = 0; k < 3; ++k) {
+                    plotData.push([xIndex++, 0]);
+                }
+            }
+        }
+        return plotData;
+    }
+}
+exports.default = new MemoryBarChart();

package/dist/trace-cluster/TraceBucket.d.ts

@@ -7,13 +7,16 @@
  * @format
  * @oncall web_perf_infra
  */
-import type { IHeapNode, IHeapSnapshot, LeakTrace, LeakTracePathItem, Optional, TraceCluster, TraceClusterDiff, IClusterStrategy } from '../lib/Types';
+import type { IHeapNode, IHeapSnapshot, LeakTrace, LeakTracePathItem, Optional, TraceCluster, TraceClusterDiff, IClusterStrategy, ControlTreatmentClusterResult } from '../lib/Types';
 import type { NormalizedTraceElement } from './TraceElement';
 declare type AggregateNodeCb = (ids: Set<number>, snapshot: IHeapSnapshot, checkCb: (node: IHeapNode) => boolean, calculateCb: (node: IHeapNode) => number) => number;
 export default class NormalizedTrace {
     private trace;
     private traceSummary;
     constructor(p?: LeakTracePathItem | null, snapshot?: IHeapSnapshot | null);
+    static getPathLastNode(p: LeakTracePathItem, options?: {
+        untilFirstDetachedDOMElem?: boolean;
+    }): Optional<IHeapNode>;
     static pathToTrace(p: LeakTracePathItem, options?: {
         untilFirstDetachedDOMElem?: boolean;
     }): NormalizedTraceElement[];
@@ -30,6 +33,12 @@ export default class NormalizedTrace {
     static clusterPaths(paths: LeakTracePathItem[], snapshot: IHeapSnapshot, aggregateDominatorMetrics: AggregateNodeCb, option?: {
         strategy?: IClusterStrategy;
     }): TraceCluster[];
+    private static buildTraceToPathMap;
+    private static pushLeakPathToCluster;
+    private static initEmptyCluster;
+    static clusterControlTreatmentPaths(controlPaths: LeakTracePathItem[], controlSnapshot: IHeapSnapshot, treatmentPaths: LeakTracePathItem[], treatmentSnapshot: IHeapSnapshot, aggregateDominatorMetrics: AggregateNodeCb, option?: {
+        strategy?: IClusterStrategy;
+    }): ControlTreatmentClusterResult;
     static generateUnClassifiedClusters(paths: LeakTracePathItem[], snapshot: IHeapSnapshot, aggregateDominatorMetrics: AggregateNodeCb): TraceCluster[];
     static loadCluster(): NormalizedTrace[];
     static saveCluster(clusters: NormalizedTrace[]): void;

package/dist/trace-cluster/TraceBucket.js

@@ -38,6 +38,27 @@ class NormalizedTrace {
                 : '';
         }
     }
+    static getPathLastNode(p, options = {}) {
+        const skipRest = !!options.untilFirstDetachedDOMElem;
+        const shouldSkip = (node) => {
+            // only consider the trace from GC root to the first detached element
+            // NOTE: do not use utils.isDetachedDOMNode, which relies on
+            //       the fact that p.node is a HeapNode
+            return (skipRest &&
+                node.name.startsWith('Detached ') &&
+                node.name !== 'Detached InternalNode');
+        };
+        let curItem = p;
+        while (curItem.next) {
+            if (curItem.node) {
+                if (shouldSkip(curItem.node)) {
+                    break;
+                }
+            }
+            curItem = curItem.next;
+        }
+        return curItem === null || curItem === void 0 ? void 0 : curItem.node;
+    }
     // convert path to leak trace
     static pathToTrace(p, options = {}) {
         const skipRest = !!options.untilFirstDetachedDOMElem;
@@ -114,6 +135,13 @@ class NormalizedTrace {
             if (Math.random() < sampleRatio) {
                 ret.push(p);
             }
+            else {
+                // force sample objects with non-trvial self size
+                const lastNode = NormalizedTrace.getPathLastNode(p);
+                if (lastNode && lastNode.self_size >= 100000) {
+                    ret.push(p);
+                }
+            }
         }
         return ret;
     }
@@ -233,6 +261,109 @@ class NormalizedTrace {
         clusters.sort((c1, c2) => { var _a, _b; return ((_a = c2.retainedSize) !== null && _a !== void 0 ? _a : 0) - ((_b = c1.retainedSize) !== null && _b !== void 0 ? _b : 0); });
         return clusters;
     }
+    static buildTraceToPathMap(paths) {
+        const traceToPathMap = new Map();
+        for (const p of paths) {
+            const trace = NormalizedTrace.pathToTrace(p, {
+                untilFirstDetachedDOMElem: true,
+            });
+            traceToPathMap.set(trace, p);
+        }
+        return traceToPathMap;
+    }
+    static pushLeakPathToCluster(traceToPathMap, trace, cluster) {
+        // if this is a control path, update control cluster
+        const curPath = traceToPathMap.get(trace);
+        if (cluster.count === 0) {
+            cluster.path = curPath;
+            // add representative object id if there is one
+            const lastNode = trace[trace.length - 1];
+            if ('id' in lastNode) {
+                cluster.id = lastNode.id;
+            }
+        }
+        cluster.count = cluster.count + 1;
+        NormalizedTrace.addLeakedNodeToCluster(cluster, curPath);
+    }
+    static initEmptyCluster(snapshot) {
+        return {
+            path: {},
+            count: 0,
+            snapshot,
+            retainedSize: 0,
+            leakedNodeIds: new Set(),
+        };
+    }
+    static clusterControlTreatmentPaths(controlPaths, controlSnapshot, treatmentPaths, treatmentSnapshot, aggregateDominatorMetrics, option = {}) {
+        const result = {
+            controlOnlyClusters: [],
+            treatmentOnlyClusters: [],
+            hybridClusters: [],
+        };
+        Console_1.default.overwrite('Clustering leak traces');
+        if (controlPaths.length === 0 && treatmentPaths.length === 0) {
+            Console_1.default.midLevel('No leaks found');
+            return result;
+        }
+        // sample paths if there are too many
+        controlPaths = this.samplePaths(controlPaths);
+        treatmentPaths = this.samplePaths(treatmentPaths);
+        // build control trace to control path map
+        const controlTraceToPathMap = NormalizedTrace.buildTraceToPathMap(controlPaths);
+        const controlTraces = Array.from(controlTraceToPathMap.keys());
+        // build treatment trace to treatment path map
+        const treatmentTraceToPathMap = NormalizedTrace.buildTraceToPathMap(treatmentPaths);
+        const treatmentTraces = Array.from(treatmentTraceToPathMap.keys());
+        // cluster traces from both the control group and the treatment group
+        const { allClusters } = NormalizedTrace.diffTraces([...controlTraces, ...treatmentTraces], [], option);
+        // construct TraceCluster from clustering result
+        allClusters.forEach((traces) => {
+            var _a, _b;
+            const controlCluster = NormalizedTrace.initEmptyCluster(controlSnapshot);
+            const treatmentCluster = NormalizedTrace.initEmptyCluster(treatmentSnapshot);
+            for (const trace of traces) {
+                const normalizedTrace = trace;
+                if (controlTraceToPathMap.has(normalizedTrace)) {
+                    NormalizedTrace.pushLeakPathToCluster(controlTraceToPathMap, normalizedTrace, controlCluster);
+                }
+                else {
+                    NormalizedTrace.pushLeakPathToCluster(treatmentTraceToPathMap, normalizedTrace, treatmentCluster);
+                }
+            }
+            const controlClusterSize = (_a = controlCluster.count) !== null && _a !== void 0 ? _a : 0;
+            const treatmentClusterSize = (_b = treatmentCluster.count) !== null && _b !== void 0 ? _b : 0;
+            // calculate aggregated cluster size for control cluster
+            if (controlClusterSize > 0) {
+                this.calculateClusterRetainedSize(controlCluster, controlSnapshot, aggregateDominatorMetrics);
+            }
+            // calculate aggregated cluster size for treatment cluster
+            if (treatmentClusterSize > 0) {
+                this.calculateClusterRetainedSize(treatmentCluster, treatmentSnapshot, aggregateDominatorMetrics);
+            }
+            if (controlClusterSize === 0) {
+                result.treatmentOnlyClusters.push(treatmentCluster);
+            }
+            else if (treatmentClusterSize === 0) {
+                result.controlOnlyClusters.push(controlCluster);
+            }
+            else {
+                result.hybridClusters.push({
+                    control: controlCluster,
+                    treatment: treatmentCluster,
+                });
+            }
+        });
+        result.treatmentOnlyClusters.sort((c1, c2) => { var _a, _b; return ((_a = c2.retainedSize) !== null && _a !== void 0 ? _a : 0) - ((_b = c1.retainedSize) !== null && _b !== void 0 ? _b : 0); });
+        result.controlOnlyClusters.sort((c1, c2) => { var _a, _b; return ((_a = c2.retainedSize) !== null && _a !== void 0 ? _a : 0) - ((_b = c1.retainedSize) !== null && _b !== void 0 ? _b : 0); });
+        result.hybridClusters.sort((g1, g2) => {
+            var _a, _b, _c, _d;
+            return ((_a = g2.control.retainedSize) !== null && _a !== void 0 ? _a : 0) +
+                ((_b = g2.treatment.retainedSize) !== null && _b !== void 0 ? _b : 0) -
+                ((_c = g1.control.retainedSize) !== null && _c !== void 0 ? _c : 0) -
+                ((_d = g1.treatment.retainedSize) !== null && _d !== void 0 ? _d : 0);
+        });
+        return result;
+    }
     static generateUnClassifiedClusters(paths, snapshot, aggregateDominatorMetrics) {
         return this.clusterPaths(paths, snapshot, aggregateDominatorMetrics, {
             strategy: new TraceAsClusterStrategy_1.default(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memlab/core",
-  "version": "1.1.17",
+  "version": "1.1.19",
   "license": "MIT",
   "description": "memlab core libraries",
   "author": "Liang Gong <lgong@fb.com>",