constant refactor

crypto-org-chain · Oct 31, 2023 · b65f2b1 · b65f2b1
1 parent 3a4335c
commit b65f2b1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/x/cronos/keeper/keeper.go b/x/cronos/keeper/keeper.go
@@ -304,7 +304,7 @@ func (k Keeper) onPacketResult(
 		return err
 	}
 
-	if stateDB, ok := ctx.Value("statedb").(vm.StateDB); ok {
+	if stateDB, ok := ctx.Value(types.StateDBContextKey).(vm.StateDB); ok {
 		for _, l := range rsp.Logs {
 			stateDB.AddLog(l.ToEthereum())
 		}

diff --git a/x/cronos/keeper/precompiles/utils.go b/x/cronos/keeper/precompiles/utils.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/crypto-org-chain/cronos/v2/x/cronos/types"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/evmos/ethermint/x/evm/statedb"
 )
@@ -45,7 +46,7 @@ func exec[Req any, PReq interface {
 	var res Resp
 	if err := stateDB.ExecuteNativeAction(contract, converter, func(ctx sdk.Context) error {
 		var err error
-		ctx = ctx.WithValue("statedb", stateDB)
+		ctx = ctx.WithValue(types.StateDBContextKey, stateDB)
 		res, err = action(ctx, msg)
 		return err
 	}); err != nil {

diff --git a/x/cronos/types/keys.go b/x/cronos/types/keys.go
@@ -27,6 +27,8 @@ const (
 	prefixContractToDenom
 	paramsKey
 	prefixAdminToPermissions
+
+	StateDBContextKey = "statedb"
 )
 
 // KVStore key prefixes