TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-04 16:00:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化测试用例

Browse Source
This commit is contained in:
GoEdgeLab
2024-01-22 11:21:19 +08:00
parent 9337fdc18b
commit fa2205bb26
1 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
internal/utils/byte_pool_test.go
View File

@@ -1,6 +1,8 @@
package utils package utils_test
import ( import (
"bytes"
"github.com/TeaOSLab/EdgeNode/internal/utils"
"runtime" "runtime"
"sync" "sync"
"testing" "testing"
@@ -10,7 +12,7 @@ func TestBytePool_Memory(t *testing.T) {
var stat1 = &runtime.MemStats{} var stat1 = &runtime.MemStats{}
runtime.ReadMemStats(stat1) runtime.ReadMemStats(stat1)
var pool = NewBytePool(32 * 1024) var pool = utils.NewBytePool(32 * 1024)
for i := 0; i < 20480; i++ { for i := 0; i < 20480; i++ {
pool.Put(make([]byte, 32*1024)) pool.Put(make([]byte, 32*1024))
} }
@@ -29,7 +31,7 @@ func TestBytePool_Memory(t *testing.T) {
func BenchmarkBytePool_Get(b *testing.B) { func BenchmarkBytePool_Get(b *testing.B) {
runtime.GOMAXPROCS(1) runtime.GOMAXPROCS(1)
var pool = NewBytePool(1) var pool = utils.NewBytePool(1)
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
@@ -42,7 +44,7 @@ func BenchmarkBytePool_Get(b *testing.B) {
func BenchmarkBytePool_Get_Parallel(b *testing.B) { func BenchmarkBytePool_Get_Parallel(b *testing.B) {
runtime.GOMAXPROCS(1) runtime.GOMAXPROCS(1)
var pool = NewBytePool(1024) var pool = utils.NewBytePool(1024)
b.ResetTimer() b.ResetTimer()
b.RunParallel(func(pb *testing.PB) { b.RunParallel(func(pb *testing.PB) {
@@ -72,3 +74,20 @@ func BenchmarkBytePool_Get_Sync(b *testing.B) {
} }
}) })
} }
func BenchmarkBytePool_Copy(b *testing.B) {
var data = bytes.Repeat([]byte{'A'}, 8<<10)
var pool = &sync.Pool{
New: func() any {
return make([]byte, 8<<10)
},
}
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var buf = pool.Get().([]byte)
copy(buf, data)
pool.Put(buf)
}
})
}