79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
package cachebs
|
|
|
|
import (
|
|
"context"
|
|
|
|
lru "github.com/hashicorp/golang-lru"
|
|
block "github.com/ipfs/go-block-format"
|
|
"github.com/ipfs/go-cid"
|
|
blockstore "github.com/ipfs/go-ipfs-blockstore"
|
|
bstore "github.com/ipfs/go-ipfs-blockstore"
|
|
logging "github.com/ipfs/go-log/v2"
|
|
)
|
|
|
|
var log = logging.Logger("cachebs")
|
|
|
|
type CacheBS struct {
|
|
cache *lru.ARCCache
|
|
bs bstore.Blockstore
|
|
}
|
|
|
|
func NewBufferedBstore(base blockstore.Blockstore, size int) *CacheBS {
|
|
c, err := lru.NewARC(size)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &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 {
|
|
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
|
|
}
|
|
|
|
return bs.bs.Get(c)
|
|
}
|
|
|
|
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)
|
|
}
|