Fix:module name in Milestone logger

llamanodes · Sep 11, 2023 · 00fbcab · 00fbcab
1 parent 846f292
commit 00fbcab
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/checkpoint/handler_milestone.go b/checkpoint/handler_milestone.go
@@ -14,7 +14,7 @@ import (
 
 // handleMsgMilestone validates milestone transaction
 func handleMsgMilestone(ctx sdk.Context, msg types.MsgMilestone, k Keeper) sdk.Result {
-	logger := k.Logger(ctx)
+	logger := k.MilestoneLogger(ctx)
 	milestoneLength := helper.MilestoneLength
 
 	//
@@ -114,7 +114,7 @@ func handleMsgMilestone(ctx sdk.Context, msg types.MsgMilestone, k Keeper) sdk.R
 
 // Handles milestone timeout transaction
 func handleMsgMilestoneTimeout(ctx sdk.Context, _ types.MsgMilestoneTimeout, k Keeper) sdk.Result {
-	logger := k.Logger(ctx)
+	logger := k.MilestoneLogger(ctx)
 
 	// Get current block time
 	currentTime := ctx.BlockTime()

diff --git a/checkpoint/keeper_milestone.go b/checkpoint/keeper_milestone.go
@@ -4,6 +4,7 @@ import (
 	"strconv"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/tendermint/tendermint/libs/log"
 
 	cmn "github.com/maticnetwork/heimdall/common"
 	"github.com/maticnetwork/heimdall/helper"
@@ -19,6 +20,11 @@ var (
 	BlockNumberKey        = []byte{0x70} //Key to store the count
 )
 
+// Logger returns a module-specific logger
+func (k Keeper) MilestoneLogger(ctx sdk.Context) log.Logger {
+	return ctx.Logger().With("module", "Milestone")
+}
+
 // AddMilestone adds milestone in the store
 func (k *Keeper) AddMilestone(ctx sdk.Context, milestone hmTypes.Milestone) error {
 	milestoneNumber := k.GetMilestoneCount(ctx) + 1 //GetCount gives the number of previous milestone

diff --git a/checkpoint/side_milestone_handler.go b/checkpoint/side_milestone_handler.go
@@ -20,7 +20,7 @@ func SideHandleMsgMilestone(ctx sdk.Context, k Keeper, msg types.MsgMilestone, c
 	milestoneLength := helper.MilestoneLength
 
 	// logger
-	logger := k.Logger(ctx)
+	logger := k.MilestoneLogger(ctx)
 
 	//Get the milestone count
 	count := k.GetMilestoneCount(ctx)
@@ -72,7 +72,7 @@ func SideHandleMsgMilestone(ctx sdk.Context, k Keeper, msg types.MsgMilestone, c
 
 // PostHandleMsgMilestone handles msg milestone
 func PostHandleMsgMilestone(ctx sdk.Context, k Keeper, msg types.MsgMilestone, sideTxResult abci.SideTxResultType) sdk.Result {
-	logger := k.Logger(ctx)
+	logger := k.MilestoneLogger(ctx)
 	timeStamp := uint64(ctx.BlockTime().Unix())
 
 	// TX bytes