fix(bitswap) always add addresses

ipfs · Dec 23, 2014 · 897e595 · 897e595
1 parent 71f780d
commit 897e595
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/exchange/bitswap/bitswap.go b/exchange/bitswap/bitswap.go
@@ -211,6 +211,9 @@ func (bs *bitswap) sendWantlistToProviders(ctx context.Context, wantlist *wantli
 			defer wg.Done()
 			child, _ := context.WithTimeout(ctx, providerRequestTimeout)
 			providers := bs.routing.FindProvidersAsync(child, k, maxProvidersPerRequest)
+			for _, info := range providers {
+				bs.network.Peerstore().AddAddresses(info.ID, info.Addresses)
+			}
 
 			for prov := range providers {
 				if set.TryAdd(prov.ID) { //Do once per peer
@@ -264,6 +267,9 @@ func (bs *bitswap) clientWorker(parent context.Context) {
 			//		newer bitswap strategies.
 			child, _ := context.WithTimeout(ctx, providerRequestTimeout)
 			providers := bs.routing.FindProvidersAsync(child, ks[0], maxProvidersPerRequest)
+			for _, info := range providers {
+				bs.network.Peerstore().AddAddresses(info.ID, info.Addresses)
+			}
 
 			err := bs.sendWantListTo(ctx, providers)
 			if err != nil {