make event emission functions methods on keeper

cosmos · Jan 27, 2025 · 0d8f78b · 0d8f78b
1 parent 44b3ce1
commit 0d8f78b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/modules/light-clients/08-wasm/keeper/events.go b/modules/light-clients/08-wasm/keeper/events.go
@@ -1,6 +1,7 @@
 package keeper
 
 import (
+	"context"
 	"encoding/hex"
 	"errors"
 
@@ -12,7 +13,8 @@ import (
 )
 
 // emitStoreWasmCodeEvent emits a store wasm code event
-func emitStoreWasmCodeEvent(em event.Manager, checksum types.Checksum) error {
+func (k Keeper) emitStoreWasmCodeEvent(ctx context.Context, checksum types.Checksum) error {
+	em := k.EventService.EventManager(ctx)
 	return errors.Join(
 		em.EmitKV(
 			types.EventTypeStoreWasmCode,
@@ -26,7 +28,8 @@ func emitStoreWasmCodeEvent(em event.Manager, checksum types.Checksum) error {
 }
 
 // emitMigrateContractEvent emits a migrate contract event
-func emitMigrateContractEvent(em event.Manager, clientID string, checksum, newChecksum types.Checksum) error {
+func (k Keeper) emitMigrateContractEvent(ctx context.Context, clientID string, checksum, newChecksum types.Checksum) error {
+	em := k.EventService.EventManager(ctx)
 	return errors.Join(
 		em.EmitKV(
 			types.EventTypeMigrateContract,

diff --git a/modules/light-clients/08-wasm/keeper/keeper.go b/modules/light-clients/08-wasm/keeper/keeper.go
@@ -177,7 +177,7 @@ func (k Keeper) migrateContractCode(ctx context.Context, clientID string, newChe
 
 	k.clientKeeper.SetClientState(ctx, clientID, wasmClientState)
 
-	if err = emitMigrateContractEvent(k.EventService.EventManager(ctx), clientID, oldChecksum, newChecksum); err != nil {
+	if err = k.emitMigrateContractEvent(ctx, clientID, oldChecksum, newChecksum); err != nil {
 		return fmt.Errorf("failed to emit migrate contract events: %w", err)
 	}
 

diff --git a/modules/light-clients/08-wasm/keeper/msg_server.go b/modules/light-clients/08-wasm/keeper/msg_server.go
@@ -27,7 +27,7 @@ func (k Keeper) StoreCode(goCtx context.Context, msg *types.MsgStoreCode) (*type
 		return nil, errorsmod.Wrap(err, "failed to store wasm bytecode")
 	}
 
-	if err := emitStoreWasmCodeEvent(k.EventService.EventManager(ctx), checksum); err != nil {
+	if err := k.emitStoreWasmCodeEvent(ctx, checksum); err != nil {
 		return nil, fmt.Errorf("failed to emit store wasm code events: %w", err)
 	}