libp2p/go-libp2p-resource-manager
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-resource-manager.git synced 2026-08-22 19:43:28 +08:00
Code Issues Projects Releases Wiki Activity

make connection/stream Done idempotent

Browse Source
This commit is contained in:
vyzo
2021-12-23 19:45:05 +02:00
parent 0b5b506b8c
commit a42700a439
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
rcmgr.go
View File

@@ -81,6 +81,8 @@ type ConnectionScope struct {
system *SystemScope
transient *TransientScope
peer *PeerScope
onceDone sync.Once
}
var _ network.ConnectionScope = (*ConnectionScope)(nil)
@@ -95,6 +97,8 @@ type StreamScope struct {
peer *PeerScope
svc *ServiceScope
proto *ProtocolScope
onceDone sync.Once
}
var _ network.StreamScope = (*StreamScope)(nil)
@@ -381,9 +385,11 @@ func (s *ConnectionScope) SetPeer(p peer.ID) error {
}
func (s *ConnectionScope) Done() {
if s.peer != nil {
s.peer.DecRef()
}
s.onceDone.Do(func() {
if s.peer != nil {
s.peer.DecRef()
}
})
s.ResourceScope.Done()
}
@@ -492,11 +498,13 @@ func (s *StreamScope) PeerScope() network.PeerScope {
}
func (s *StreamScope) Done() {
s.peer.DecRef()
s.onceDone.Do(func() {
s.peer.DecRef()
if s.proto != nil {
s.proto.DecRef()
}
if s.proto != nil {
s.proto.DecRef()
}
})
s.ResourceScope.Done()
}