-
Notifications
You must be signed in to change notification settings - Fork 15
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* add rollback cmd * bump deps * add height condition check * lint --------- Co-authored-by: beer-1 <[email protected]>
- Loading branch information
Showing
5 changed files
with
82 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,65 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "path/filepath" | ||
| "strconv" | ||
|
|
||
| cmtcmd "github.com/cometbft/cometbft/cmd/cometbft/commands" | ||
| "github.com/spf13/cobra" | ||
|
|
||
| dbm "github.com/cosmos/cosmos-db" | ||
| "github.com/cosmos/cosmos-sdk/server" | ||
| "github.com/cosmos/cosmos-sdk/server/types" | ||
| ) | ||
|
|
||
| // NewMultipleRollbackCmd creates a command to rollback CometBFT and multistore state by one height. | ||
| func NewMultipleRollbackCmd(appCreator types.AppCreator) *cobra.Command { | ||
| cmd := &cobra.Command{ | ||
| Use: "mrollback [height]", | ||
| Short: "rollback Cosmos SDK and CometBFT state to the given height", | ||
| Long: ` | ||
| A state rollback is performed to recover from an incorrect application state transition, | ||
| when CometBFT has persisted an incorrect app hash and is thus unable to make | ||
| progress. Rollback overwrites a state with the state at the given height. All | ||
| blocks after the given height are removed from the blockchain. | ||
| `, | ||
| Args: cobra.ExactArgs(1), | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| ctx := server.GetServerContextFromCmd(cmd) | ||
|
|
||
| height, err := strconv.ParseInt(args[0], 10, 64) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if height <= 0 { | ||
| return fmt.Errorf("height must be greater than 0") | ||
| } | ||
|
|
||
| dataDir := filepath.Join(ctx.Config.RootDir, "data") | ||
| db, err := dbm.NewDB("application", server.GetAppDBBackend(ctx.Viper), dataDir) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| app := appCreator(ctx.Logger, db, nil, ctx.Viper) | ||
| if curHeight := app.CommitMultiStore().LatestVersion(); height >= curHeight { | ||
| return fmt.Errorf("height must be less than the current height %d", curHeight) | ||
| } | ||
|
|
||
| // rollback CometBFT state | ||
| height, hash, err := cmtcmd.RollbackMultipleState(ctx.Config, height) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to rollback CometBFT state: %w", err) | ||
| } | ||
| // rollback the multistore | ||
|
|
||
| if err := app.CommitMultiStore().RollbackToVersion(height); err != nil { | ||
| return fmt.Errorf("failed to rollback to version: %w", err) | ||
| } | ||
|
|
||
| fmt.Printf("Rolled back state to height %d and hash %X\n", height, hash) | ||
| return nil | ||
| }, | ||
| } | ||
| return cmd | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters