Add BlockWriter bench (#114)

mkvcore/blockwriter_test.go

@@ -515,3 +515,50 @@ func TestBlockWriter_WithSeekHead(t *testing.T) {
 		}
 	})
 }
+
+func BenchmarkBlockWriter_InitFinalize(b *testing.B) {
+	tracks := []TrackDescription{
+		{TrackNumber: 1},
+	}
+
+	for i := 0; i < b.N; i++ {
+		buf := buffercloser.New()
+		ws, err := NewSimpleBlockWriter(buf, tracks,
+			WithBlockInterceptor(NewMultiTrackBlockSorter(10, BlockSorterDropOutdated)),
+		)
+		if err != nil {
+			b.Fatalf("Failed to create BlockWriter: %v", err)
+		}
+		for _, w := range ws {
+			w.Close()
+		}
+	}
+}
+
+func BenchmarkBlockWriter_SimpleBlock(b *testing.B) {
+	tracks := []TrackDescription{
+		{TrackNumber: 1},
+	}
+
+	buf := buffercloser.New()
+	ws, err := NewSimpleBlockWriter(buf, tracks,
+		WithBlockInterceptor(NewMultiTrackBlockSorter(10, BlockSorterDropOutdated)),
+	)
+	if err != nil {
+		b.Fatalf("Failed to create BlockWriter: %v", err)
+	}
+
+	data := []byte{}
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		for _, w := range ws {
+			if _, err := w.Write(true, int64(i*20), data); err != nil {
+				b.Fatalf("Failed to Write: %v", err)
+			}
+		}
+	}
+	b.StopTimer()
+	for _, w := range ws {
+		w.Close()
+	}
+}