Merge pull request #20 from jenyasd209/fix/visit_all_blocks

Visit all blocks
Wondertan · Oct 12, 2020 · c6d9634 · c6d9634
2 parents b561ea8 + ac89c02
commit c6d9634
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/ipld/walk.go b/ipld/walk.go
@@ -71,9 +71,19 @@ func Walk(ctx context.Context, id cid.Cid, bs blockstream.BlockStreamer, handler
 				return err
 			}
 
-			err = wo.handle(nd)
-			if err != nil {
-				return err
+			shouldHandle := true
+			if nd.Cid().Type() != cid.Raw {
+				shouldHandle, err = wo.visit(nd.Cid())
+				if err != nil {
+					return err
+				}
+			}
+
+			if shouldHandle {
+				err = wo.handle(nd)
+				if err != nil {
+					return err
+				}
 			}
 
 			ids := make([]cid.Cid, 0, len(nd.Links()))