blockstore: add helper functions to convert from Cid to DsKey and back

License: MIT Signed-off-by: Kevin Atkinson <k@kevina.org>
ipfs · Oct 16, 2016 · 2b09019 · 2b09019
1 parent d5c716a
commit 2b09019
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 14 deletions.
diff --git a/blocks/blockstore/blockstore.go b/blocks/blockstore/blockstore.go
@@ -9,7 +9,6 @@ import (
 	"sync/atomic"
 
 	blocks "github.com/ipfs/go-ipfs/blocks"
-	dshelp "github.com/ipfs/go-ipfs/thirdparty/ds-help"
 
 	logging "gx/ipfs/QmSpJByNKFX1sCsHBEp3R73FL4NF6FnQTEGyNAXHm2GS52/go-log"
 	cid "gx/ipfs/QmXUuRadqDq5BuFWzVU6VuKaSjTcNm1gNCtLvvP1TJCW4z/go-cid"
@@ -87,7 +86,7 @@ func (bs *blockstore) Get(k *cid.Cid) (blocks.Block, error) {
 		return nil, ErrNotFound
 	}
 
-	maybeData, err := bs.datastore.Get(dshelp.NewKeyFromBinary(k.KeyString()))
+	maybeData, err := bs.datastore.Get(CidToDsKey(k))
 	if err == ds.ErrNotFound {
 		return nil, ErrNotFound
 	}
@@ -112,7 +111,7 @@ func (bs *blockstore) Get(k *cid.Cid) (blocks.Block, error) {
 }
 
 func (bs *blockstore) Put(block blocks.Block) error {
-	k := dshelp.NewKeyFromBinary(block.Cid().KeyString())
+	k := CidToDsKey(block.Cid())
 
 	// Has is cheaper than Put, so see if we already have it
 	exists, err := bs.datastore.Has(k)
@@ -128,7 +127,7 @@ func (bs *blockstore) PutMany(blocks []blocks.Block) error {
 		return err
 	}
 	for _, b := range blocks {
-		k := dshelp.NewKeyFromBinary(b.Cid().KeyString())
+		k := CidToDsKey(b.Cid())
 		exists, err := bs.datastore.Has(k)
 		if err == nil && exists {
 			continue
@@ -143,11 +142,11 @@ func (bs *blockstore) PutMany(blocks []blocks.Block) error {
 }
 
 func (bs *blockstore) Has(k *cid.Cid) (bool, error) {
-	return bs.datastore.Has(dshelp.NewKeyFromBinary(k.KeyString()))
+	return bs.datastore.Has(CidToDsKey(k))
 }
 
 func (s *blockstore) DeleteBlock(k *cid.Cid) error {
-	return s.datastore.Delete(dshelp.NewKeyFromBinary(k.KeyString()))
+	return s.datastore.Delete(CidToDsKey(k))
 }
 
 // AllKeysChan runs a query for keys from the blockstore.
@@ -180,17 +179,12 @@ func (bs *blockstore) AllKeysChan(ctx context.Context) (<-chan *cid.Cid, error)
 			}
 
 			// need to convert to key.Key using key.KeyFromDsKey.
-			kb, err := dshelp.BinaryFromDsKey(ds.NewKey(e.Key)) // TODO: calling NewKey isnt free
+			c, err := DsKeyToCid(ds.NewKey(e.Key)) // TODO: calling NewKey isnt free
 			if err != nil {
 				log.Warningf("error parsing key from DsKey: ", err)
 				return nil, true
 			}
 
-			c, err := cid.Cast(kb)
-			if err != nil {
-				log.Warning("error parsing cid from decoded DsKey: ", err)
-				return nil, true
-			}
 			log.Debug("blockstore: query got key", c)
 
 			return c, true

diff --git a/blocks/blockstore/blockstore_test.go b/blocks/blockstore/blockstore_test.go
@@ -7,7 +7,6 @@ import (
 	"testing"
 
 	blocks "github.com/ipfs/go-ipfs/blocks"
-	dshelp "github.com/ipfs/go-ipfs/thirdparty/ds-help"
 
 	cid "gx/ipfs/QmXUuRadqDq5BuFWzVU6VuKaSjTcNm1gNCtLvvP1TJCW4z/go-cid"
 	u "gx/ipfs/Qmb912gdngC1UWwTkhuW8knyRbcWeu5kqkxBpveLmW8bSr/go-ipfs-util"
@@ -190,7 +189,7 @@ func TestValueTypeMismatch(t *testing.T) {
 	block := blocks.NewBlock([]byte("some data"))
 
 	datastore := ds.NewMapDatastore()
-	k := BlockPrefix.Child(dshelp.NewKeyFromBinary(block.Cid().KeyString()))
+	k := BlockPrefix.Child(CidToDsKey(block.Cid()))
 	datastore.Put(k, "data that isn't a block!")
 
 	blockstore := NewBlockstore(ds_sync.MutexWrap(datastore))

diff --git a/blocks/blockstore/key.go b/blocks/blockstore/key.go
@@ -0,0 +1,20 @@
+package blockstore
+
+import (
+	dshelp "github.com/ipfs/go-ipfs/thirdparty/ds-help"
+
+	cid "gx/ipfs/QmXUuRadqDq5BuFWzVU6VuKaSjTcNm1gNCtLvvP1TJCW4z/go-cid"
+	ds "gx/ipfs/QmbzuUusHqaLLoNTDEVLcSF6vZDHZDLPC7p4bztRvvkXxU/go-datastore"
+)
+
+func CidToDsKey(k *cid.Cid) ds.Key {
+	return dshelp.NewKeyFromBinary(k.KeyString())
+}
+
+func DsKeyToCid(dsKey ds.Key) (*cid.Cid, error) {
+	kb, err := dshelp.BinaryFromDsKey(dsKey)
+	if err != nil {
+		return nil, err
+	}
+	return cid.Cast(kb)
+}