status-im
/
go-libp2p-pubsub
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

defer stream removal instead of doing it inline.

This commit is contained in:
vyzo 2021-01-12 16:26:38 +02:00
parent b080f595bb
commit 9e129db766
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comm.go
View File

@ -52,6 +52,14 @@ func (p *PubSub) handleNewStream(s network.Stream) {
p.inboundStreams[peer] = s p.inboundStreams[peer] = s
p.inboundStreamsMx.Unlock() p.inboundStreamsMx.Unlock()
defer func() {
p.inboundStreamsMx.Lock()
if p.inboundStreams[peer] == s {
delete(p.inboundStreams, peer)
}
p.inboundStreamsMx.Unlock()
}()
r := protoio.NewDelimitedReader(s, p.maxMessageSize) r := protoio.NewDelimitedReader(s, p.maxMessageSize)
for { for {
rpc := new(RPC) rpc := new(RPC)
@ -66,12 +74,6 @@ func (p *PubSub) handleNewStream(s network.Stream) {
s.Close() s.Close()
} }
p.inboundStreamsMx.Lock()
if p.inboundStreams[peer] == s {
delete(p.inboundStreams, peer)
}
p.inboundStreamsMx.Unlock()
return return
} }