Add kickMetaNodeVotePool to genesis file

stratosnet · Mar 12, 2024 · e2f3de2 · e2f3de2
1 parent a29695c
commit e2f3de2
Show file tree

Hide file tree

Showing 6 changed files with 457 additions and 197 deletions.
diff --git a/api/stratos/register/v1/genesis.pulsar.go b/api/stratos/register/v1/genesis.pulsar.go
diff --git a/proto/stratos/register/v1/genesis.proto b/proto/stratos/register/v1/genesis.proto
@@ -68,6 +68,12 @@ message GenesisState {
     (gogoproto.jsontag) = "unbonding_nodes",
     (gogoproto.moretags) = "yaml:\"unbonding_nodes\""
   ];
+  repeated KickMetaNodeVotePool kick_meta_node_vote_pool = 9 [
+    (gogoproto.nullable) = false,
+    (amino.dont_omitempty) = true,
+    (gogoproto.jsontag) = "kick_meta_node_vote_pool",
+    (gogoproto.moretags) = "yaml:\"kick_meta_node_vote_pool\""
+  ];
 }
 
 message GenesisMetaNode {

diff --git a/x/register/keeper/genesis.go b/x/register/keeper/genesis.go
@@ -157,6 +157,10 @@ func (k Keeper) InitGenesis(ctx sdk.Context, data *types.GenesisState) {
 		}
 	}
 
+	for _, kickMetaNodeVoteInfo := range data.GetKickMetaNodeVotePool() {
+		k.SetKickMetaNodeVotePool(ctx, kickMetaNodeVoteInfo)
+	}
+
 	return
 }
 
@@ -182,6 +186,8 @@ func (k Keeper) ExportGenesis(ctx sdk.Context) (data *types.GenesisState) {
 		return false
 	})
 
+	kickMetaNodeVotePool := k.GetAllKickMetaNodeVotePool(ctx)
+
 	return types.NewGenesisState(params, resourceNodes, metaNodes, remainingNozLimit, slashingInfo, depositNozRate,
-		metaNodeRegVotePool, unbondingNodes)
+		metaNodeRegVotePool, unbondingNodes, kickMetaNodeVotePool)
 }
diff --git a/x/register/keeper/store.go b/x/register/keeper/store.go
@@ -241,6 +241,19 @@ func (k Keeper) GetKickMetaNodeVotePool(ctx sdk.Context, targetNetworkAddr strat
 	return votePool, true
 }
 
+func (k Keeper) GetAllKickMetaNodeVotePool(ctx sdk.Context) (votePools []types.KickMetaNodeVotePool) {
+	store := ctx.KVStore(k.storeKey)
+	iterator := sdk.KVStorePrefixIterator(store, types.KickMetaNodeVotesKey)
+	defer iterator.Close()
+
+	for ; iterator.Valid(); iterator.Next() {
+		voteInfo := types.KickMetaNodeVotePool{}
+		k.cdc.MustUnmarshalLengthPrefixed(iterator.Value(), &voteInfo)
+		votePools = append(votePools, voteInfo)
+	}
+	return
+}
+
 func (k Keeper) GetAllExpiredKickMetaNodeVotePool(ctx sdk.Context) (votePools []types.KickMetaNodeVotePool) {
 	store := ctx.KVStore(k.storeKey)
 	iterator := sdk.KVStorePrefixIterator(store, types.KickMetaNodeVotesKey)

diff --git a/x/register/types/genesis.go b/x/register/types/genesis.go
@@ -21,30 +21,33 @@ func NewGenesisState(params Params,
 	depositNozRate sdkmath.LegacyDec,
 	metaNodeRegVotePool []MetaNodeRegistrationVotePool,
 	unbondingNodes []UnbondingNode,
+	kickMetaNodeVotePool []KickMetaNodeVotePool,
 ) *GenesisState {
 	return &GenesisState{
-		Params:              params,
-		ResourceNodes:       resourceNodes,
-		MetaNodes:           metaNodes,
-		RemainingNozLimit:   remainingNozLimit,
-		Slashing:            slashingInfo,
-		DepositNozRate:      depositNozRate,
-		MetaNodeRegVotePool: metaNodeRegVotePool,
-		UnbondingNodes:      unbondingNodes,
+		Params:               params,
+		ResourceNodes:        resourceNodes,
+		MetaNodes:            metaNodes,
+		RemainingNozLimit:    remainingNozLimit,
+		Slashing:             slashingInfo,
+		DepositNozRate:       depositNozRate,
+		MetaNodeRegVotePool:  metaNodeRegVotePool,
+		UnbondingNodes:       unbondingNodes,
+		KickMetaNodeVotePool: kickMetaNodeVotePool,
 	}
 }
 
 // DefaultGenesisState - default GenesisState used by Cosmos Hub
 func DefaultGenesisState() *GenesisState {
 	return &GenesisState{
-		Params:              DefaultParams(),
-		ResourceNodes:       ResourceNodes{},
-		MetaNodes:           MetaNodes{},
-		RemainingNozLimit:   DefaultRemainingNozLimit,
-		Slashing:            make([]Slashing, 0),
-		DepositNozRate:      DefaultDepositNozRate,
-		MetaNodeRegVotePool: make([]MetaNodeRegistrationVotePool, 0),
-		UnbondingNodes:      make([]UnbondingNode, 0),
+		Params:               DefaultParams(),
+		ResourceNodes:        ResourceNodes{},
+		MetaNodes:            MetaNodes{},
+		RemainingNozLimit:    DefaultRemainingNozLimit,
+		Slashing:             make([]Slashing, 0),
+		DepositNozRate:       DefaultDepositNozRate,
+		MetaNodeRegVotePool:  make([]MetaNodeRegistrationVotePool, 0),
+		UnbondingNodes:       make([]UnbondingNode, 0),
+		KickMetaNodeVotePool: make([]KickMetaNodeVotePool, 0),
 	}
 }