ipld-eth-server/pkg/repositories/in_memory.go

104 lines
2.7 KiB
Go
Raw Normal View History

2017-11-06 16:42:33 +00:00
package repositories
import (
"fmt"
2017-11-06 18:53:43 +00:00
"github.com/8thlight/vulcanizedb/pkg/core"
2017-11-06 16:42:33 +00:00
)
type InMemory struct {
blocks map[int64]core.Block
contracts map[string]core.Contract
logs map[string][]core.Log
}
func (repository *InMemory) CreateLogs(logs []core.Log) error {
for _, log := range logs {
key := fmt.Sprintf("%s%s", log.BlockNumber, log.Index)
var logs []core.Log
repository.logs[key] = append(logs, log)
}
return nil
}
func (repository *InMemory) FindLogs(address string, blockNumber int64) []core.Log {
var matchingLogs []core.Log
for _, logs := range repository.logs {
for _, log := range logs {
if log.Address == address && log.BlockNumber == blockNumber {
matchingLogs = append(matchingLogs, log)
}
}
}
return matchingLogs
}
2017-12-04 22:54:35 +00:00
func (repository *InMemory) CreateContract(contract core.Contract) error {
repository.contracts[contract.Hash] = contract
return nil
}
2017-12-04 22:54:35 +00:00
func (repository *InMemory) ContractExists(contractHash string) bool {
_, present := repository.contracts[contractHash]
return present
2017-11-06 16:42:33 +00:00
}
func (repository *InMemory) FindContract(contractHash string) (core.Contract, error) {
2017-12-04 22:54:35 +00:00
contract, ok := repository.contracts[contractHash]
if !ok {
return core.Contract{}, ErrContractDoesNotExist(contractHash)
}
for _, block := range repository.blocks {
for _, transaction := range block.Transactions {
if transaction.To == contractHash {
2017-12-04 22:54:35 +00:00
contract.Transactions = append(contract.Transactions, transaction)
}
}
}
return contract, nil
}
func (repository *InMemory) MissingBlockNumbers(startingBlockNumber int64, endingBlockNumber int64) []int64 {
missingNumbers := []int64{}
for blockNumber := int64(startingBlockNumber); blockNumber <= endingBlockNumber; blockNumber++ {
if _, ok := repository.blocks[blockNumber]; !ok {
missingNumbers = append(missingNumbers, blockNumber)
}
}
return missingNumbers
}
2017-11-06 16:42:33 +00:00
func NewInMemory() *InMemory {
return &InMemory{
blocks: make(map[int64]core.Block),
contracts: make(map[string]core.Contract),
logs: make(map[string][]core.Log),
2017-11-06 16:42:33 +00:00
}
}
func (repository *InMemory) CreateBlock(block core.Block) error {
repository.blocks[block.Number] = block
return nil
2017-11-06 16:42:33 +00:00
}
func (repository *InMemory) BlockCount() int {
return len(repository.blocks)
}
func (repository *InMemory) FindBlockByNumber(blockNumber int64) (core.Block, error) {
if block, ok := repository.blocks[blockNumber]; ok {
return block, nil
}
return core.Block{}, ErrBlockDoesNotExist(blockNumber)
2017-11-06 16:42:33 +00:00
}
func (repository *InMemory) MaxBlockNumber() int64 {
highestBlockNumber := int64(-1)
for key := range repository.blocks {
if key > highestBlockNumber {
highestBlockNumber = key
}
}
return highestBlockNumber
}