mv "SetHeightHash" fron endBlock to beginBlock

x/evm/keeper/abci.go

@@ -1,15 +1,14 @@
 package keeper
 
 import (
+	"github.com/ethereum/go-ethereum/common"
 	"math/big"
 
 	abci "github.com/tendermint/tendermint/abci/types"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	ethtypes "github.com/ethereum/go-ethereum/core/types"
-
-	"github.com/okex/okexchain/x/evm/types"
 )
 
 // BeginBlock sets the block hash -> block height map for the previous block height
@@ -22,7 +21,12 @@ func (k *Keeper) BeginBlock(ctx sdk.Context, req abci.RequestBeginBlock) {
 	// Gas costs are handled within msg handler so costs should be ignored
 	ctx = ctx.WithGasMeter(sdk.NewInfiniteGasMeter())
 
-	k.SetBlockHash(ctx, req.Header.LastBlockId.GetHash(), req.Header.GetHeight()-1)
+	// Set the hash -> height and height -> hash mapping.
+	hash := req.Header.LastBlockId.GetHash()
+	height := req.Header.GetHeight() - 1
+
+	k.SetHeightHash(ctx, uint64(height), common.BytesToHash(hash))
+	k.SetBlockHash(ctx, hash, height)
 
 	// reset counters that are used on CommitStateDB.Prepare
 	k.Bloom = big.NewInt(0)
@@ -37,13 +41,6 @@ func (k Keeper) EndBlock(ctx sdk.Context, req abci.RequestEndBlock) []abci.Valid
 	// Gas costs are handled within msg handler so costs should be ignored
 	ctx = ctx.WithGasMeter(sdk.NewInfiniteGasMeter())
 
-	// Set the hash for the current height.
-	// NOTE: we set the hash here instead of on BeginBlock in order to set the final block prior to
-	// an upgrade. If we set it on BeginBlock the last block from prior to the upgrade wouldn't be
-	// included on the store.
-	hash := types.HashFromContext(ctx)
-	k.SetHeightHash(ctx, uint64(ctx.BlockHeight()), hash)
-
 	// Update account balances before committing other parts of state
 	k.UpdateAccounts(ctx)
 

x/evm/types/statedb.go

@@ -32,7 +32,7 @@ type revision struct {
 
 // CommitStateDB implements the Geth state.StateDB interface. Instead of using
 // a trie and database for querying and persistence, the Keeper uses KVStores
-// and an account mapper is used to facilitate state transitions.
+// and an AccountKeeper to facilitate state transitions.
 //
 // TODO: This implementation is subject to change in regards to its statefull
 // manner. In otherwords, how this relates to the keeper in this module.