118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package state
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/lib/blockstore"
|
|
|
|
"github.com/filecoin-project/specs-actors/actors/util/adt"
|
|
|
|
blocks "github.com/ipfs/go-block-format"
|
|
"github.com/ipfs/go-blockservice"
|
|
"github.com/ipfs/go-cid"
|
|
ds "github.com/ipfs/go-datastore"
|
|
exchange "github.com/ipfs/go-ipfs-exchange-interface"
|
|
offline "github.com/ipfs/go-ipfs-exchange-offline"
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
|
format "github.com/ipfs/go-ipld-format"
|
|
"github.com/ipfs/go-merkledag"
|
|
)
|
|
|
|
// Stores is a collection of the different stores and services that are needed
|
|
// to deal with the data layer of Filecoin, conveniently interlinked with one
|
|
// another.
|
|
type Stores struct {
|
|
CBORStore cbor.IpldStore
|
|
ADTStore adt.Store
|
|
Datastore ds.Batching
|
|
Blockstore blockstore.Blockstore
|
|
BlockService blockservice.BlockService
|
|
Exchange exchange.Interface
|
|
DAGService format.DAGService
|
|
}
|
|
|
|
func newStores(ctx context.Context, ds ds.Batching, bs blockstore.Blockstore) *Stores {
|
|
var (
|
|
cborstore = cbor.NewCborStore(bs)
|
|
offl = offline.Exchange(bs)
|
|
blkserv = blockservice.New(bs, offl)
|
|
dserv = merkledag.NewDAGService(blkserv)
|
|
)
|
|
|
|
return &Stores{
|
|
CBORStore: cborstore,
|
|
ADTStore: adt.WrapStore(ctx, cborstore),
|
|
Datastore: ds,
|
|
Blockstore: bs,
|
|
Exchange: offl,
|
|
BlockService: blkserv,
|
|
DAGService: dserv,
|
|
}
|
|
}
|
|
|
|
// NewLocalStores creates a Stores object that operates entirely in-memory with
|
|
// no read-through remote fetch fallback.
|
|
func NewLocalStores(ctx context.Context) *Stores {
|
|
ds := ds.NewMapDatastore()
|
|
bs := blockstore.NewBlockstore(ds)
|
|
return newStores(ctx, ds, bs)
|
|
}
|
|
|
|
type proxyingBlockstore struct {
|
|
ctx context.Context
|
|
api api.FullNode
|
|
|
|
online bool
|
|
lock sync.RWMutex
|
|
blockstore.Blockstore
|
|
}
|
|
|
|
func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
|
|
pb.lock.RLock()
|
|
if block, err := pb.Blockstore.Get(cid); (err == nil || !pb.online) {
|
|
pb.lock.RUnlock();
|
|
return block, err
|
|
}
|
|
pb.lock.RUnlock();
|
|
|
|
// fmt.Printf("fetching cid via rpc: %v\n", cid)
|
|
item, err := pb.api.ChainReadObj(pb.ctx, cid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
block, err := blocks.NewBlockWithCid(item, cid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pb.lock.Lock();
|
|
defer pb.lock.Unlock();
|
|
err = pb.Blockstore.Put(block)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return block, nil
|
|
}
|
|
|
|
func (pb *proxyingBlockstore) SetOnline(online bool) {
|
|
pb.online = online
|
|
}
|
|
|
|
// NewProxyingStore is a Stores that proxies get requests for unknown CIDs
|
|
// to a Filecoin node, via the ChainReadObj RPC.
|
|
func NewProxyingStore(ctx context.Context, api api.FullNode) *Stores {
|
|
ds := ds.NewMapDatastore()
|
|
|
|
bs := &proxyingBlockstore{
|
|
ctx: ctx,
|
|
api: api,
|
|
online: true,
|
|
Blockstore: blockstore.NewBlockstore(ds),
|
|
}
|
|
|
|
return newStores(ctx, ds, bs)
|
|
}
|