handle id cids in internal versions of view/get

filecoin-project · Jul 9, 2021 · 870a47f · 870a47f
1 parent f5ae10e
commit 870a47f
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/blockstore/splitstore/splitstore.go b/blockstore/splitstore/splitstore.go
@@ -1316,18 +1316,31 @@ func (s *SplitStore) walkObjectIncomplete(c cid.Cid, walked *cid.Set, f, missing
 }
 
 // internal version used by walk
-func (s *SplitStore) view(cid cid.Cid, cb func([]byte) error) error {
-	err := s.hot.View(cid, cb)
+func (s *SplitStore) view(c cid.Cid, cb func([]byte) error) error {
+	if isIdentiyCid(c) {
+		data, err := decodeIdentityCid(c)
+		if err != nil {
+			return err
+		}
+
+		return cb(data)
+	}
+
+	err := s.hot.View(c, cb)
 	switch err {
 	case bstore.ErrNotFound:
-		return s.cold.View(cid, cb)
+		return s.cold.View(c, cb)
 
 	default:
 		return err
 	}
 }
 
 func (s *SplitStore) has(c cid.Cid) (bool, error) {
+	if isIdentiyCid(c) {
+		return true, nil
+	}
+
 	has, err := s.hot.Has(c)
 
 	if has || err != nil {