Java单元测试：如何测量方法调用的内存占用

Question

假设我有一个可以进行大量处理的类，并使用多个集合进行操作。我要做的是确保这种操作不会导致内存不足，甚至更好，我想设置一个阈值来确定它可以使用多少内存。

class MyClass()
{
   public void myMethod()
   {
      for(int i=0; i<10000000; i++)
      {
         // Allocate some memory, may be several collections
      }
   }
}

class MyClassTest
{
   @Test
   public void myMethod_makeSureMemoryFootprintIsNotBiggerThanMax()
   {
      new MyClass().myMethod(); 
      // How do I measure amount of memory it may try to allocate?
   }
}

什么是正确的方法？还是这不可能/不可行？

Answer 1

我可以想到几种选择：

• 通过微基准测试（即jmh）找出您的方法需要多少内存。
• 基于启发式估计的建筑分配策略。有几种实现类大小估计的开源解决方案，即ClassSize。一种更简单的方法可能是利用释放很少使用的对象的缓存（例如Guava的缓存）。如@EnnoShioji所述，Guava的缓存具有基于内存的驱逐策略。

您也可以编写自己的基准测试来计算内存。这个想法是为了

1. 运行一个线程。
2. 创建一个新的数组来存储要分配的对象。因此，在GC运行期间不会收集这些对象。
3. System.gc()， memoryBefore = runtime.totalMemory() - runtime.freeMemory()
4. 分配您的对象。将它们放入数组。
5. System.gc()， memoryAfter = runtime.totalMemory() - runtime.freeMemory()

这是我在轻量级微基准测试工具中使用的一种技术，该工具能够以字节精度测量内存分配。