fix(bitswap) always add addresses
Brian Tiger Chow committed
Dec 23, 2014 at 07:15 UTC
c6294646e9d3fc408a954a29216af52d37039445
1 file changed
+2
-2
exchange/bitswap/bitswap.go
+2
-2
@@ -180,6 +180,7 @@ func (bs *bitswap) sendWantListTo(ctx context.Context, peers <-chan peer.PeerInf
180
for peerToQuery := range peers {
181
log.Event(ctx, "PeerToQuery", peerToQuery.ID)
182
wg.Add(1)
183
+ bs.network.Peerstore().AddAddresses(peerToQuery.ID, peerToQuery.Addrs)
184
go func(p peer.ID) {
185
defer wg.Done()
186
if err := bs.send(ctx, p, message); err != nil {
@@ -212,8 +213,8 @@ func (bs *bitswap) sendWantlistToProviders(ctx context.Context, wantlist *wantli
213
defer wg.Done()
214
child, _ := context.WithTimeout(ctx, providerRequestTimeout)
215
providers := bs.routing.FindProvidersAsync(child, k, maxProvidersPerRequest)
215
-
216
for prov := range providers {
217
+ bs.network.Peerstore().AddAddresses(prov.ID, prov.Addrs)
218
if set.TryAdd(prov.ID) { //Do once per peer
219
bs.send(ctx, prov.ID, message)
220
}
@@ -265,7 +266,6 @@ func (bs *bitswap) clientWorker(parent context.Context) {
266
// newer bitswap strategies.
267
child, _ := context.WithTimeout(ctx, providerRequestTimeout)
268
providers := bs.routing.FindProvidersAsync(child, ks[0], maxProvidersPerRequest)
268
-
269
err := bs.sendWantListTo(ctx, providers)
270
if err != nil {
271
log.Errorf("error sending wantlist: %s", err)