package chain import ( "fmt" lru "github.com/hashicorp/golang-lru" "github.com/ipfs/go-cid" "github.com/filecoin-project/lotus/build" ) type BadBlockCache struct { badBlocks *lru.ARCCache } type BadBlockReason struct { Reason string TipSet []cid.Cid OriginalReason *BadBlockReason } func NewBadBlockReason(cid []cid.Cid, format string, i ...interface{}) BadBlockReason { return BadBlockReason{ TipSet: cid, Reason: fmt.Sprintf(format, i...), } } func (bbr BadBlockReason) Linked(reason string, i ...interface{}) BadBlockReason { or := &bbr if bbr.OriginalReason != nil { or = bbr.OriginalReason } return BadBlockReason{Reason: fmt.Sprintf(reason, i...), OriginalReason: or} } func (bbr BadBlockReason) String() string { res := bbr.Reason if bbr.OriginalReason != nil { res += " caused by: " + fmt.Sprintf("%s %s", bbr.OriginalReason.TipSet, bbr.OriginalReason.String()) } return res } func NewBadBlockCache() *BadBlockCache { cache, err := lru.NewARC(build.BadBlockCacheSize) if err != nil { panic(err) // ok } return &BadBlockCache{ badBlocks: cache, } } func (bts *BadBlockCache) Add(c cid.Cid, bbr BadBlockReason) { bts.badBlocks.Add(c, bbr) } func (bts *BadBlockCache) Remove(c cid.Cid) { bts.badBlocks.Remove(c) } func (bts *BadBlockCache) Purge() { bts.badBlocks.Purge() } func (bts *BadBlockCache) Has(c cid.Cid) (BadBlockReason, bool) { rval, ok := bts.badBlocks.Get(c) if !ok { return BadBlockReason{}, false } return rval.(BadBlockReason), true }