cerc-io/lotus
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

wrap close(wait) with sync.Once to avoid panic

Browse Source
This commit is contained in:
Anton Evangelatov 2021-07-20 11:49:59 +02:00 committed by Steven Allen
parent 8ddd12d144
commit cd4571f0b0
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/rpcenc/reader.go
View File

@ -81,18 +81,23 @@ func ReaderParamEncoder(addr string) jsonrpc.Option {
type waitReadCloser struct { type waitReadCloser struct {
io.ReadCloser io.ReadCloser
wait chan struct{} wait chan struct{}
closeOnce sync.Once
} }
func (w *waitReadCloser) Read(p []byte) (int, error) { func (w *waitReadCloser) Read(p []byte) (int, error) {
n, err := w.ReadCloser.Read(p) n, err := w.ReadCloser.Read(p)
if err != nil { if err != nil {
w.closeOnce.Do(func() {
close(w.wait) close(w.wait)
})
} }
return n, err return n, err
} }
func (w *waitReadCloser) Close() error { func (w *waitReadCloser) Close() error {
w.closeOnce.Do(func() {
close(w.wait) close(w.wait)
})
return w.ReadCloser.Close() return w.ReadCloser.Close()
} }