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{} ...@@ -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 // Take advantage of blockservice/bitswap batched requests to fetch all
// child nodes of a given node // child nodes of a given node
// TODO: finish this // TODO: finish this
func (ds *dagService) BatchFetch(ctx context.Context, root *Node) chan struct{} { func (ds *dagService) BatchFetch(ctx context.Context, root *Node) <-chan int {
sig := make(chan struct{}) sig := make(chan int)
go func() { go func() {
var keys []u.Key var keys []u.Key
for _, lnk := range root.Links { for _, lnk := range root.Links {
...@@ -323,11 +323,11 @@ func (ds *dagService) BatchFetch(ctx context.Context, root *Node) chan struct{} ...@@ -323,11 +323,11 @@ func (ds *dagService) BatchFetch(ctx context.Context, root *Node) chan struct{}
// //
if next == i { if next == i {
sig <- struct{}{} sig <- next
next++ next++
for { for {
if _, ok := seen[next]; ok { if _, ok := seen[next]; ok {
sig <- struct{}{} sig <- next
next++ next++
} else { } else {
break 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