WitNote / internal /bridge /tab_executor_benchmark_test.go
AUXteam's picture
Upload folder using huggingface_hub
6a7089a verified
package bridge
import (
"context"
"fmt"
"sync"
"testing"
"time"
)
func BenchmarkTabExecutor_SequentialSameTab(b *testing.B) {
te := NewTabExecutor(4)
ctx := context.Background()
b.ResetTimer()
for i := 0; i < b.N; i++ {
_ = te.Execute(ctx, "tab1", func(ctx context.Context) error {
return nil
})
}
}
func BenchmarkTabExecutor_ParallelDifferentTabs(b *testing.B) {
te := NewTabExecutor(8)
ctx := context.Background()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
i := 0
for pb.Next() {
tabID := fmt.Sprintf("tab%d", i%8)
_ = te.Execute(ctx, tabID, func(ctx context.Context) error {
return nil
})
i++
}
})
}
func BenchmarkTabExecutor_ParallelSameTab(b *testing.B) {
te := NewTabExecutor(8)
ctx := context.Background()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_ = te.Execute(ctx, "tab1", func(ctx context.Context) error {
return nil
})
}
})
}
func BenchmarkTabExecutor_WithWork(b *testing.B) {
te := NewTabExecutor(4)
ctx := context.Background()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
i := 0
for pb.Next() {
tabID := fmt.Sprintf("tab%d", i%4)
_ = te.Execute(ctx, tabID, func(ctx context.Context) error {
// Simulate light work
sum := 0
for j := 0; j < 100; j++ {
sum += j
}
_ = sum
return nil
})
i++
}
})
}
func BenchmarkSequentialVsParallel(b *testing.B) {
workDuration := time.Microsecond * 100
b.Run("Sequential_4Tabs", func(b *testing.B) {
te := NewTabExecutor(1)
ctx := context.Background()
b.ResetTimer()
for i := 0; i < b.N; i++ {
for j := 0; j < 4; j++ {
_ = te.Execute(ctx, fmt.Sprintf("tab%d", j), func(ctx context.Context) error {
time.Sleep(workDuration)
return nil
})
}
}
})
b.Run("Parallel_4Tabs", func(b *testing.B) {
te := NewTabExecutor(4)
ctx := context.Background()
b.ResetTimer()
for i := 0; i < b.N; i++ {
var wg sync.WaitGroup
for j := 0; j < 4; j++ {
wg.Add(1)
tabID := fmt.Sprintf("tab%d", j)
go func() {
defer wg.Done()
_ = te.Execute(ctx, tabID, func(ctx context.Context) error {
time.Sleep(workDuration)
return nil
})
}()
}
wg.Wait()
}
})
}