fix: remove validator info after hook

terra-money · Jul 21, 2023 · 1cdf4e3 · 1cdf4e3
1 parent a1bc98f
commit 1cdf4e3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/x/alliance/keeper/hooks.go b/x/alliance/keeper/hooks.go
@@ -19,7 +19,8 @@ func (h Hooks) BeforeValidatorModified(_ sdk.Context, _ sdk.ValAddress) error {
 	return nil
 }
 
-func (h Hooks) AfterValidatorRemoved(ctx sdk.Context, _ sdk.ConsAddress, _ sdk.ValAddress) error {
+func (h Hooks) AfterValidatorRemoved(ctx sdk.Context, _ sdk.ConsAddress, valAddr sdk.ValAddress) error {
+	h.k.DeleteValidatorInfo(ctx, valAddr)
 	h.k.QueueAssetRebalanceEvent(ctx)
 	return nil
 }

diff --git a/x/alliance/keeper/validator.go b/x/alliance/keeper/validator.go
@@ -72,3 +72,9 @@ func (k Keeper) GetAllAllianceValidatorInfo(ctx sdk.Context) []types.AllianceVal
 	}
 	return infos
 }
+
+func (k Keeper) DeleteValidatorInfo(ctx sdk.Context, valAddr sdk.ValAddress) {
+	store := ctx.KVStore(k.storeKey)
+	key := types.GetAllianceValidatorInfoKey(valAddr)
+	store.Delete(key)
+}
diff --git a/x/alliance/tests/benchmark/benchmark_genesis.json b/x/alliance/tests/benchmark/benchmark_genesis.json