56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package store
|
|
|
|
import (
|
|
"github.com/ipfs/go-cid"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
// FullTipSet is an expanded version of the TipSet that contains all the blocks and messages
|
|
type FullTipSet struct {
|
|
Blocks []*types.FullBlock
|
|
tipset *types.TipSet
|
|
cids []cid.Cid
|
|
}
|
|
|
|
func NewFullTipSet(blks []*types.FullBlock) *FullTipSet {
|
|
return &FullTipSet{
|
|
Blocks: blks,
|
|
}
|
|
}
|
|
|
|
func (fts *FullTipSet) Cids() []cid.Cid {
|
|
if fts.cids != nil {
|
|
return fts.cids
|
|
}
|
|
|
|
var cids []cid.Cid
|
|
for _, b := range fts.Blocks {
|
|
cids = append(cids, b.Cid())
|
|
}
|
|
fts.cids = cids
|
|
|
|
return cids
|
|
}
|
|
|
|
// TipSet returns a narrower view of this FullTipSet elliding the block
|
|
// messages.
|
|
func (fts *FullTipSet) TipSet() *types.TipSet {
|
|
if fts.tipset != nil {
|
|
// FIXME: fts.tipset is actually never set. Should it memoize?
|
|
return fts.tipset
|
|
}
|
|
|
|
var headers []*types.BlockHeader
|
|
for _, b := range fts.Blocks {
|
|
headers = append(headers, b.Header)
|
|
}
|
|
|
|
ts, err := types.NewTipSet(headers)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return ts
|
|
}
|