take a pin lock while performing ipfs files operations

whyrusleeping · whyrusleeping · commit f20b727acf2f · 2016-01-05T04:20:15.000-08:00
License: MIT
Signed-off-by: Jeromy &lt;jeromyj@gmail.com&gt;
diff --git a/core/commands/files/files.go b/core/commands/files/files.go
@@ -59,6 +59,9 @@ var FilesStatCmd = &cmds.Command{
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer node.Blockstore.PinLock()()
+
 		path, err := checkPath(req.Arguments()[0])
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -138,6 +141,9 @@ var FilesCpCmd = &cmds.Command{
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer node.Blockstore.PinLock()()
+
 		src, err := checkPath(req.Arguments()[0])
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -233,6 +239,9 @@ Examples:
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer nd.Blockstore.PinLock()()
+
 		fsn, err := mfs.Lookup(nd.FilesRoot, path)
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -323,6 +332,9 @@ Examples:
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer n.Blockstore.PinLock()()
+
 		path, err := checkPath(req.Arguments()[0])
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -409,6 +421,9 @@ Example:
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer n.Blockstore.PinLock()()
+
 		src, err := checkPath(req.Arguments()[0])
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -484,6 +499,9 @@ Warning:
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer nd.Blockstore.PinLock()()
+
 		offset, _, err := req.Option("offset").Int()
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
@@ -579,6 +597,9 @@ Examples:
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer n.Blockstore.PinLock()()
+
 		dashp, _, _ := req.Option("parents").Bool()
 		dirtomake, err := checkPath(req.Arguments()[0])
 		if err != nil {
@@ -628,6 +649,9 @@ remove files or directories
 			return
 		}
 
+		// take the lock and defer the unlock
+		defer nd.Blockstore.PinLock()()
+
 		path, err := checkPath(req.Arguments()[0])
 		if err != nil {
 			res.SetError(err, cmds.ErrNormal)
