Unverified Commit 7d6f9521 authored by Steven Allen's avatar Steven Allen Committed by GitHub
Browse files

Merge pull request #380 from libp2p/fix/reset-ping-stream

always reset ping streams when done
parents 6f0dcfa1 72df25af
......@@ -44,19 +44,14 @@ func (p *PingService) PingHandler(s inet.Stream) {
select {
case <-timer.C:
log.Debug("ping timeout")
s.Reset()
case err, ok := <-errCh:
if ok {
log.Debug(err)
if err == io.EOF {
s.Close()
} else {
s.Reset()
}
} else {
log.Error("ping loop failed without error")
}
}
s.Reset()
}()
for {
......@@ -85,7 +80,7 @@ func (ps *PingService) Ping(ctx context.Context, p peer.ID) (<-chan time.Duratio
out := make(chan time.Duration)
go func() {
defer close(out)
defer s.Close()
defer s.Reset()
for {
select {
case <-ctx.Done():
......@@ -93,7 +88,6 @@ func (ps *PingService) Ping(ctx context.Context, p peer.ID) (<-chan time.Duratio
default:
t, err := ping(s)
if err != nil {
s.Reset()
log.Debugf("ping error: %s", err)
return
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment