Fixing logs and adding restore snapshot flag

maxpert · Oct 10, 2022 · acb1a29 · acb1a29
1 parent a4352b2
commit acb1a29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/cfg/config.go b/cfg/config.go
@@ -7,8 +7,9 @@ import (
 	"github.com/nats-io/nats.go"
 )
 
-var Cleanup = flag.Bool("cleanup", false, "Cleanup all trigger hooks for marmot")
+var Cleanup = flag.Bool("cleanup", false, "Only cleanup marmot triggers and changelogs")
 var SaveSnapshot = flag.Bool("save-snapshot", false, "Only take snapshot and upload")
+var RestoreSnapshot = flag.Bool("restore-snapshot", true, "Restore snapshot at boot")
 var DBPathString = flag.String("db-path", "/tmp/marmot.db", "Path to SQLite database")
 var NodeID = flag.Uint64("node-id", rand.Uint64(), "Node ID")
 var NatsAddr = flag.String("nats-url", nats.DefaultURL, "NATS server URL")

diff --git a/marmot.go b/marmot.go
@@ -56,9 +56,11 @@ func main() {
 		return
 	}
 
-	err = rep.RestoreSnapshot()
-	if err != nil {
-		log.Panic().Err(err).Msg("Unable to restore snapshot")
+	if *cfg.RestoreSnapshot {
+		err = rep.RestoreSnapshot()
+		if err != nil {
+			log.Panic().Err(err).Msg("Unable to restore snapshot")
+		}
 	}
 
 	log.Info().Msg("Listing tables to watch...")