diff --git a/internal/utils/byte_pool_test.go b/internal/utils/byte_pool_test.go index 04c1faf..ed13e7d 100644 --- a/internal/utils/byte_pool_test.go +++ b/internal/utils/byte_pool_test.go @@ -1,6 +1,8 @@ -package utils +package utils_test import ( + "bytes" + "github.com/TeaOSLab/EdgeNode/internal/utils" "runtime" "sync" "testing" @@ -10,7 +12,7 @@ func TestBytePool_Memory(t *testing.T) { var stat1 = &runtime.MemStats{} runtime.ReadMemStats(stat1) - var pool = NewBytePool(32 * 1024) + var pool = utils.NewBytePool(32 * 1024) for i := 0; i < 20480; i++ { pool.Put(make([]byte, 32*1024)) } @@ -29,7 +31,7 @@ func TestBytePool_Memory(t *testing.T) { func BenchmarkBytePool_Get(b *testing.B) { runtime.GOMAXPROCS(1) - var pool = NewBytePool(1) + var pool = utils.NewBytePool(1) b.ResetTimer() for i := 0; i < b.N; i++ { @@ -42,7 +44,7 @@ func BenchmarkBytePool_Get(b *testing.B) { func BenchmarkBytePool_Get_Parallel(b *testing.B) { runtime.GOMAXPROCS(1) - var pool = NewBytePool(1024) + var pool = utils.NewBytePool(1024) b.ResetTimer() 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) + } + }) +}