added paralell benchmkarks for sequence window

1 files changed, 52 insertions, 0 deletions

diff --git a/satp/sequence-window_test.go b/satp/sequence-window_test.go
index 3678d85..d52dead 100644
--- a/satp/sequence-window_test.go
+++ b/satp/sequence-window_test.go
@@ -31,6 +31,8 @@
 package satp
 
 import (
+	"runtime"
+	"sync"
 	"testing"
 )
 
@@ -239,6 +241,31 @@ func BenchmarkSeqWindowCheck(b *testing.B) {
 	}
 }
 
+func BenchmarkSeqWindowCheckParallel(b *testing.B) {
+	w, err := NewSequenceWindow(100, 0)
+	if err != nil {
+		b.Fatal("unexpected error:", err)
+	}
+
+	wg := sync.WaitGroup{}
+	numThreads := runtime.NumCPU()
+	b.Logf("running test with %d threads", numThreads)
+	wg.Add(numThreads)
+
+	b.ResetTimer()
+
+	for i := 0; i < numThreads; i++ {
+		go func() {
+			for seq := uint32(0); seq < uint32(b.N); seq += uint32(numThreads) {
+				w.Check(seq)
+			}
+			wg.Done()
+		}()
+	}
+
+	wg.Wait()
+}
+
 func BenchmarkSeqWindowCheckAndSet(b *testing.B) {
 	w, err := NewSequenceWindow(100, 0)
 	if err != nil {
@@ -251,3 +278,28 @@ func BenchmarkSeqWindowCheckAndSet(b *testing.B) {
 		w.CheckAndSet(seq)
 	}
 }
+
+func BenchmarkSeqWindowCheckAndSetParallel(b *testing.B) {
+	w, err := NewSequenceWindow(100, 0)
+	if err != nil {
+		b.Fatal("unexpected error:", err)
+	}
+
+	wg := sync.WaitGroup{}
+	numThreads := runtime.NumCPU()
+	b.Logf("running test with %d threads", numThreads)
+	wg.Add(numThreads)
+
+	b.ResetTimer()
+
+	for i := 0; i < numThreads; i++ {
+		go func() {
+			for seq := uint32(0); seq < uint32(b.N); seq += uint32(numThreads) {
+				w.CheckAndSet(seq)
+			}
+			wg.Done()
+		}()
+	}
+
+	wg.Wait()
+}