Merge pull request #480 from iamqizhao/master
Fix the benchmark client
diff --git a/benchmark/client/main.go b/benchmark/client/main.go
index f9e2a83..e7f0a8f 100644
--- a/benchmark/client/main.go
+++ b/benchmark/client/main.go
@@ -92,13 +92,6 @@
func closeLoopStream() {
s, conn, tc := buildConnection()
- stream, err := tc.StreamingCall(context.Background())
- if err != nil {
- grpclog.Fatalf("%v.StreamingCall(_) = _, %v", tc, err)
- }
- for i := 0; i < 100; i++ {
- streamCaller(tc, stream)
- }
ch := make(chan int, *maxConcurrentRPCs*4)
var (
mu sync.Mutex
@@ -108,7 +101,15 @@
// Distribute RPCs over maxConcurrentCalls workers.
for i := 0; i < *maxConcurrentRPCs; i++ {
go func() {
- for _ = range ch {
+ stream, err := tc.StreamingCall(context.Background())
+ if err != nil {
+ grpclog.Fatalf("%v.StreamingCall(_) = _, %v", tc, err)
+ }
+ // Do some warm up.
+ for i := 0; i < 100; i++ {
+ streamCaller(tc, stream)
+ }
+ for range ch {
start := time.Now()
streamCaller(tc, stream)
elapse := time.Since(start)
