Genesis export stream

server/config/config.go

@@ -192,6 +192,15 @@
 	SnapshotDirectory string `mapstructure:"snapshot-directory"`
 }
 
+// GenesisConfig defines the genesis export, validation, and import configuration
+type GenesisConfig struct {
+	// StreamImport defines if the genesis.json is in stream form or not.
+	StreamImport bool `mapstructure:"stream-import"`
+
+	// ImportFile sets the genesis json file from which to stream from
+	ImportFile string `mapstructure:"import-file"`
+}
+
 // Config defines the server's top level configuration
 type Config struct {
 	BaseConfig `mapstructure:",squash"`
@@ -205,6 +214,7 @@
 	StateSync   StateSyncConfig          `mapstructure:"state-sync"`
 	StateCommit config.StateCommitConfig `mapstructure:"state-commit"`
 	StateStore  config.StateStoreConfig  `mapstructure:"state-store"`
+	Genesis     GenesisConfig            `mapstructure:genesis`
 }
 
 // SetMinGasPrices sets the validator's minimum gas prices.
@@ -288,6 +298,10 @@
 		},
 		StateCommit: config.DefaultStateCommitConfig(),
 		StateStore:  config.DefaultStateStoreConfig(),
+		Genesis: GenesisConfig{
+			StreamImport: false,
+			ImportFile:   "",
+		},
 	}
 }
 
@@ -391,6 +405,10 @@
 			PruneIntervalSeconds: v.GetInt("state-store.prune-interval-seconds"),
 			ImportNumWorkers:     v.GetInt("state-store.import-num-workers"),
 		},
+		Genesis: GenesisConfig{
+			StreamImport: v.GetBool("genesis.stream-import"),
+			ImportFile:   v.GetString("genesis.import-file"),
+		},
 	}, nil
 }
 

server/config/toml.go

@@ -246,6 +246,20 @@ snapshot-keep-recent = {{ .StateSync.SnapshotKeepRecent }}
 # default is emtpy which will then store under the app home directory same as before.
 snapshot-directory = "{{ .StateSync.SnapshotDirectory }}"
 
+###############################################################################
+###                         Genesis Configuration                           ###
+###############################################################################
+
+
+# Genesis config allows configuring whether to stream from an genesis json file in streamed form
+[genesis]
+
+# stream-import specifies whether to the stream the import from the genesis json file. The genesis
+# file must be in stream form and exported in a streaming fashion.
+stream-import = {{ .Genesis.StreamImport }}
+
+# import-file specifies the path of the genesis json file to stream from.
+import-file = "{{ .Genesis.ImportFile }}"
 ` + config.DefaultConfigTemplate
 
 var configTemplate *template.Template

server/export.go

@@ -7,8 +7,10 @@
 	"fmt"
 	"io/ioutil"
 	"os"
+	"time"
 
 	"github.com/spf13/cobra"
+	tmbytes "github.com/tendermint/tendermint/libs/bytes"
 	tmtypes "github.com/tendermint/tendermint/types"
 
 	"github.com/cosmos/cosmos-sdk/client/flags"
@@ -17,11 +19,22 @@
 )
 
 const (
+	FlagIsStreaming      = "streaming"
+	FlagStreamingFile    = "streaming-file"
 	FlagHeight           = "height"
 	FlagForZeroHeight    = "for-zero-height"
 	FlagJailAllowedAddrs = "jail-allowed-addrs"
 )
 
+type GenesisDocNoAppState struct {
+	GenesisTime     time.Time                  `json:"genesis_time"`
+	ChainID         string                     `json:"chain_id"`
+	InitialHeight   int64                      `json:"initial_height,string"`
+	ConsensusParams *tmtypes.ConsensusParams   `json:"consensus_params,omitempty"`
+	Validators      []tmtypes.GenesisValidator `json:"validators,omitempty"`
+	AppHash         tmbytes.HexBytes           `json:"app_hash"`
+}
+
 // ExportCmd dumps app state to JSON.
 func ExportCmd(appExporter types.AppExporter, defaultNodeHome string) *cobra.Command {
 	cmd := &cobra.Command{
@@ -38,6 +51,20 @@
 				return err
 			}
 
+			isStreaming, err := cmd.Flags().GetBool(FlagIsStreaming)
+			if err != nil {
+				return err
+			}
+
+			streamingFile, err := cmd.Flags().GetString(FlagStreamingFile)
+			if err != nil {
+				return err
+			}
+
+			if isStreaming && streamingFile == "" {
+				return fmt.Errorf("file to export stream to not provided")
+			}
+
 			db, err := openDB(config.RootDir)
 			if err != nil {
 				return err
@@ -67,7 +94,56 @@
 			forZeroHeight, _ := cmd.Flags().GetBool(FlagForZeroHeight)
 			jailAllowedAddrs, _ := cmd.Flags().GetStringSlice(FlagJailAllowedAddrs)
 
-			exported, err := appExporter(serverCtx.Logger, db, traceWriter, height, forZeroHeight, jailAllowedAddrs, serverCtx.Viper)
+			if isStreaming {
+				file, err := os.Create(streamingFile)
+				if err != nil {
+					return err
+				}
+				exported, err := appExporter(serverCtx.Logger, db, traceWriter, height, forZeroHeight, jailAllowedAddrs, serverCtx.Viper, file)
+				if err != nil {
+					return fmt.Errorf("error exporting state: %v", err)
+				}
+
+				doc, err := tmtypes.GenesisDocFromFile(serverCtx.Config.GenesisFile())
+				if err != nil {
+					return err
+				}
+
+				genesisDocNoAppHash := GenesisDocNoAppState{
+					GenesisTime:   doc.GenesisTime,
+					ChainID:       doc.ChainID,
+					AppHash:       doc.AppHash,
+					InitialHeight: exported.Height,
+					ConsensusParams: &tmtypes.ConsensusParams{
+						Block: tmtypes.BlockParams{
+							MaxBytes: exported.ConsensusParams.Block.MaxBytes,
+							MaxGas:   exported.ConsensusParams.Block.MaxGas,
+						},
+						Evidence: tmtypes.EvidenceParams{
+							MaxAgeNumBlocks: exported.ConsensusParams.Evidence.MaxAgeNumBlocks,
+							MaxAgeDuration:  exported.ConsensusParams.Evidence.MaxAgeDuration,
+							MaxBytes:        exported.ConsensusParams.Evidence.MaxBytes,
+						},
+						Validator: tmtypes.ValidatorParams{
+							PubKeyTypes: exported.ConsensusParams.Validator.PubKeyTypes,
+						},
+					},
+					Validators: exported.Validators,
+				}
+
+				// NOTE: Tendermint uses a custom JSON decoder for GenesisDoc
+				// (except for stuff inside AppState). Inside AppState, we're free
+				// to encode as protobuf or amino.
+				encoded, err := json.Marshal(genesisDocNoAppHash)
+				if err != nil {
+					return err
+				}
+
+				file.Write([]byte(fmt.Sprintf("%s", string(sdk.MustSortJSON(encoded)))))
+				return nil
+			}
+
+			exported, err := appExporter(serverCtx.Logger, db, traceWriter, height, forZeroHeight, jailAllowedAddrs, serverCtx.Viper, nil)
 			if err != nil {
 				return fmt.Errorf("error exporting state: %v", err)
 			}
@@ -108,6 +184,8 @@
 		},
 	}
 
+	cmd.Flags().Bool(FlagIsStreaming, false, "Whether to stream the export in chunks. Useful when genesis is extremely large and cannot fit into memory.")
+	cmd.Flags().String(FlagStreamingFile, "genesis-stream.json", "The file to export the streamed genesis to")
 	cmd.Flags().String(flags.FlagHome, defaultNodeHome, "The application home directory")
 	cmd.Flags().Int64(FlagHeight, -1, "Export state from a particular height (-1 means latest height)")
 	cmd.Flags().Bool(FlagForZeroHeight, false, "Export state to start at height zero (perform preproccessing)")

server/export_test.go

@@ -147,7 +147,7 @@ func setupApp(t *testing.T, tempDir string) (*simapp.SimApp, context.Context, *t
 	app.Commit(context.Background())
 
 	cmd := server.ExportCmd(
-		func(_ log.Logger, _ dbm.DB, _ io.Writer, height int64, forZeroHeight bool, jailAllowedAddrs []string, appOptons types.AppOptions) (types.ExportedApp, error) {
+		func(_ log.Logger, _ dbm.DB, _ io.Writer, height int64, forZeroHeight bool, jailAllowedAddrs []string, appOptons types.AppOptions, file *os.File) (types.ExportedApp, error) {
 			encCfg := simapp.MakeTestEncodingConfig()
 
 			var simApp *simapp.SimApp

server/start.go

@@ -13,6 +13,7 @@
 
 	clientconfig "github.com/cosmos/cosmos-sdk/client/config"
 
+	genesistypes "github.com/cosmos/cosmos-sdk/types/genesis"
 	"github.com/spf13/cobra"
 	abciclient "github.com/tendermint/tendermint/abci/client"
 	"github.com/tendermint/tendermint/abci/server"
@@ -170,11 +171,6 @@
 
 			serverCtx.Viper.Set(flags.FlagChainID, chainID)
 
-			genesisFile, _ := tmtypes.GenesisDocFromFile(serverCtx.Config.GenesisFile())
-			if genesisFile.ChainID != clientCtx.ChainID {
-				panic(fmt.Sprintf("genesis file chain-id=%s does not equal config.toml chain-id=%s", genesisFile.ChainID, clientCtx.ChainID))
-			}
-
 			if enableTracing, _ := cmd.Flags().GetBool(tracing.FlagTracing); !enableTracing {
 				serverCtx.Logger.Info("--tracing not passed in, tracing is not enabled")
 				tracerProviderOptions = []trace.TracerProviderOption{}
@@ -197,6 +193,12 @@
 			if err != nil {
 				return fmt.Errorf("failed to initialize telemetry: %w", err)
 			}
+			if !config.Genesis.StreamImport {
+				genesisFile, _ := tmtypes.GenesisDocFromFile(serverCtx.Config.GenesisFile())
+				if genesisFile.ChainID != clientCtx.ChainID {
+					panic(fmt.Sprintf("genesis file chain-id=%s does not equal config.toml chain-id=%s", genesisFile.ChainID, clientCtx.ChainID))
+				}
+			}
 
 			restartCoolDownDuration := time.Second * time.Duration(serverCtx.Config.SelfRemediation.RestartCooldownSeconds)
 			// Set the first restart time to be now - restartCoolDownDuration so that the first restart can trigger whenever
@@ -386,13 +388,27 @@
 		config.GRPC.Enable = true
 	} else {
 		ctx.Logger.Info("starting node with ABCI Tendermint in-process")
+		var gen *tmtypes.GenesisDoc
+		if config.Genesis.StreamImport {
+			lines := genesistypes.IngestGenesisFileLineByLine(config.Genesis.ImportFile)
+			for line := range lines {
+				genDoc, err := tmtypes.GenesisDocFromJSON([]byte(line))
+				if err != nil {
+					continue
+				}
+				if gen != nil {
+					return fmt.Errorf("error: multiple genesis docs found in stream")
+				}
+				gen = genDoc
+			}
+		}
 		tmNode, err = node.New(
 			goCtx,
 			ctx.Config,
 			ctx.Logger,
 			restartCh,
 			abciclient.NewLocalClient(ctx.Logger, app),
-			nil,
+			gen,
 			tracerProviderOptions,
 			nodeMetricsProvider,
 		)

server/types/app.go

@@ -3,6 +3,7 @@ package types
 import (
 	"encoding/json"
 	"io"
+	"os"
 	"time"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
@@ -84,5 +85,6 @@ type (
 
 	// AppExporter is a function that dumps all app state to
 	// JSON-serializable structure and returns the current validator set.
-	AppExporter func(log.Logger, dbm.DB, io.Writer, int64, bool, []string, AppOptions) (ExportedApp, error)
+	// If a file is specified,
+	AppExporter func(log.Logger, dbm.DB, io.Writer, int64, bool, []string, AppOptions, *os.File) (ExportedApp, error)
 )

simapp/app.go

@@ -32,6 +32,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/store/streaming"
 	"github.com/cosmos/cosmos-sdk/testutil/testdata"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	genesistypes "github.com/cosmos/cosmos-sdk/types/genesis"
 	"github.com/cosmos/cosmos-sdk/types/module"
 	"github.com/cosmos/cosmos-sdk/utils"
 	"github.com/cosmos/cosmos-sdk/version"
@@ -594,7 +595,7 @@ func (app *SimApp) InitChainer(ctx sdk.Context, req abci.RequestInitChain) abci.
 		panic(err)
 	}
 	app.UpgradeKeeper.SetModuleVersionMap(ctx, app.mm.GetVersionMap())
-	return app.mm.InitGenesis(ctx, app.appCodec, genesisState)
+	return app.mm.InitGenesis(ctx, app.appCodec, genesisState, genesistypes.GenesisImportConfig{})
 }
 
 // LoadHeight loads a particular height

simapp/sim_test.go

@@ -18,6 +18,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/simapp/helpers"
 	"github.com/cosmos/cosmos-sdk/store"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/genesis"
 	simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
 	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
 	authzkeeper "github.com/cosmos/cosmos-sdk/x/authz/keeper"
@@ -155,7 +156,7 @@ func TestAppImportExport(t *testing.T) {
 
 	ctxA := app.NewContext(true, tmproto.Header{Height: app.LastBlockHeight()})
 	ctxB := newApp.NewContext(true, tmproto.Header{Height: app.LastBlockHeight()})
-	newApp.mm.InitGenesis(ctxB, app.AppCodec(), genesisState)
+	newApp.mm.InitGenesis(ctxB, app.AppCodec(), genesisState, genesis.GenesisImportConfig{})
 	newApp.StoreConsensusParams(ctxB, exported.ConsensusParams)
 
 	fmt.Printf("comparing stores...\n")

simapp/simd/cmd/root.go

@@ -313,7 +313,7 @@
 // and exports state.
 func (a appCreator) appExport(
 	logger log.Logger, db dbm.DB, traceStore io.Writer, height int64, forZeroHeight bool, jailAllowedAddrs []string,
-	appOpts servertypes.AppOptions) (servertypes.ExportedApp, error) {
+	appOpts servertypes.AppOptions, file *os.File) (servertypes.ExportedApp, error) {
 
 	var simApp *simapp.SimApp
 	homePath, ok := appOpts.Get(flags.FlagHome).(string)