52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
|
package inmemory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/core"
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/repositories"
|
||
|
|
)
|
||
|
|
|
||
|
|
type BlockRepository struct {
|
||
|
|
*InMemory
|
||
|
|
}
|
||
|
|
|
||
|
|
func (blockRepository *BlockRepository) CreateOrUpdateBlock(block core.Block) error {
|
||
|
|
blockRepository.CreateOrUpdateBlockCallCount++
|
||
|
|
blockRepository.blocks[block.Number] = block
|
||
|
|
for _, transaction := range block.Transactions {
|
||
|
|
blockRepository.receipts[transaction.Hash] = transaction.Receipt
|
||
|
|
blockRepository.logs[transaction.TxHash] = transaction.Logs
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (blockRepository *BlockRepository) GetBlock(blockNumber int64) (core.Block, error) {
|
||
|
|
if block, ok := blockRepository.blocks[blockNumber]; ok {
|
||
|
|
return block, nil
|
||
|
|
}
|
||
|
|
return core.Block{}, repositories.ErrBlockDoesNotExist(blockNumber)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (blockRepository *BlockRepository) MissingBlockNumbers(startingBlockNumber int64, endingBlockNumber int64) []int64 {
|
||
|
|
missingNumbers := []int64{}
|
||
|
|
for blockNumber := int64(startingBlockNumber); blockNumber <= endingBlockNumber; blockNumber++ {
|
||
|
|
if _, ok := blockRepository.blocks[blockNumber]; !ok {
|
||
|
|
missingNumbers = append(missingNumbers, blockNumber)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return missingNumbers
|
||
|
|
}
|
||
|
|
|
||
|
|
func (blockRepository *BlockRepository) SetBlocksStatus(chainHead int64) {
|
||
|
|
for key, block := range blockRepository.blocks {
|
||
|
|
if key < (chainHead - blocksFromHeadBeforeFinal) {
|
||
|
|
tmp := block
|
||
|
|
tmp.IsFinal = true
|
||
|
|
blockRepository.blocks[key] = tmp
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (blockRepository *BlockRepository) BlockCount() int {
|
||
|
|
return len(blockRepository.blocks)
|
||
|
|
}
|