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

refac(commands2/pin) loop

Browse Source
This commit is contained in:
Brian Tiger Chow 2014-11-06 04:14:39 -08:00 committed by Juan Batiz-Benet
parent 353cb39921
commit 86f0aac88c
1 changed files with 32 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
core/commands2/pin.go
View File

@ -5,6 +5,8 @@ import (
"fmt"
cmds "github.com/jbenet/go-ipfs/commands"
"github.com/jbenet/go-ipfs/core"
"github.com/jbenet/go-ipfs/merkledag"
)
var pinCmd = &cmds.Command{
@ -35,27 +37,17 @@ var pinCmd = &cmds.Command{
}
}*/
paths := make([]string, 0)
for _, arg := range req.Arguments() {
path, ok := arg.(string)
if !ok {
res.SetError(errors.New("cast error"), cmds.ErrNormal)
return
}
dagnode, err := n.Resolver.ResolvePath(path)
if err != nil {
res.SetError(fmt.Errorf("pin error: %v", err), cmds.ErrNormal)
return
}
err = n.Pinning.Pin(dagnode, recursive)
if err != nil {
res.SetError(fmt.Errorf("pin: %v", err), cmds.ErrNormal)
return
}
paths = append(paths, path)
}
err := n.Pinning.Flush()
_, err := pin(n, paths, recursive)
if err != nil {
res.SetError(err, cmds.ErrNormal)
}
@ -107,3 +99,30 @@ var unpinCmd = &cmds.Command{
// TODO: create some output to show what got unpinned
},
}
func pin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, error) {
dagnodes := make([]*merkledag.Node, 0)
for _, path := range paths {
dagnode, err := n.Resolver.ResolvePath(path)
if err != nil {
return nil, fmt.Errorf("pin error: %v", err)
}
dagnodes = append(dagnodes, dagnode)
}
for _, dagnode := range dagnodes {
err := n.Pinning.Pin(dagnode, recursive)
if err != nil {
return nil, fmt.Errorf("pin: %v", err)
}
}
err := n.Pinning.Flush()
if err != nil {
return nil, err
}
return dagnodes, nil
}