From 930fde056a15b238c10861377bbf2fb7a29b171c Mon Sep 17 00:00:00 2001 From: Ethan Buchman Date: Sat, 20 Jan 2018 21:28:00 -0500 Subject: [PATCH] p2p: add back lost func --- p2p/addrbook.go | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/p2p/addrbook.go b/p2p/addrbook.go index 7591c307..fee179cb 100644 --- a/p2p/addrbook.go +++ b/p2p/addrbook.go @@ -324,6 +324,30 @@ func (a *AddrBook) GetSelection() []*NetAddress { return allAddr[:numAddresses] } +// ListOfKnownAddresses returns the new and old addresses. +func (a *AddrBook) ListOfKnownAddresses() []*knownAddress { + a.mtx.Lock() + defer a.mtx.Unlock() + + addrs := []*knownAddress{} + for _, addr := range a.addrLookup { + addrs = append(addrs, addr.copy()) + } + return addrs +} + +func (ka *knownAddress) copy() *knownAddress { + return &knownAddress{ + Addr: ka.Addr, + Src: ka.Src, + Attempts: ka.Attempts, + LastAttempt: ka.LastAttempt, + LastSuccess: ka.LastSuccess, + BucketType: ka.BucketType, + Buckets: ka.Buckets, + } +} + /* Loading & Saving */ type addrBookJSON struct {