implement database.StateAccount()

This commit is contained in:
i-norden 2023-02-28 11:43:07 -06:00
parent 16b0b5b0f2
commit 8754278d18
2 changed files with 35 additions and 2 deletions

View File

@ -3,6 +3,8 @@ package ipld_eth_statedb
import (
"context"
"errors"
"github.com/ethereum/go-ethereum/crypto"
"math/big"
"github.com/VictoriaMetrics/fastcache"
"github.com/ethereum/go-ethereum/common"
@ -81,7 +83,23 @@ func (sd *stateDatabase) ContractCodeSize(_, codeHash common.Hash) (int, error)
}
func (sd *stateDatabase) StateAccount(address common.Address) (*types.StateAccount, error) {
panic("implement me")
res := StateAccountResult{}
key := crypto.Keccak256Hash(address.Bytes())
if err := sd.pgdb.QueryRow(context.Background(), GetStateAccount, key.Hex()).Scan(&res); err != nil {
return nil, errNotFound
}
if res.Removed {
// TODO: check expected behavior for deleted/non existing accounts
return nil, nil
}
bal := new(big.Int)
bal.SetString(res.Balance, 10)
return &types.StateAccount{
Nonce: res.Nonce,
Balance: bal,
Root: common.HexToHash(res.StorageRoot),
CodeHash: res.CodeHash,
}, nil
}
func (sd *stateDatabase) StorageSlot(addressHash, slotHash common.Hash) ([]byte, error) {

17
sql.go
View File

@ -14,7 +14,7 @@ const (
AND header_cids.block_hash = (SELECT canonical_header_hash(header_cids.block_number))
ORDER BY header_cids.block_number DESC
LIMIT 1`
GetStorageSlot = `SELECT value, removed FROM eth.storage_cids
GetStorageSlot = `SELECT val, removed FROM eth.storage_cids
INNER JOIN eth.header_cids ON (
storage_cids.header_id = header_cids.block_hash
AND storage_cids.block_number = header_cids.block_number
@ -25,3 +25,18 @@ const (
ORDER BY header_cids.block_number DESC
LIMIT 1`
)
// StorageSlotResult struct for unpacking GetStorageSlot result
type StorageSlotResult struct {
Value string `db:"val"`
Removed bool `db:"removed"`
}
// StateAccountResult struct for unpacking GetStateAccount result
type StateAccountResult struct {
Balance string `db:"balance"`
Nonce uint64 `db:"nonce"`
CodeHash []byte `db:"code_hash"`
StorageRoot string `db:"storage_root"`
Removed bool `db:"removed"`
}