Implement consumer removal

cosmos · Oct 25, 2022 · 4830e7e · 4830e7e
1 parent 8ef0e58
commit 4830e7e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/x/ccv/provider/keeper/keymap.go b/x/ccv/provider/keeper/keymap.go
@@ -564,6 +564,21 @@ func (s *KeyMapStore) IterateCcaToLastUpdateMemo(cb func(ConsumerConsAddr, ccvty
 	}
 }
 
+func (k Keeper) DeleteKeyMap(ctx sdk.Context, chainID string) {
+	store := ctx.KVStore(k.storeKey)
+	for _, pref := range [][]byte{
+		types.KeyMapPkToCkChainPrefix(chainID),
+		types.KeyMapCkToPkChainPrefix(chainID),
+		types.KeyMapCkToMemoChainPrefix(chainID),
+	} {
+		iter := sdk.KVStorePrefixIterator(store, pref)
+		defer iter.Close()
+		for ; iter.Valid(); iter.Next() {
+			store.Delete(iter.Key())
+		}
+	}
+}
+
 func (k Keeper) KeyMap(ctx sdk.Context, chainID string) *KeyMap {
 	store := KeyMapStore{ctx.KVStore(k.storeKey), chainID}
 	km := MakeKeyMap(&store)

diff --git a/x/ccv/provider/keeper/proposal.go b/x/ccv/provider/keeper/proposal.go
@@ -124,6 +124,7 @@ func (k Keeper) StopConsumerChain(ctx sdk.Context, chainID string, lockUbd, clos
 	// clean up states
 	k.DeleteConsumerClientId(ctx, chainID)
 	k.DeleteConsumerGenesis(ctx, chainID)
+	k.DeleteKeyMap(ctx, chainID)
 	k.DeleteLockUnbondingOnTimeout(ctx, chainID)
 
 	// close channel and delete the mappings between chain ID and channel ID