Commit 43c07c7e authored by Jeromy's avatar Jeromy

change BatchFetch to return indices

parent 52e9cde2
......@@ -289,8 +289,8 @@ func FetchGraph(ctx context.Context, root *Node, serv DAGService) chan struct{}
// Take advantage of blockservice/bitswap batched requests to fetch all
// child nodes of a given node
// TODO: finish this
func (ds *dagService) BatchFetch(ctx context.Context, root *Node) chan struct{} {
sig := make(chan struct{})
func (ds *dagService) BatchFetch(ctx context.Context, root *Node) <-chan int {
sig := make(chan int)
go func() {
var keys []u.Key
for _, lnk := range root.Links {
......@@ -323,11 +323,11 @@ func (ds *dagService) BatchFetch(ctx context.Context, root *Node) chan struct{}
//
if next == i {
sig <- struct{}{}
sig <- next
next++
for {
if _, ok := seen[next]; ok {
sig <- struct{}{}
sig <- next
next++
} else {
break
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment