coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-03-09 18:28:08 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Don't do extra work in provider queue loop

Browse Source 
License: MIT
Signed-off-by: Erik Ingenito <erik@carbonfive.com>
This commit is contained in:
Erik Ingenito 2019-03-16 09:52:24 -07:00
parent 6c1eca959e
commit 906d2bd093
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
provider/queue.go
View File

@ -2,13 +2,14 @@ package provider
import (
"context"
"github.com/ipfs/go-cid"
"github.com/ipfs/go-datastore"
"github.com/ipfs/go-datastore/namespace"
"github.com/ipfs/go-datastore/query"
"math"
"strconv"
"strings"
cid "github.com/ipfs/go-cid"
datastore "github.com/ipfs/go-datastore"
namespace "github.com/ipfs/go-datastore/namespace"
query "github.com/ipfs/go-datastore/query"
)
// Queue provides a durable, FIFO interface to the datastore for storing cids
@ -100,11 +101,16 @@ func (q *Queue) nextEntry() (datastore.Key, cid.Cid) {
// Run dequeues and enqueues when available.
func (q *Queue) work() {
go func() {
var k datastore.Key = datastore.Key{}
var c cid.Cid = cid.Undef
for {
k, c := q.nextEntry()
var dequeue chan cid.Cid
if c == cid.Undef {
k, c = q.nextEntry()
}
// If c != cid.Undef set dequeue and attempt write, otherwise wait for enqueue
var dequeue chan cid.Cid
if c != cid.Undef {
dequeue = q.dequeue
}
@ -126,7 +132,7 @@ func (q *Queue) work() {
log.Errorf("Failed to delete queued cid %s with key %s: %s", c, k, err)
continue
}
c = cid.Undef
q.head++
case <-q.ctx.Done():
return