package operation import ( "context" "fmt" "github.com/seaweedfs/seaweedfs/weed/pb" "google.golang.org/grpc" "testing" "time" ) func BenchmarkWithConcurrency(b *testing.B) { concurrencyLevels := []int{1, 10, 100, 1000} ap, _ := NewAssignProxy(func(_ context.Context) pb.ServerAddress { return pb.ServerAddress("localhost:9333") }, grpc.WithInsecure(), 16) for _, concurrency := range concurrencyLevels { b.Run( fmt.Sprintf("Concurrency-%d", concurrency), func(b *testing.B) { for i := 0; i < b.N; i++ { done := make(chan struct{}) startTime := time.Now() for j := 0; j < concurrency; j++ { go func() { ap.Assign(&VolumeAssignRequest{ Count: 1, }) done <- struct{}{} }() } for j := 0; j < concurrency; j++ { <-done } duration := time.Since(startTime) b.Logf("Concurrency: %d, Duration: %v", concurrency, duration) } }, ) } } func BenchmarkStreamAssign(b *testing.B) { ap, _ := NewAssignProxy(func(_ context.Context) pb.ServerAddress { return pb.ServerAddress("localhost:9333") }, grpc.WithInsecure(), 16) for i := 0; i < b.N; i++ { ap.Assign(&VolumeAssignRequest{ Count: 1, }) } } func BenchmarkUnaryAssign(b *testing.B) { for i := 0; i < b.N; i++ { Assign(func(_ context.Context) pb.ServerAddress { return pb.ServerAddress("localhost:9333") }, grpc.WithInsecure(), &VolumeAssignRequest{ Count: 1, }) } }