hotstore now memrepo and somehow it works for compaction!

blockstore/mem.go

@@ -107,3 +107,12 @@ func (m MemBlockstore) AllKeysChan(ctx context.Context) (<-chan cid.Cid, error)
 func (m MemBlockstore) HashOnRead(enabled bool) {
 	// no-op
 }
+
+func (m MemBlockstore) ForEachKey(f func(c cid.Cid) error) error {
+	for _, b := range m {
+		if err := f(b.Cid()); err != nil {
+			return err
+		}
+	}
+	return nil
+}

blockstore/sync.go

@@ -81,3 +81,9 @@ func (m *SyncBlockstore) AllKeysChan(ctx context.Context) (<-chan cid.Cid, error
 func (m *SyncBlockstore) HashOnRead(enabled bool) {
 	// noop
 }
+
+func (m *SyncBlockstore) ForEachKey(f func(c cid.Cid) error) error {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.bs.ForEachKey(f)
+}

node/repo/fsrepo.go

@@ -436,8 +436,9 @@ func (fsr *fsLockedRepo) Blockstore(ctx context.Context, domain BlockstoreDomain
 		log.Error("invalid blockstore domain", domain)
 		return nil, nil, ErrInvalidBlockstoreDomain
 	}
-	var bs *badgerbs.Blockstore
+	var bs blockstore.Blockstore
 	var err error
+	var close func() error
 	if domain == UniversalBlockstore {
 		fsr.bsOnce.Do(func() {
 			path := fsr.join(filepath.Join(fsDatastore, "chain"))
@@ -458,12 +459,14 @@ func (fsr *fsLockedRepo) Blockstore(ctx context.Context, domain BlockstoreDomain
 				opts.SyncWrites = false
 			}
 
-			bs, err = badgerbs.Open(opts)
+			bbs, err := badgerbs.Open(opts)
 			if err != nil {
 				fsr.bsErr = err
 				return
 			}
-			fsr.bs = blockstore.WrapIDStore(bs)
+			close = bbs.Close
+			fsr.bs = blockstore.WrapIDStore(bbs)
+			bs = fsr.bs
 		})
 		err = fsr.bsErr
 	} else {
@@ -485,17 +488,19 @@ func (fsr *fsLockedRepo) Blockstore(ctx context.Context, domain BlockstoreDomain
 				fsr.bsHotErr = err
 				return
 			}
-			bs, err = badgerbs.Open(opts)
+			bbs, err := badgerbs.Open(opts)
 			if err != nil {
 				fsr.bsHotErr = err
 				return
 			}
-			fsr.bsHot = bs
+			fsr.bsHot = bbs
+			close = bbs.Close
+			bs = fsr.bsHot
 		})
 		err = fsr.bsHotErr
 	}
 
-	return bs, bs.Close, err
+	return bs, close, err
 }
 
 func (fsr *fsLockedRepo) SplitstorePath() (string, error) {