add method to peerset for returning all peers in set

rargulati · Mar 5, 2017 · 962a676 · 962a676
1 parent c022ceb
commit 962a676
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/peerset/peerset.go b/peerset/peerset.go
@@ -60,3 +60,13 @@ func (ps *PeerSet) TryAdd(p peer.ID) bool {
 	ps.lk.Unlock()
 	return success
 }
+
+func (ps *PeerSet) Peers() []peer.ID {
+	ps.lk.Lock()
+	out := make([]peer.ID, 0, len(ps.ps))
+	for p, _ := range ps.ps {
+		out = append(out, p)
+	}
+	ps.lk.Unlock()
+	return out
+}