opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix leak goroutine problem

Browse Source
This commit is contained in:
Menghan Li
2016-04-29 16:43:01 -07:00
parent e7801b635a
commit 02bd4c80b5
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
benchmark/worker/benchmark_client.go
View File

@ -202,7 +202,10 @@ func (bc *benchmarkClient) doCloseLoopUnary(conns []*grpc.ClientConn, rpcCountPe
go func() {
start := time.Now()
if err := benchmark.DoUnaryCall(client, reqSize, respSize); err != nil {
done <- false
select {
case <-bc.stop:
case done <- false:
}
return
}
elapse := time.Since(start)
@ -259,7 +262,10 @@ func (bc *benchmarkClient) doCloseLoopStreaming(conns []*grpc.ClientConn, rpcCou
go func() {
start := time.Now()
if err := doRPC(stream, reqSize, respSize); err != nil {
done <- false
select {
case <-bc.stop:
case done <- false:
}
return
}
elapse := time.Since(start)