opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-30 18:13:54 +08:00
Code Issues Projects Releases Wiki Activity

simplify provideCollector

Browse Source
This commit is contained in:
Jeromy
2015-03-05 16:27:47 -08:00
parent 8937f5fbda
commit bfee4894b4
1 changed files with 9 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
exchange/bitswap/workers.go
View File

@ -83,17 +83,7 @@ func (bs *Bitswap) provideCollector(ctx context.Context) {
defer close(bs.provideKeys)
var toprovide []u.Key
var nextKey u.Key
select {
case blk, ok := <-bs.newBlocks:
if !ok {
log.Debug("newBlocks channel closed")
return
}
nextKey = blk.Key()
case <-ctx.Done():
return
}
var keysOut chan u.Key
for {
select {
@ -102,21 +92,18 @@ func (bs *Bitswap) provideCollector(ctx context.Context) {
log.Debug("newBlocks channel closed")
return
}
if keysOut == nil {
nextKey = blk.Key()
keysOut = bs.provideKeys
} else {
toprovide = append(toprovide, blk.Key())
case bs.provideKeys <- nextKey:
}
case keysOut <- nextKey:
if len(toprovide) > 0 {
nextKey = toprovide[0]
toprovide = toprovide[1:]
} else {
select {
case blk, ok := <-bs.newBlocks:
if !ok {
return
}
nextKey = blk.Key()
case <-ctx.Done():
return
}
keysOut = nil
}
case <-ctx.Done():
return