TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-12-09 19:30:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

测试用内存统计增加回调函数

Browse Source
This commit is contained in:
GoEdgeLab
2023-10-05 09:15:17 +08:00
parent 7fb050fc80
commit 4cebb4d876
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/utils/testutils/memory.go
View File

@@ -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) var ticker = time.NewTicker(1 * time.Second)
go func() { go func() {
var stat = &runtime.MemStats{} var stat = &runtime.MemStats{}
@@ -46,6 +46,12 @@ func StartMemoryStats(t *testing.T) {
lastHeapInUse = stat.HeapInuse lastHeapInUse = stat.HeapInuse
t.Log(timeutil.Format("H:i:s"), "HeapInuse:", fmt.Sprintf("%.2fM", float64(stat.HeapInuse)/1024/1024), "NumGC:", stat.NumGC) 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()
}
}
} }
}() }()
} }