coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-02-26 12:57:44 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

add priorities to GetBlocks requests, and add waitgroup to sendWantListTo

Browse Source
This commit is contained in:
Jeromy 2014-12-10 21:12:07 +00:00 committed by Juan Batiz-Benet
parent 3778eedff0
commit e7bba82dcb
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
exchange/bitswap/bitswap.go
View File

@ -174,10 +174,12 @@ func (bs *bitswap) sendWantListTo(ctx context.Context, peers <-chan peer.Peer) e
for _, wanted := range bs.wantlist.Entries() {
message.AddEntry(wanted.Value, wanted.Priority, false)
}
wg := sync.WaitGroup{}
for peerToQuery := range peers {
log.Debug("sending query to: %s", peerToQuery)
log.Event(ctx, "PeerToQuery", peerToQuery)
wg.Add(1)
go func(p peer.Peer) {
defer wg.Done()
log.Event(ctx, "DialPeer", p)
err := bs.sender.DialPeer(ctx, p)
@ -197,6 +199,7 @@ func (bs *bitswap) sendWantListTo(ctx context.Context, peers <-chan peer.Peer) e
bs.ledgerset.MessageSent(p, message)
}(peerToQuery)
}
wg.Wait()
return nil
}
@ -305,8 +308,8 @@ func (bs *bitswap) clientWorker(parent context.Context) {
log.Warning("Received batch request for zero blocks")
continue
}
for _, k := range ks {
bs.wantlist.Add(k, 1)
for i, k := range ks {
bs.wantlist.Add(k, len(ks)-i)
}
// NB: send want list to providers for the first peer in this list.
// the assumption is made that the providers of the first key in