99 "sync"
1010 "time"
1111
12+ "github.com/libp2p/go-libp2p/p2p/host/pstoremanager"
1213 "github.com/libp2p/go-libp2p/p2p/host/relaysvc"
1314 relayv2 "github.com/libp2p/go-libp2p/p2p/protocol/circuitv2/relay"
1415 "github.com/libp2p/go-libp2p/p2p/protocol/holepunch"
@@ -74,6 +75,7 @@ type BasicHost struct {
7475 refCount sync.WaitGroup
7576
7677 network network.Network
78+ psManager * pstoremanager.PeerstoreManager
7779 mux * msmux.MultistreamMuxer
7880 ids identify.IDService
7981 hps * holepunch.Service
@@ -156,18 +158,24 @@ type HostOpts struct {
156158
157159// NewHost constructs a new *BasicHost and activates it by attaching its stream and connection handlers to the given inet.Network.
158160func NewHost (n network.Network , opts * HostOpts ) (* BasicHost , error ) {
161+ eventBus := eventbus .NewBus ()
162+ psManager , err := pstoremanager .NewPeerstoreManager (n .Peerstore (), eventBus )
163+ if err != nil {
164+ return nil , err
165+ }
159166 hostCtx , cancel := context .WithCancel (context .Background ())
160167 if opts == nil {
161168 opts = & HostOpts {}
162169 }
163170
164171 h := & BasicHost {
165172 network : n ,
173+ psManager : psManager ,
166174 mux : msmux .NewMultistreamMuxer (),
167175 negtimeout : DefaultNegotiationTimeout ,
168176 AddrsFactory : DefaultAddrsFactory ,
169177 maResolver : madns .DefaultResolver ,
170- eventbus : eventbus . NewBus () ,
178+ eventbus : eventBus ,
171179 addrChangeChan : make (chan struct {}, 1 ),
172180 ctx : hostCtx ,
173181 ctxCancel : cancel ,
@@ -176,7 +184,6 @@ func NewHost(n network.Network, opts *HostOpts) (*BasicHost, error) {
176184
177185 h .updateLocalIpAddr ()
178186
179- var err error
180187 if h .emitters .evtLocalProtocolsUpdated , err = h .eventbus .Emitter (& event.EvtLocalProtocolsUpdated {}); err != nil {
181188 return nil , err
182189 }
@@ -352,6 +359,7 @@ func (h *BasicHost) updateLocalIpAddr() {
352359
353360// Start starts background tasks in the host
354361func (h * BasicHost ) Start () {
362+ h .psManager .Start ()
355363 h .refCount .Add (1 )
356364 go h .background ()
357365}
@@ -1036,6 +1044,7 @@ func (h *BasicHost) Close() error {
10361044 _ = h .emitters .evtLocalAddrsUpdated .Close ()
10371045 h .Network ().Close ()
10381046
1047+ h .psManager .Close ()
10391048 if h .Peerstore () != nil {
10401049 h .Peerstore ().Close ()
10411050 }
0 commit comments