diff --git a/core/block_cache.go b/core/block_cache.go index ea39e78e84..768d3bf197 100644 --- a/core/block_cache.go +++ b/core/block_cache.go @@ -56,6 +56,23 @@ func (bc *BlockCache) Push(block *types.Block) { bc.hashes[len(bc.hashes)-1] = hash } +func (bc *BlockCache) Delete(hash common.Hash) { + bc.mu.Lock() + defer bc.mu.Unlock() + + if _, ok := bc.blocks[hash]; ok { + delete(bc.blocks, hash) + for i, h := range bc.hashes { + if hash == h { + bc.hashes = bc.hashes[:i+copy(bc.hashes[i:], bc.hashes[i+1:])] + // or ? => bc.hashes = append(bc.hashes[:i], bc.hashes[i+1]...) + + break + } + } + } +} + func (bc *BlockCache) Get(hash common.Hash) *types.Block { bc.mu.RLock() defer bc.mu.RUnlock() diff --git a/core/block_cache_test.go b/core/block_cache_test.go index d4f610b719..434b97792e 100644 --- a/core/block_cache_test.go +++ b/core/block_cache_test.go @@ -46,3 +46,15 @@ func TestInclusion(t *testing.T) { } } } + +func TestDeletion(t *testing.T) { + chain := newChain(3) + cache := NewBlockCache(3) + insertChainCache(cache, chain) + + cache.Delete(chain[1].Hash()) + + if cache.Has(chain[1].Hash()) { + t.Errorf("expected %x not to be included") + } +}