115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package splitstore
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
blocks "github.com/ipfs/go-block-format"
|
|
cid "github.com/ipfs/go-cid"
|
|
|
|
bstore "github.com/filecoin-project/lotus/blockstore"
|
|
)
|
|
|
|
type exposedSplitStore struct {
|
|
s *SplitStore
|
|
}
|
|
|
|
var _ bstore.Blockstore = (*exposedSplitStore)(nil)
|
|
|
|
func (s *SplitStore) Expose() bstore.Blockstore {
|
|
return &exposedSplitStore{s: s}
|
|
}
|
|
|
|
func (es *exposedSplitStore) DeleteBlock(_ context.Context, _ cid.Cid) error {
|
|
return errors.New("DeleteBlock: operation not supported")
|
|
}
|
|
|
|
func (es *exposedSplitStore) DeleteMany(_ context.Context, _ []cid.Cid) error {
|
|
return errors.New("DeleteMany: operation not supported")
|
|
}
|
|
|
|
func (es *exposedSplitStore) Has(ctx context.Context, c cid.Cid) (bool, error) {
|
|
if isIdentiyCid(c) {
|
|
return true, nil
|
|
}
|
|
|
|
has, err := es.s.hot.Has(ctx, c)
|
|
if has || err != nil {
|
|
return has, err
|
|
}
|
|
|
|
return es.s.cold.Has(ctx, c)
|
|
}
|
|
|
|
func (es *exposedSplitStore) Get(ctx context.Context, c cid.Cid) (blocks.Block, error) {
|
|
if isIdentiyCid(c) {
|
|
data, err := decodeIdentityCid(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return blocks.NewBlockWithCid(data, c)
|
|
}
|
|
|
|
blk, err := es.s.hot.Get(ctx, c)
|
|
switch err {
|
|
case bstore.ErrNotFound:
|
|
return es.s.cold.Get(ctx, c)
|
|
default:
|
|
return blk, err
|
|
}
|
|
}
|
|
|
|
func (es *exposedSplitStore) GetSize(ctx context.Context, c cid.Cid) (int, error) {
|
|
if isIdentiyCid(c) {
|
|
data, err := decodeIdentityCid(c)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return len(data), nil
|
|
}
|
|
|
|
size, err := es.s.hot.GetSize(ctx, c)
|
|
switch err {
|
|
case bstore.ErrNotFound:
|
|
return es.s.cold.GetSize(ctx, c)
|
|
default:
|
|
return size, err
|
|
}
|
|
}
|
|
|
|
func (es *exposedSplitStore) Put(ctx context.Context, blk blocks.Block) error {
|
|
return es.s.Put(ctx, blk)
|
|
}
|
|
|
|
func (es *exposedSplitStore) PutMany(ctx context.Context, blks []blocks.Block) error {
|
|
return es.s.PutMany(ctx, blks)
|
|
}
|
|
|
|
func (es *exposedSplitStore) AllKeysChan(ctx context.Context) (<-chan cid.Cid, error) {
|
|
return es.s.AllKeysChan(ctx)
|
|
}
|
|
|
|
func (es *exposedSplitStore) HashOnRead(enabled bool) {}
|
|
|
|
func (es *exposedSplitStore) View(ctx context.Context, c cid.Cid, f func([]byte) error) error {
|
|
if isIdentiyCid(c) {
|
|
data, err := decodeIdentityCid(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return f(data)
|
|
}
|
|
|
|
err := es.s.hot.View(ctx, c, f)
|
|
switch err {
|
|
case bstore.ErrNotFound:
|
|
return es.s.cold.View(ctx, c, f)
|
|
|
|
default:
|
|
return err
|
|
}
|
|
}
|