benchmarks_test.go

package easytcp

import (
	"bytes"
	"github.com/stretchr/testify/assert"
	"io"
	"net"
	"testing"
)

// go test -bench="^Benchmark_\w+$" -run=none -benchmem -benchtime=250000x

type mutedLogger struct{}

func (m *mutedLogger) Errorf(_ string, _ ...interface{}) {}
func (m *mutedLogger) Tracef(_ string, _ ...interface{}) {}

func Benchmark_NoHandler(b *testing.B) {
	s := NewServer(&ServerOption{
		DoNotPrintRoutes: true,
	})
	go s.Run("127.0.0.1:0") // nolint
	defer s.Stop()          // nolint

	<-s.acceptingC

	// client
	client, err := net.Dial("tcp", s.Listener.Addr().String())
	if err != nil {
		panic(err)
	}
	defer client.Close() // nolint

	packedBytes, _ := s.Packer.Pack(NewMessage(1, []byte("ping")))
	beforeBench(b)
	for i := 0; i < b.N; i++ {
		_, _ = client.Write(packedBytes)
	}
}

func Benchmark_OneHandler(b *testing.B) {
	s := NewServer(&ServerOption{
		DoNotPrintRoutes: true,
	})
	s.AddRoute(1, func(ctx Context) {})
	go s.Run("127.0.0.1:0") // nolint
	defer s.Stop()          // nolint

	<-s.acceptingC

	// client
	client, err := net.Dial("tcp", s.Listener.Addr().String())
	if err != nil {
		panic(err)
	}
	defer client.Close() // nolint

	packedMsg, _ := s.Packer.Pack(NewMessage(1, []byte("ping")))
	beforeBench(b)
	for i := 0; i < b.N; i++ {
		_, _ = client.Write(packedMsg)
	}
}

func Benchmark_DefaultPacker_Pack(b *testing.B) {
	packer := NewDefaultPacker()
	msg := NewMessage(1, []byte("test"))
	beforeBench(b)
	for i := 0; i < b.N; i++ {
		_, _ = packer.Pack(msg)
	}
}

func Benchmark_DefaultPacker_Unpack(b *testing.B) {
	packer := NewDefaultPacker()
	msg := NewMessage(1, []byte("test"))
	dataBytes, err := packer.Pack(msg)
	assert.NoError(b, err)

	r := bytes.NewReader(dataBytes)
	beforeBench(b)
	for i := 0; i < b.N; i++ {
		_, _ = packer.Unpack(r)
		_, _ = r.Seek(0, io.SeekStart)
	}
}

func beforeBench(b *testing.B) {
	_log = &mutedLogger{}
	b.ReportAllocs()
	b.ResetTimer()
}