libp2p/go-libp2p-peerstore
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-peerstore.git synced 2026-08-21 15:23:28 +08:00
Code Issues Projects Releases Wiki Activity

peer supports any protocol

Browse Source
This commit is contained in:
Aarsh Shah
2020-05-08 12:10:14 +05:30
parent b1f58c0365
commit 7a58f873f4
5 changed files with 56 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pstoremem/protobook.go
View File

@@ -163,3 +163,20 @@ func (pb *memoryProtoBook) SupportsProtocols(p peer.ID, protos ...string) ([]str
return out, nil
}
func (pb *memoryProtoBook) SupportsAnyProtocol(p peer.ID, protos ...string) (bool, error) {
if err := p.Validate(); err != nil {
return false, err
}
s := pb.segments.get(p)
s.RLock()
defer s.RUnlock()
for _, proto := range protos {
if _, ok := s.protocols[p][proto]; ok {
return true, nil
}
}
return false, nil
}