add head memory to perf logging

facebook · Oct 17, 2024 · eebcb64 · eebcb64
1 parent 2471678
commit eebcb64
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 6 deletions.
diff --git a/packages/docusaurus-logger/src/logger.ts b/packages/docusaurus-logger/src/logger.ts
@@ -178,6 +178,7 @@ const logger = {
   red: (msg: string | number): string => chalk.red(msg),
   yellow: (msg: string | number): string => chalk.yellow(msg),
   green: (msg: string | number): string => chalk.green(msg),
+  cyan: (msg: string | number): string => chalk.cyan(msg),
   bold: (msg: string | number): string => chalk.bold(msg),
   dim: (msg: string | number): string => chalk.dim(msg),
   path,

diff --git a/packages/docusaurus-logger/src/perfLogger.ts b/packages/docusaurus-logger/src/perfLogger.ts
@@ -37,6 +37,11 @@ type PerfLoggerAPI = {
   ) => Promise<Result>;
 };
 
+type Memory = {
+  before: NodeJS.MemoryUsage;
+  after: NodeJS.MemoryUsage;
+};
+
 function createPerfLogger(): PerfLoggerAPI {
   if (!PerfDebuggingEnabled) {
     const noop = () => {};
@@ -58,19 +63,56 @@ function createPerfLogger(): PerfLoggerAPI {
     }
   };
 
-  const logDuration = (label: string, duration: number) => {
+  const formatMemory = (memory: Memory): string => {
+    const fmtHead = (bytes: number) =>
+      logger.cyan(`${(bytes / 1000000).toFixed(0)}mb`);
+    return logger.dim(
+      `(${fmtHead(memory.before.heapUsed)} -> ${fmtHead(
+        memory.after.heapUsed,
+      )})`,
+    );
+  };
+
+  const printPerfLog = ({
+    label,
+    duration,
+    memory,
+  }: {
+    label: string;
+    duration: number;
+    memory: Memory;
+  }) => {
     if (duration < Thresholds.min) {
       return;
     }
-    console.log(`${PerfPrefix + label} - ${formatDuration(duration)}`);
+    console.log(
+      `${PerfPrefix + label} - ${formatDuration(duration)} - ${formatMemory(
+        memory,
+      )}`,
+    );
   };
 
-  const start: PerfLoggerAPI['start'] = (label) => performance.mark(label);
+  const start: PerfLoggerAPI['start'] = (label) =>
+    performance.mark(label, {
+      detail: {
+        memoryUsage: process.memoryUsage(),
+      },
+    });
 
   const end: PerfLoggerAPI['end'] = (label) => {
-    const {duration} = performance.measure(label);
+    const {
+      duration,
+      detail: {memoryUsage},
+    } = performance.measure(label);
     performance.clearMarks(label);
-    logDuration(applyParentPrefix(label), duration);
+    printPerfLog({
+      label: applyParentPrefix(label),
+      duration,
+      memory: {
+        before: memoryUsage,
+        after: process.memoryUsage(),
+      },
+    });
   };
 
   const log: PerfLoggerAPI['log'] = (label: string) =>
@@ -79,9 +121,18 @@ function createPerfLogger(): PerfLoggerAPI {
   const async: PerfLoggerAPI['async'] = async (label, asyncFn) => {
     const finalLabel = applyParentPrefix(label);
     const before = performance.now();
+    const memoryBefore = process.memoryUsage();
     const result = await ParentPrefix.run(finalLabel, () => asyncFn());
+    const memoryAfter = process.memoryUsage();
     const duration = performance.now() - before;
-    logDuration(finalLabel, duration);
+    printPerfLog({
+      label: finalLabel,
+      duration,
+      memory: {
+        before: memoryBefore,
+        after: memoryAfter,
+      },
+    });
     return result;
   };