truncate-database command line utility

cmd/util/cmd/common/storage.go

@@ -8,10 +8,15 @@ import (
 )
 
 func InitStorage(datadir string) *badger.DB {
+	return InitStorageWithTruncate(datadir, false)
+}
+
+func InitStorageWithTruncate(datadir string, truncate bool) *badger.DB {
 	opts := badger.
 		DefaultOptions(datadir).
 		WithKeepL0InMemory(true).
-		WithLogger(nil)
+		WithLogger(nil).
+		WithTruncate(truncate)
 
 	db, err := badger.Open(opts)
 	if err != nil {

cmd/util/cmd/root.go

@@ -14,6 +14,7 @@ import (
 	extract "github.com/onflow/flow-go/cmd/util/cmd/execution-state-extract"
 	"github.com/onflow/flow-go/cmd/util/cmd/find-block"
 	"github.com/onflow/flow-go/cmd/util/cmd/read-execution-state"
+	truncate_database "github.com/onflow/flow-go/cmd/util/cmd/truncate-database"
 )
 
 var (
@@ -52,6 +53,7 @@ func addCommands() {
 	rootCmd.AddCommand(find.Cmd)
 	rootCmd.AddCommand(read.Cmd)
 	rootCmd.AddCommand(checkpoint_list_tries.Cmd)
+	rootCmd.AddCommand(truncate_database.Cmd)
 }
 
 func initConfig() {

cmd/util/cmd/truncate-database/cmd.go

@@ -0,0 +1,37 @@
+package truncate_database
+
+import (
+	"github.com/rs/zerolog/log"
+
+	"github.com/spf13/cobra"
+
+	"github.com/onflow/flow-go/cmd/util/cmd/common"
+)
+
+var (
+	flagDatadir string
+)
+
+var Cmd = &cobra.Command{
+	Use:   "truncate-database",
+	Short: "Truncates protocol state database (Possible data loss!)",
+	Run:   run,
+}
+
+func init() {
+
+	Cmd.Flags().StringVar(&flagDatadir, "datadir", "",
+		"directory that stores the protocol state")
+	_ = Cmd.MarkFlagRequired("datadir")
+
+}
+
+func run(*cobra.Command, []string) {
+
+	log.Info().Msg("Opening database with truncate")
+
+	db := common.InitStorageWithTruncate(flagDatadir, true)
+	defer db.Close()
+
+	log.Info().Msg("Truncated")
+}