Handle Errs on Genesis Init

strangelove-ventures · Sep 9, 2024 · 91843a7 · 91843a7
1 parent 1c952ab
commit 91843a7
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/x/staking/genesis.go b/x/staking/genesis.go
@@ -43,12 +43,16 @@ func InitGenesis(
 		keeper.SetValidator(ctx, validator)
 
 		// Manually set indices for the first time
-		keeper.SetValidatorByConsAddr(ctx, validator)
+		if err := keeper.SetValidatorByConsAddr(ctx, validator); err != nil {
+			panic(err)
+		}
 		keeper.SetValidatorByPowerIndex(ctx, validator)
 
 		// Call the creation hook if not exported
 		if !data.Exported {
-			keeper.Hooks().AfterValidatorCreated(ctx, validator.GetOperator())
+			if err := keeper.Hooks().AfterValidatorCreated(ctx, validator.GetOperator()); err != nil {
+				panic(err)
+			}
 		}
 
 		// update timeslice if necessary
@@ -71,13 +75,17 @@ func InitGenesis(
 
 		// Call the before-creation hook if not exported
 		if !data.Exported {
-			keeper.Hooks().BeforeDelegationCreated(ctx, delegatorAddress, delegation.GetValidatorAddr())
+			if err := keeper.Hooks().BeforeDelegationCreated(ctx, delegatorAddress, delegation.GetValidatorAddr()); err != nil {
+				panic(err)
+			}
 		}
 
 		keeper.SetDelegation(ctx, delegation)
 		// Call the after-modification hook if not exported
 		if !data.Exported {
-			keeper.Hooks().AfterDelegationModified(ctx, delegatorAddress, delegation.GetValidatorAddr())
+			if err := keeper.Hooks().AfterDelegationModified(ctx, delegatorAddress, delegation.GetValidatorAddr()); err != nil {
+				panic(err)
+			}
 		}
 	}