UNPKG

next

Version:

The React Framework

82 lines (81 loc) 3.57 kB
import v8 from 'v8'; import { info, warn } from '../../build/output/log'; import { trace } from '../../trace'; import { bold, italic } from '../picocolors'; import { join } from 'path'; import { traceGlobals } from '../../trace/shared'; const HEAP_SNAPSHOT_THRESHOLD_PERCENT = 70; let alreadyGeneratedHeapSnapshot = false; const TRACE_MEMORY_USAGE_TIMER_MS = 20000; let traceMemoryUsageTimer; const allMemoryUsage = []; /** * Begins a timer that will record memory usage periodically to understand * memory usage across the lifetime of the process. */ export function startPeriodicMemoryUsageTracing() { traceMemoryUsageTimer = setTimeout(()=>{ traceMemoryUsage('periodic memory snapshot'); startPeriodicMemoryUsageTracing(); }, TRACE_MEMORY_USAGE_TIMER_MS); } export function stopPeriodicMemoryUsageTracing() { if (traceMemoryUsageTimer) { clearTimeout(traceMemoryUsageTimer); } } /** * Returns the list of all recorded memory usage snapshots from the process. */ export function getAllMemoryUsageSpans() { return allMemoryUsage; } /** * Records a snapshot of memory usage at this moment in time to the .next/trace * file. */ export function traceMemoryUsage(description, parentSpan) { const memoryUsage = process.memoryUsage(); const v8HeapStatistics = v8.getHeapStatistics(); const heapUsed = v8HeapStatistics.used_heap_size; const heapMax = v8HeapStatistics.heap_size_limit; const tracedMemoryUsage = { 'memory.rss': memoryUsage.rss, 'memory.heapUsed': heapUsed, 'memory.heapTotal': memoryUsage.heapTotal, 'memory.heapMax': heapMax }; allMemoryUsage.push(tracedMemoryUsage); const tracedMemoryUsageAsStrings = Object.fromEntries(Object.entries(tracedMemoryUsage).map(([key, value])=>[ key, String(value) ])); if (parentSpan) { parentSpan.traceChild('memory-usage', tracedMemoryUsageAsStrings); } else { trace('memory-usage', undefined, tracedMemoryUsageAsStrings); } if (process.env.EXPERIMENTAL_DEBUG_MEMORY_USAGE) { const percentageHeapUsed = 100 * heapUsed / heapMax; info(''); info('***************************************'); info(`Memory usage report at "${description}":`); info(` - RSS: ${(memoryUsage.rss / 1024 / 1024).toFixed(2)} MB`); info(` - Heap Used: ${(heapUsed / 1024 / 1024).toFixed(2)} MB`); info(` - Heap Total Allocated: ${(memoryUsage.heapTotal / 1024 / 1024).toFixed(2)} MB`); info(` - Heap Max: ${(heapMax / 1024 / 1024).toFixed(2)} MB`); info(` - Percentage Heap Used: ${percentageHeapUsed.toFixed(2)}%`); info('***************************************'); info(''); if (percentageHeapUsed > HEAP_SNAPSHOT_THRESHOLD_PERCENT) { const distDir = traceGlobals.get('distDir'); const heapFilename = join(distDir, `${description.replace(' ', '-')}.heapsnapshot`); warn(bold(`Heap usage is close to the limit. ${percentageHeapUsed.toFixed(2)}% of heap has been used.`)); if (!alreadyGeneratedHeapSnapshot) { warn(bold(`Saving heap snapshot to ${heapFilename}. ${italic('Note: this will take some time.')}`)); v8.writeHeapSnapshot(heapFilename); alreadyGeneratedHeapSnapshot = true; } else { warn('Skipping heap snapshot generation since heap snapshot has already been generated.'); } } } } //# sourceMappingURL=trace.js.map