Fix client RPC stream close mutex (#1643)

This commit is contained in:
Maarten Bezemer 2020-05-18 18:22:33 +02:00 committed by GitHub
parent 56a7897c91
commit 3d36398818
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -130,15 +130,15 @@ func (r *rpcStream) Error() error {
} }
func (r *rpcStream) Close() error { func (r *rpcStream) Close() error {
r.RLock() r.Lock()
select { select {
case <-r.closed: case <-r.closed:
r.RUnlock() r.Unlock()
return nil return nil
default: default:
close(r.closed) close(r.closed)
r.RUnlock() r.Unlock()
// send the end of stream message // send the end of stream message
if r.sendEOS { if r.sendEOS {