Skip to content

Commit 15fd011

Browse files
authored
close empty streams (#9)
1 parent 754f452 commit 15fd011

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

network/libp2p_impl.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,7 @@ func (gsnet *libp2pGraphSyncNetwork) handleNewStream(s network.Stream) {
231231
reader := msgio.NewVarintReaderSize(s, network.MessageSizeMax)
232232
for {
233233
p = s.Conn().RemotePeer()
234+
_ = s.SetReadDeadline(time.Now().Add(time.Second * 10))
234235
received, err := gsnet.messageHandlerSelector.Select(s.Protocol()).FromMsgReader(s.Conn().RemotePeer(), reader)
235236

236237
if err != nil {
@@ -239,8 +240,10 @@ func (gsnet *libp2pGraphSyncNetwork) handleNewStream(s network.Stream) {
239240
go gsnet.receiver.ReceiveError(p, err)
240241
log.Debugf("graphsync net handleNewStream from %s error: %s", s.Conn().RemotePeer(), err)
241242
}
243+
_ = s.SetReadDeadline(time.Time{})
242244
return
243245
}
246+
_ = s.SetReadDeadline(time.Time{})
244247

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

0 commit comments

Comments
 (0)