adds a cli to override database dirty flag (#629)

* adds a cli to override database dirty flag

* gofmt

plugins/database/parameters.go

@@ -9,9 +9,12 @@ const (
 	CfgDatabaseDir = "database.directory"
 	// CfgDatabaseInMemory defines whether to use an in-memory database.
 	CfgDatabaseInMemory = "database.inMemory"
+	// CfgDatabaseDirty defines whether to override the database dirty flag.
+	CfgDatabaseDirty = "database.dirty"
 )
 
 func init() {
 	flag.String(CfgDatabaseDir, "mainnetdb", "path to the database folder")
 	flag.Bool(CfgDatabaseInMemory, false, "whether the database is only kept in memory and not persisted")
+	flag.String(CfgDatabaseDirty, "", "set the dirty flag of the database")
 }

plugins/database/plugin.go

@@ -3,6 +3,7 @@ package database
 
 import (
 	"errors"
+	"strconv"
 	"sync"
 	"time"
 
@@ -77,6 +78,17 @@ func configure(_ *node.Plugin) {
 		log.Fatalf("Failed to check database version: %s", err)
 	}
 
+	if str := config.Node().GetString(CfgDatabaseDirty); str != "" {
+		val, err := strconv.ParseBool(str)
+		if err != nil {
+			log.Warnf("Invalid %s: %s", CfgDatabaseDirty, err)
+		} else if val {
+			MarkDatabaseUnhealthy()
+		} else {
+			MarkDatabaseHealthy()
+		}
+	}
+
 	if IsDatabaseUnhealthy() {
 		log.Fatal("The database is marked as not properly shutdown/corrupted, please delete the database folder and restart.")
 	}