Skip to content

Commit

Permalink
add stream timeout
Browse files Browse the repository at this point in the history
  • Loading branch information
aarshkshah1992 committed May 15, 2024
1 parent 74d7274 commit fc40cda
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion network/libp2p_impl.go
Original file line number Diff line number Diff line change
Expand Up @@ -226,16 +226,18 @@ func (gsnet *libp2pGraphSyncNetwork) handleNewStream(s network.Stream) {
reader := msgio.NewVarintReaderSize(s, network.MessageSizeMax)
for {
p = s.Conn().RemotePeer()
_ = s.SetReadDeadline(time.Now().Add(time.Second * 10))
received, err := gsnet.messageHandlerSelector.Select(s.Protocol()).FromMsgReader(s.Conn().RemotePeer(), reader)

if err != nil {
if err != io.EOF {
_ = s.Reset()
go gsnet.receiver.ReceiveError(p, err)
log.Debugf("graphsync net handleNewStream from %s error: %s", s.Conn().RemotePeer(), err)
}
_ = s.SetReadDeadline(time.Time{})
return
}
_ = s.SetReadDeadline(time.Time{})

ctx := context.Background()
log.Debugf("graphsync net handleNewStream from %s", s.Conn().RemotePeer())
Expand Down

0 comments on commit fc40cda

Please sign in to comment.