feat: support badger.path in consensus (#36)

steady-bytes · Aug 9, 2024 · 1af42c4 · 1af42c4
1 parent 160dea6
commit 1af42c4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/chassis/consensus.go b/pkg/chassis/consensus.go
@@ -95,7 +95,11 @@ func (c *Runtime) bootstrapRaft(registrar ConsensusRegistrar) {
 	raftConf.SnapshotThreshold = 1024
 
 	// set the path to the directory bolt will use to write to the filesystem
-	store, err := raftboltdb.NewBoltStore(filepath.Join(raftNodeID, "raft.dataRepo"))
+	basePath := c.config.GetString("badger.path")
+	if basePath == "" {
+		basePath = raftNodeID
+	}
+	store, err := raftboltdb.NewBoltStore(filepath.Join(basePath, "raft.dataRepo"))
 	if err != nil {
 		c.logger.Fatal(err.Error())
 	}
@@ -107,7 +111,7 @@ func (c *Runtime) bootstrapRaft(registrar ConsensusRegistrar) {
 	}
 
 	// TODO -> understand this more when diving into snapshots, and how they can be used to recover data
-	snapshotStore, err := raft.NewFileSnapshotStore(raftNodeID, raftSnapShotRetain, os.Stdout)
+	snapshotStore, err := raft.NewFileSnapshotStore(basePath, raftSnapShotRetain, os.Stdout)
 	if err != nil {
 		c.logger.Fatal(err.Error())
 	}