mirror of
https://github.com/mochi-mqtt/server.git
synced 2025-09-26 20:21:12 +08:00
c6c7c296f6
* Dynamically allocate buffer for writes if needed * Remove unused net.Buffer * Return bytes written to buffer instead of conn * Dynamic write buffer * Reduce double write of pk.Payload * Use memory pool for packet encode * Pool doesn't guarantee value between Put and Get * Add benchmark for bufpool * Fix issue #346 * Change default pool not to have size cap --------- Co-authored-by: JB <28275108+mochi-co@users.noreply.github.com>
97 lines
1.7 KiB
Go
97 lines
1.7 KiB
Go
package mempool
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"runtime/debug"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewBuffer(t *testing.T) {
|
|
defer debug.SetGCPercent(debug.SetGCPercent(-1))
|
|
bp := NewBuffer(1000)
|
|
require.Equal(t, "*mempool.BufferWithCap", reflect.TypeOf(bp).String())
|
|
|
|
bp = NewBuffer(0)
|
|
require.Equal(t, "*mempool.Buffer", reflect.TypeOf(bp).String())
|
|
|
|
bp = NewBuffer(-1)
|
|
require.Equal(t, "*mempool.Buffer", reflect.TypeOf(bp).String())
|
|
}
|
|
|
|
func TestBuffer(t *testing.T) {
|
|
defer debug.SetGCPercent(debug.SetGCPercent(-1))
|
|
Size := 101
|
|
|
|
bp := NewBuffer(0)
|
|
buf := bp.Get()
|
|
|
|
for i := 0; i < Size; i++ {
|
|
buf.WriteByte('a')
|
|
}
|
|
|
|
bp.Put(buf)
|
|
buf = bp.Get()
|
|
require.Equal(t, 0, buf.Len())
|
|
}
|
|
|
|
func TestBufferWithCap(t *testing.T) {
|
|
defer debug.SetGCPercent(debug.SetGCPercent(-1))
|
|
Size := 101
|
|
bp := NewBuffer(100)
|
|
buf := bp.Get()
|
|
|
|
for i := 0; i < Size; i++ {
|
|
buf.WriteByte('a')
|
|
}
|
|
|
|
bp.Put(buf)
|
|
buf = bp.Get()
|
|
require.Equal(t, 0, buf.Len())
|
|
require.Equal(t, 0, buf.Cap())
|
|
}
|
|
|
|
func BenchmarkBufferPool(b *testing.B) {
|
|
bp := NewBuffer(0)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b := bp.Get()
|
|
b.WriteString("this is a test")
|
|
bp.Put(b)
|
|
}
|
|
}
|
|
|
|
func BenchmarkBufferPoolWithCapLarger(b *testing.B) {
|
|
bp := NewBuffer(64 * 1024)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b := bp.Get()
|
|
b.WriteString("this is a test")
|
|
bp.Put(b)
|
|
}
|
|
}
|
|
|
|
func BenchmarkBufferPoolWithCapLesser(b *testing.B) {
|
|
bp := NewBuffer(10)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b := bp.Get()
|
|
b.WriteString("this is a test")
|
|
bp.Put(b)
|
|
}
|
|
}
|
|
|
|
func BenchmarkBufferWithoutPool(b *testing.B) {
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b := new(bytes.Buffer)
|
|
b.WriteString("this is a test")
|
|
_ = b
|
|
}
|
|
}
|