feat: allow passing options to memoryAddrBook (#3062)

Co-authored-by: Daniel N <2color@users.noreply.github.com>
libp2p · Nov 25, 2024 · 666d1ed · 666d1ed
1 parent 288868c
commit 666d1ed
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/p2p/host/peerstore/pstoremem/addr_book.go b/p2p/host/peerstore/pstoremem/addr_book.go
@@ -201,7 +201,7 @@ type memoryAddrBook struct {
 var _ peerstore.AddrBook = (*memoryAddrBook)(nil)
 var _ peerstore.CertifiedAddrBook = (*memoryAddrBook)(nil)
 
-func NewAddrBook() *memoryAddrBook {
+func NewAddrBook(opts ...AddrBookOption) *memoryAddrBook {
 	ctx, cancel := context.WithCancel(context.Background())
 
 	ab := &memoryAddrBook{
@@ -211,8 +211,12 @@ func NewAddrBook() *memoryAddrBook {
 		cancel:               cancel,
 		clock:                realclock{},
 		maxUnconnectedAddrs:  defaultMaxUnconnectedAddrs,
-		maxSignedPeerRecords: defaultMaxUnconnectedAddrs,
+		maxSignedPeerRecords: defaultMaxSignedPeerRecords,
 	}
+	for _, opt := range opts {
+		opt(ab)
+	}
+
 	ab.refCount.Add(1)
 	go ab.background(ctx)
 	return ab