load genesis state

cudoventures · Sep 16, 2021 · c2f4158 · c2f4158
1 parent 0b68110
commit c2f4158
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 0 deletions.
diff --git a/local-run.sh b/local-run.sh
@@ -0,0 +1,17 @@
+ADDRESS_PREFIX="cudos"
+FEES="100acudos"
+GRPC="http://localhost:9090"
+ETHRPC="http://104.198.157.197:8545"
+CONTRACT_ADDR="0x9fdE6D55dDa637806DbF016a03B6970613630333"
+COSMOS_ORCH_MNEMONIC="pencil spirit also middle brave celery obtain merge hurt rocket slice damp account actor fire first science organ charge ring vessel square extra general"
+ETH_PRIV_KEY_HEX="ed1eb2a53e4f15d2d60009b6b4be1d999cb0f64712678d09de972ce681e82882"
+
+cd ./orchestrator && cargo build && cp ./target/debug/gbt /usr/local/bin/gbt
+
+gbt --address-prefix="$ADDRESS_PREFIX" orchestrator \
+    --fees="$FEES" \
+    --cosmos-grpc="$GRPC" \
+    --ethereum-rpc="$ETHRPC" \
+    --gravity-contract-address="$CONTRACT_ADDR" \
+    --ethereum-key="${ETH_PRIV_KEY_HEX}" \
+    --cosmos-phrase="$COSMOS_ORCH_MNEMONIC"
diff --git a/module/x/gravity/keeper/genesis.go b/module/x/gravity/keeper/genesis.go
@@ -8,6 +8,9 @@ import (
 
 // InitGenesis starts a chain from a genesis state
 func InitGenesis(ctx sdk.Context, k Keeper, data types.GenesisState) {
+	var lastTxPoolId uint64 = 1
+	var lastBatchNonce uint64 = 1
+
 	k.SetParams(ctx, *data.Params)
 	// reset valsets in state
 	for _, vs := range data.Valsets {
@@ -24,7 +27,19 @@ func InitGenesis(ctx sdk.Context, k Keeper, data types.GenesisState) {
 	for _, batch := range data.Batches {
 		// TODO: block height?
 		k.StoreBatchUnsafe(ctx, batch)
+		if batch.BatchNonce > lastBatchNonce {
+			lastBatchNonce = batch.BatchNonce
+		}
+		for _, tx := range batch.Transactions {
+			if err := k.setPoolEntry(ctx, tx); err != nil {
+				panic(err)
+			}
+			if tx.Id > lastTxPoolId {
+				lastTxPoolId = tx.Id
+			}
+		}
 	}
+	k.setIncrementID(ctx, types.KeyLastOutgoingBatchID, lastTxPoolId+1)
 
 	// reset batch confirmations in state
 	for _, conf := range data.BatchConfirms {
@@ -48,8 +63,14 @@ func InitGenesis(ctx sdk.Context, k Keeper, data types.GenesisState) {
 		if err := k.setPoolEntry(ctx, tx); err != nil {
 			panic(err)
 		}
+		k.appendToUnbatchedTXIndex(ctx, tx.Erc20Fee.Contract, *tx.Erc20Fee, tx.Id)
+		if tx.Id > lastTxPoolId {
+			lastTxPoolId = tx.Id
+		}
 	}
 
+	k.setIncrementID(ctx, types.KeyLastTXPoolID, lastTxPoolId+1)
+
 	// reset attestations in state
 	for _, att := range data.Attestations {
 		att := att

diff --git a/module/x/gravity/keeper/pool.go b/module/x/gravity/keeper/pool.go
@@ -381,3 +381,10 @@ func (k Keeper) autoIncrementID(ctx sdk.Context, idKey []byte) uint64 {
 	store.Set(idKey, bz)
 	return id
 }
+
+func (k Keeper) setIncrementID(ctx sdk.Context, idKey []byte, id uint64) uint64 {
+	store := ctx.KVStore(k.storeKey)
+	bz := sdk.Uint64ToBigEndian(id)
+	store.Set(idKey, bz)
+	return id
+}