如何从Rust中记录WASM堆内存使用以检测内存泄漏？

核心思路 #

《Rust and WebAssembly》一书中推荐利用WASM分配器的内存统计API直接获取已使用的堆内存大小，这是最精准且低成本的方案，无需额外工具或复杂插桩操作。

具体实现步骤 #

1. 在Rust侧暴露内存统计函数
   书中主推轻量分配器wee_alloc，它内置了内存使用统计方法。先在Cargo.toml添加依赖：

   [dependencies]
   wasm-bindgen = "0.2"
   wee_alloc = "0.4"
   

   接着在Rust代码中配置全局分配器并导出统计函数：

   use wasm_bindgen::prelude::*;
   use wee_alloc::WeeAlloc;
   
   // 设置全局分配器为wee_alloc
   #[global_allocator]
   static ALLOC: WeeAlloc = WeeAlloc::INIT;
   
   // 导出给JS调用的函数，返回当前已分配堆内存大小（字节）
   #[wasm_bindgen]
   pub fn get_used_heap() -> usize {
       ALLOC.size_of_current_allocations()
   }
   

   若使用Rust默认的dlmalloc分配器，需手动实现统计逻辑，但wee_alloc的方案更简洁，完全契合书中建议。

2. 在JavaScript侧定期记录数据
   编译WASM后，在JS中定时调用导出的函数，记录或监控内存变化：

   // 假设wasmModule是加载完成的WASM实例
   const monitorInterval = setInterval(() => {
       const usedBytes = wasmModule.get_used_heap();
       console.log(`WASM已用堆内存: ${usedBytes} bytes`);
       // 可选：将数据发送到监控系统或写入日志
   }, 2000); // 每2秒记录一次，可按需调整间隔
   
   // 停止监控时调用
   // clearInterval(monitorInterval);
   

关键提示 #

• 不要直接读取WASM的memory.buffer.byteLength，这个值是整个线性内存的总大小，包含代码段、静态数据等非堆内存部分，无法准确反映堆内存使用情况。
• 若发现内存持续增长且无回落，大概率存在泄漏，可结合Rust的Rc/Arc引用计数检查、或JS侧console.trace()追踪分配来源进一步定位。

内容的提问来源于stack exchange，提问作者prideout