mirror of
https://github.com/libp2p/go-libp2p-core.git
synced 2025-01-14 02:20:06 +08:00
ea8b36912f
* run go mod tidy
* omit receiver name if unused
* remove unused type testkey in tests
* fix duplicate import of go-multiaddr
* fix use of deprecated peer.IDB58{Encode,Decode}
* use bytes.Equal instead of bytes.Compare
* fix unnecessary assigments to blank identifier
* use time.Until instead of t.Sub(time.Now())
* fix use of deprecated go-multihash.ID
* add missing error check in envelope test
* fix error check in tests
72 lines
1.2 KiB
Go
72 lines
1.2 KiB
Go
package peer
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// PeerSet is a threadsafe set of peers.
|
|
type Set struct {
|
|
lk sync.RWMutex
|
|
ps map[ID]struct{}
|
|
|
|
size int
|
|
}
|
|
|
|
func NewSet() *Set {
|
|
ps := new(Set)
|
|
ps.ps = make(map[ID]struct{})
|
|
ps.size = -1
|
|
return ps
|
|
}
|
|
|
|
func NewLimitedSet(size int) *Set {
|
|
ps := new(Set)
|
|
ps.ps = make(map[ID]struct{})
|
|
ps.size = size
|
|
return ps
|
|
}
|
|
|
|
func (ps *Set) Add(p ID) {
|
|
ps.lk.Lock()
|
|
ps.ps[p] = struct{}{}
|
|
ps.lk.Unlock()
|
|
}
|
|
|
|
func (ps *Set) Contains(p ID) bool {
|
|
ps.lk.RLock()
|
|
_, ok := ps.ps[p]
|
|
ps.lk.RUnlock()
|
|
return ok
|
|
}
|
|
|
|
func (ps *Set) Size() int {
|
|
ps.lk.RLock()
|
|
defer ps.lk.RUnlock()
|
|
return len(ps.ps)
|
|
}
|
|
|
|
// TryAdd Attempts to add the given peer into the set.
|
|
// This operation can fail for one of two reasons:
|
|
// 1) The given peer is already in the set
|
|
// 2) The number of peers in the set is equal to size
|
|
func (ps *Set) TryAdd(p ID) bool {
|
|
var success bool
|
|
ps.lk.Lock()
|
|
if _, ok := ps.ps[p]; !ok && (len(ps.ps) < ps.size || ps.size == -1) {
|
|
success = true
|
|
ps.ps[p] = struct{}{}
|
|
}
|
|
ps.lk.Unlock()
|
|
return success
|
|
}
|
|
|
|
func (ps *Set) Peers() []ID {
|
|
ps.lk.Lock()
|
|
out := make([]ID, 0, len(ps.ps))
|
|
for p := range ps.ps {
|
|
out = append(out, p)
|
|
}
|
|
ps.lk.Unlock()
|
|
return out
|
|
}
|