From 4cebb4d876c88ed0b22f45578be7824ba349ec6b Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Thu, 5 Oct 2023 09:15:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E7=94=A8=E5=86=85=E5=AD=98?= =?UTF-8?q?=E7=BB=9F=E8=AE=A1=E5=A2=9E=E5=8A=A0=E5=9B=9E=E8=B0=83=E5=87=BD?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/utils/testutils/memory.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/internal/utils/testutils/memory.go b/internal/utils/testutils/memory.go index 49a50c6..c981bb4 100644 --- a/internal/utils/testutils/memory.go +++ b/internal/utils/testutils/memory.go @@ -32,7 +32,7 @@ func StartMemoryStatsGC(t *testing.T) { }() } -func StartMemoryStats(t *testing.T) { +func StartMemoryStats(t *testing.T, callbacks ...func()) { var ticker = time.NewTicker(1 * time.Second) go func() { var stat = &runtime.MemStats{} @@ -46,6 +46,12 @@ func StartMemoryStats(t *testing.T) { lastHeapInUse = stat.HeapInuse t.Log(timeutil.Format("H:i:s"), "HeapInuse:", fmt.Sprintf("%.2fM", float64(stat.HeapInuse)/1024/1024), "NumGC:", stat.NumGC) + + if len(callbacks) > 0 { + for _, callback := range callbacks { + callback() + } + } } }() }