测试内存消耗

Question

I need to verify how much memory a specific function consumes during execution, and make sure that it stays under a specific limit.

Ideally I'd like to do this in a test or benchmark. As far as I can see the only way to do this is to create a separate test binary and use the BenchmarkResult from

func Benchmark(f func(b *B)) BenchmarkResult

Is this the correct way to do this?

Answer 1

Actually it's very simple:

1. Read memstats with runtime.ReadMemStats(&m) into m
2. Invoke f()
3. Read memstats again into m2
4. Calculate diff between m and m2