mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-04 16:00:25 +08:00
42 lines
628 B
Go
42 lines
628 B
Go
package utils
|
|
|
|
import (
|
|
"github.com/iwind/TeaGo/assert"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewBytePool(t *testing.T) {
|
|
a := assert.NewAssertion(t)
|
|
|
|
pool := NewBytePool(5, 8)
|
|
buf := pool.Get()
|
|
a.IsTrue(len(buf) == 8)
|
|
a.IsTrue(len(pool.c) == 0)
|
|
|
|
pool.Put(buf)
|
|
a.IsTrue(len(pool.c) == 1)
|
|
|
|
pool.Get()
|
|
a.IsTrue(len(pool.c) == 0)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
pool.Put(buf)
|
|
}
|
|
t.Log(len(pool.c))
|
|
a.IsTrue(len(pool.c) == 5)
|
|
}
|
|
|
|
func BenchmarkBytePool_Get(b *testing.B) {
|
|
runtime.GOMAXPROCS(1)
|
|
|
|
pool := NewBytePool(1024, 1)
|
|
for i := 0; i < b.N; i++ {
|
|
buf := pool.Get()
|
|
_ = buf
|
|
pool.Put(buf)
|
|
}
|
|
|
|
b.Log(pool.Size())
|
|
}
|