repomix/lib/shared/memoryUtils.js

A tool to pack repository contents to single file for AI consumption

import { logger } from './logger.js';
function bytesToMB(bytes) {
    return Math.round((bytes / 1024 / 1024) * 100) / 100;
}
export function getMemoryStats() {
    const usage = process.memoryUsage();
    const heapUsed = bytesToMB(usage.heapUsed);
    const heapTotal = bytesToMB(usage.heapTotal);
    const external = bytesToMB(usage.external);
    const rss = bytesToMB(usage.rss);
    const heapUsagePercent = Math.round((heapUsed / heapTotal) * 10000) / 100;
    return {
        heapUsed,
        heapTotal,
        external,
        rss,
        heapUsagePercent,
    };
}
export function logMemoryUsage(context) {
    const stats = getMemoryStats();
    logger.trace(`Memory [${context}] | Heap: ${stats.heapUsed}/${stats.heapTotal}MB (${stats.heapUsagePercent}%) | RSS: ${stats.rss}MB | Ext: ${stats.external}MB`);
}
export function logMemoryDifference(context, before, after) {
    const heapDiff = after.heapUsed - before.heapUsed;
    const rssDiff = after.rss - before.rss;
    const externalDiff = after.external - before.external;
    const formatDiff = (diff) => `${diff >= 0 ? '+' : ''}${diff.toFixed(2)}`;
    logger.trace(`Memory [${context} - Delta] | Heap: ${formatDiff(heapDiff)}MB | RSS: ${formatDiff(rssDiff)}MB | Ext: ${formatDiff(externalDiff)}MB`);
}
export async function withMemoryLogging(context, fn) {
    const before = getMemoryStats();
    logMemoryUsage(`${context} - Before`);
    try {
        const result = await fn();
        const after = getMemoryStats();
        logMemoryUsage(`${context} - After`);
        logMemoryDifference(context, before, after);
        return result;
    }
    catch (error) {
        const after = getMemoryStats();
        logMemoryUsage(`${context} - After (Error)`);
        logMemoryDifference(context, before, after);
        throw error;
    }
}