feat(store/v2): build the migration manager in the root store factory (#22336)

Co-authored-by: marbar3778 <marbar3778@yahoo.com>
Co-authored-by: Marko <marko@baricevic.me>
Co-authored-by: Alex | Interchain Labs <alex@skip.money>

Author: 3 people

api/cosmos/store/v2/commit_info.pulsar.go

@@ -0,0 +1,34 @@
+syntax = "proto3";
+package cosmos.store.v2;
+
+import "gogoproto/gogo.proto";
+import "google/protobuf/timestamp.proto";
+
+option go_package = "cosmossdk.io/store/v2/proof";
+
+// CommitInfo defines commit information used by the multi-store when committing
+// a version/height.
+message CommitInfo {
+  int64                     version     = 1;
+  repeated StoreInfo        store_infos = 2;
+  google.protobuf.Timestamp timestamp   = 3 [(gogoproto.nullable) = false, (gogoproto.stdtime) = true];
+  bytes                     commit_hash = 4;
+}
+
+// StoreInfo defines store-specific commit information. It contains a reference
+// between a store name and the commit ID.
+//
+message StoreInfo {
+  string   name      = 1;
+  CommitID commit_id = 2;
+  string   structure = 3;
+}
+
+// CommitID defines the commitment information when a specific store is
+// committed.
+message CommitID {
+  option (gogoproto.goproto_stringer) = false;
+
+  int64 version = 1;
+  bytes hash    = 2;
+}

client/grpc/reflection/reflection.pb.go

@@ -11,15 +11,14 @@ $SIMD_BIN config set client keyring-backend test
 $SIMD_BIN config set client keyring-default-keyname alice
 $SIMD_BIN config set app rest.enable true
 $SIMD_BIN config set app telemetry.prometheus-retention-time 600
-$SIMD_BIN config set app store.options.sc-type iavl-v2
 sed -i '' 's/timeout_commit = "5s"/timeout_commit = "1s"/' "$SIMD_HOME"/config/config.toml
 sed -i '' 's/prometheus = false/prometheus = true/' "$SIMD_HOME"/config/config.toml
 
 $SIMD_BIN keys add alice --indiscreet
 $SIMD_BIN keys add bob --indiscreet
 aliases=""
 for i in $(seq 10); do
-    alias=$(dd if=/dev/urandom bs=16 count=24 2> /dev/null | base32 | head -c 32)
+    alias=$(dd if=/dev/urandom bs=16 count=24 2> /dev/null | base64 | head -c 32)
     $SIMD_BIN keys add "$alias" --indiscreet
     aliases="$aliases $alias"
 done

proto/cosmos/store/v2/commit_info.proto

@@ -913,7 +913,7 @@ func assertStoreLatestVersion(t *testing.T, store types.Store, target uint64) {
 	require.Equal(t, target, version)
 	commitInfo, err := store.GetStateCommitment().GetCommitInfo(version)
 	require.NoError(t, err)
-	require.Equal(t, target, commitInfo.Version)
+	require.Equal(t, target, uint64(commitInfo.Version))
 }
 
 func TestOptimisticExecution(t *testing.T) {

scripts/init-simapp-v2.sh

@@ -37,7 +37,7 @@ func (s *MockStore) GetLatestVersion() (uint64, error) {
 		return 0, err
 	}
 
-	return lastCommitID.Version, nil
+	return uint64(lastCommitID.Version), nil
 }
 
 func (s *MockStore) StateLatest() (uint64, corestore.ReaderMap, error) {
@@ -99,7 +99,7 @@ func (s *MockStore) LastCommitID() (proof.CommitID, error) {
 	v, err := s.GetStateCommitment().GetLatestVersion()
 	bz := sha256.Sum256([]byte{})
 	return proof.CommitID{
-		Version: v,
+		Version: int64(v),
 		Hash:    bz[:],
 	}, err
 }

server/v2/cometbft/abci_test.go

@@ -101,8 +101,15 @@ func (s *Server[T]) Start(ctx context.Context) error {
 		}()
 	}
 
-	if err := <-resCh; err != nil {
-		return fmt.Errorf("failed to start servers: %w", err)
+	for i := 0; i < len(s.components); i++ {
+		select {
+		case err := <-resCh:
+			if err != nil {
+				return fmt.Errorf("failed to start servers: %w", err)
+			}
+		case <-ctx.Done():
+			return nil
+		}
 	}
 
 	<-ctx.Done()

server/v2/cometbft/internal/mock/mock_store.go

@@ -37,8 +37,6 @@ import (
 	_ "cosmossdk.io/x/bank"         // import for side-effects
 	banktypes "cosmossdk.io/x/bank/types"
 	_ "cosmossdk.io/x/bank/v2" // import for side-effects
-	bankv2types "cosmossdk.io/x/bank/v2/types"
-	bankmodulev2 "cosmossdk.io/x/bank/v2/types/module"
 	_ "cosmossdk.io/x/circuit" // import for side-effects
 	circuittypes "cosmossdk.io/x/circuit/types"
 	_ "cosmossdk.io/x/consensus" // import for side-effects
@@ -156,7 +154,7 @@ var (
 						accounts.ModuleName,
 						authtypes.ModuleName,
 						banktypes.ModuleName,
-						bankv2types.ModuleName,
+						// bankv2types.ModuleName,
 						distrtypes.ModuleName,
 						stakingtypes.ModuleName,
 						slashingtypes.ModuleName,
@@ -297,10 +295,10 @@ var (
 				Name:   epochstypes.ModuleName,
 				Config: appconfig.WrapAny(&epochsmodulev1.Module{}),
 			},
-			{
-				Name:   bankv2types.ModuleName,
-				Config: appconfig.WrapAny(&bankmodulev2.Module{}),
-			},
+			// {
+			// 	Name:   bankv2types.ModuleName,
+			// 	Config: appconfig.WrapAny(&bankmodulev2.Module{}),
+			// },
 		},
 	}
 )

server/v2/server.go

@@ -26,7 +26,6 @@ import (
 	"cosmossdk.io/server/v2/cometbft"
 	"cosmossdk.io/server/v2/store"
 	banktypes "cosmossdk.io/x/bank/types"
-	bankv2types "cosmossdk.io/x/bank/v2/types"
 	stakingtypes "cosmossdk.io/x/staking/types"
 
 	"github.com/cosmos/cosmos-sdk/client"
@@ -423,13 +422,6 @@ func initGenFiles[T transaction.Tx](
 	}
 	appGenState[banktypes.ModuleName] = clientCtx.Codec.MustMarshalJSON(&bankGenState)
 
-	var bankV2GenState bankv2types.GenesisState
-	clientCtx.Codec.MustUnmarshalJSON(appGenState[bankv2types.ModuleName], &bankV2GenState)
-	if len(bankV2GenState.Balances) == 0 {
-		bankV2GenState = getBankV2GenesisFromV1(bankGenState)
-	}
-	appGenState[bankv2types.ModuleName] = clientCtx.Codec.MustMarshalJSON(&bankV2GenState)
-
 	appGenStateJSON, err := json.MarshalIndent(appGenState, "", "  ")
 	if err != nil {
 		return err
@@ -532,16 +524,3 @@ func writeFile(name, dir string, contents []byte) error {
 
 	return os.WriteFile(file, contents, 0o600)
 }
-
-// getBankV2GenesisFromV1 clones bank/v1 state to bank/v2
-// since we not migrate yet
-// TODO: Remove
-func getBankV2GenesisFromV1(v1GenesisState banktypes.GenesisState) bankv2types.GenesisState {
-	var v2GenesisState bankv2types.GenesisState
-	for _, balance := range v1GenesisState.Balances {
-		v2Balance := bankv2types.Balance(balance)
-		v2GenesisState.Balances = append(v2GenesisState.Balances, v2Balance)
-		v2GenesisState.Supply = v2GenesisState.Supply.Add(balance.Coins...)
-	}
-	return v2GenesisState
-}

simapp/v2/app_config.go

@@ -38,3 +38,16 @@ func (e *Exporter) Close() error {
 
 	return nil
 }
+
+// EmptyExporter is a Exporter for an empty tree.
+type EmptyExporter struct{}
+
+// Next returns ExportDone.
+func (e *EmptyExporter) Next() (*snapshotstypes.SnapshotIAVLItem, error) {
+	return nil, commitment.ErrorExportDone
+}
+
+// Close does nothing.
+func (e *EmptyExporter) Close() error {
+	return nil
+}