Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

agent: fix a data race and deadlock

Browse Source 
add 1-size buffer for `errc`  to avoid deadlock of child goroutine
add a local variable to a void data race in `err`
when `case <-stream.Context().Done():` is taken
This commit is contained in:
Zeming YU 2019-08-06 14:18:23 -07:00 committed by Gyuho Lee
parent 89e102365d
commit 629cb7aa5e
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
functional/agent/server.go
View File

@ -126,11 +126,12 @@ func (srv *Server) Stop() {
}
// Transport communicates with etcd tester.
func (srv *Server) Transport(stream rpcpb.Transport_TransportServer) (err error) {
errc := make(chan error)
func (srv *Server) Transport(stream rpcpb.Transport_TransportServer) (reterr error) {
errc := make(chan error, 1)
go func() {
for {
var req *rpcpb.Request
var err error
req, err = stream.Recv()
if err != nil {
errc <- err
@ -161,9 +162,9 @@ func (srv *Server) Transport(stream rpcpb.Transport_TransportServer) (err error)
}()
select {
case err = <-errc:
case reterr = <-errc:
case <-stream.Context().Done():
err = stream.Context().Err()
reterr = stream.Context().Err()
}
return err
return reterr
}