Refactor.

License: MIT Signed-off-by: Kevin Atkinson <k@kevina.org>
ipfs · Mar 2, 2017 · 93ec38d · 93ec38d
1 parent 5087db5
commit 93ec38d
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/filestore/fsrefstore.go b/filestore/fsrefstore.go
@@ -84,6 +84,20 @@ func (f *FileManager) DeleteBlock(c *cid.Cid) error {
 }
 
 func (f *FileManager) Get(c *cid.Cid) (blocks.Block, error) {
+	dobj, err := f.getDataObj(c)
+	if err != nil {
+		return nil, err
+	}
+
+	out, err := f.readDataObj(c, dobj)
+	if err != nil {
+		return nil, err
+	}
+
+	return blocks.NewBlockWithCid(out, c)
+}
+
+func (f *FileManager) getDataObj(c *cid.Cid) (*pb.DataObj, error) {
 	o, err := f.ds.Get(dshelp.CidToDsKey(c))
 	switch err {
 	case ds.ErrNotFound:
@@ -104,12 +118,7 @@ func (f *FileManager) Get(c *cid.Cid) (blocks.Block, error) {
 		return nil, err
 	}
 
-	out, err := f.readDataObj(c, &dobj)
-	if err != nil {
-		return nil, err
-	}
-
-	return blocks.NewBlockWithCid(out, c)
+	return &dobj, nil
 }
 
 // reads and verifies the block