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

Merge pull request #794 from jbenet/feat/refs-faster

Browse Source 
teach ipfs refs -r how to use GetBlocks
This commit is contained in:
Jeromy Johnson 2015-02-18 20:57:19 -08:00
commit fa3a9e2ec9
1 changed files with 43 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
core/commands/refs.go
View File

@ -227,6 +227,49 @@ type RefWriter struct {
// WriteRefs writes refs of the given object to the underlying writer.
func (rw *RefWriter) WriteRefs(n *dag.Node) (int, error) {
if rw.Recursive {
return rw.writeRefsRecursive(n)
} else {
return rw.writeRefsSingle(n)
}
}
func (rw *RefWriter) writeRefsRecursive(n *dag.Node) (int, error) {
nkey, err := n.Key()
if err != nil {
return 0, err
}
if rw.skip(nkey) {
return 0, nil
}
var count int
for i, ng := range rw.DAG.GetDAG(rw.Ctx, n) {
lk := u.Key(n.Links[i].Hash)
if rw.skip(lk) {
continue
}
if err := rw.WriteEdge(nkey, lk, n.Links[i].Name); err != nil {
return count, err
}
nd, err := ng.Get()
if err != nil {
return count, err
}
c, err := rw.writeRefsRecursive(nd)
count += c
if err != nil {
return count, err
}
}
return count, nil
}
func (rw *RefWriter) writeRefsSingle(n *dag.Node) (int, error) {
nkey, err := n.Key()
if err != nil {
return 0, err
@ -248,21 +291,6 @@ func (rw *RefWriter) WriteRefs(n *dag.Node) (int, error) {
return count, err
}
count++
if !rw.Recursive {
continue
}
child, err := l.GetNode(rw.DAG)
if err != nil {
return count, err
}
c, err := rw.WriteRefs(child)
count += c
if err != nil {
return count, err
}
}
return count, nil
}