5733c71c50
We were ignoring quite a few error cases, and had one case where we weren't actually updating state where we wanted to. Unfortunately, if the linter doesn't pass, nobody has any reason to actually check lint failures in CI. There are three remaining XXXs marked in the code for lint.
90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package cachebs
|
|
|
|
import (
|
|
"context"
|
|
|
|
lru "github.com/hashicorp/golang-lru"
|
|
block "github.com/ipfs/go-block-format"
|
|
"github.com/ipfs/go-cid"
|
|
logging "github.com/ipfs/go-log/v2"
|
|
|
|
bstore "github.com/filecoin-project/lotus/lib/blockstore"
|
|
)
|
|
|
|
//nolint:deadcode,varcheck
|
|
var log = logging.Logger("cachebs")
|
|
|
|
type CacheBS struct {
|
|
cache *lru.ARCCache
|
|
bs bstore.Blockstore
|
|
}
|
|
|
|
func NewBufferedBstore(base bstore.Blockstore, size int) bstore.Blockstore {
|
|
c, err := lru.NewARC(size)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// Wrap this in an ID blockstore to avoid caching blocks inlined into
|
|
// CIDs.
|
|
return bstore.WrapIDStore(&CacheBS{
|
|
cache: c,
|
|
bs: base,
|
|
})
|
|
}
|
|
|
|
var _ (bstore.Blockstore) = &CacheBS{}
|
|
|
|
func (bs *CacheBS) AllKeysChan(ctx context.Context) (<-chan cid.Cid, error) {
|
|
return bs.bs.AllKeysChan(ctx)
|
|
}
|
|
|
|
func (bs *CacheBS) DeleteBlock(c cid.Cid) error {
|
|
bs.cache.Remove(c)
|
|
|
|
return bs.bs.DeleteBlock(c)
|
|
}
|
|
|
|
func (bs *CacheBS) Get(c cid.Cid) (block.Block, error) {
|
|
v, ok := bs.cache.Get(c)
|
|
if ok {
|
|
return v.(block.Block), nil
|
|
}
|
|
|
|
out, err := bs.bs.Get(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
bs.cache.Add(c, out)
|
|
return out, nil
|
|
}
|
|
|
|
func (bs *CacheBS) GetSize(c cid.Cid) (int, error) {
|
|
return bs.bs.GetSize(c)
|
|
}
|
|
|
|
func (bs *CacheBS) Put(blk block.Block) error {
|
|
bs.cache.Add(blk.Cid(), blk)
|
|
|
|
return bs.bs.Put(blk)
|
|
}
|
|
|
|
func (bs *CacheBS) Has(c cid.Cid) (bool, error) {
|
|
if bs.cache.Contains(c) {
|
|
return true, nil
|
|
}
|
|
|
|
return bs.bs.Has(c)
|
|
}
|
|
|
|
func (bs *CacheBS) HashOnRead(hor bool) {
|
|
bs.bs.HashOnRead(hor)
|
|
}
|
|
|
|
func (bs *CacheBS) PutMany(blks []block.Block) error {
|
|
for _, blk := range blks {
|
|
bs.cache.Add(blk.Cid(), blk)
|
|
}
|
|
return bs.bs.PutMany(blks)
|
|
}
|