Function BenchmarkExtensionReadWrite

msgp/extension_test.go:155–205

(b *testing.B)

}

func BenchmarkExtensionReadWrite(b *testing.B) {
	var buf bytes.Buffer
	en := NewWriter(&buf)
	dc := NewReader(&buf)

	b.Run("interface", func(b *testing.B) {
		b.ReportAllocs()

		for i := 0; i < b.N; i++ {
			buf.Reset()

			e := randomExt()
			err := en.WriteExtension(&e)
			if err != nil {
				b.Errorf("error writing extension: %s", err)
			}
			en.Flush()

			err = dc.ReadExtension(&e)
			if err != nil {
				b.Errorf("error reading extension: %s", err)
			}
		}
	})

	b.Run("raw", func(b *testing.B) {
		// this should have zero allocations
		b.ReportAllocs()

		for i := 0; i < b.N; i++ {
			buf.Reset()

			e := randomExt()
			err := en.WriteExtensionRaw(e.Type, e.Data)
			if err != nil {
				b.Errorf("error writing extension: %s", err)
			}
			en.Flush()

			typ, payload, err := dc.ReadExtensionRaw()
			if err != nil {
				b.Errorf("error reading extension: %s", err)
			}
			if typ != e.Type || !bytes.Equal(payload, e.Data) {
				b.Errorf("extension mismatch: %d %x != %d %x", typ, payload, e.Type, e.Data)
			}

			buf.Reset()
		}
	})
}