fix: Token generation when restarting in dev mode

polarstreams · Oct 17, 2022 · bf53c95 · bf53c95
1 parent 05e1ce7
commit bf53c95
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/internal/discovery/discoverer.go b/internal/discovery/discoverer.go
@@ -101,6 +101,9 @@ func (d *discoverer) Init() error {
 		if err := d.loadFixedTopology(0); err != nil {
 			return err
 		}
+		if err := d.loadGenerations(); err != nil {
+			return err
+		}
 		return d.startClientDiscoveryServer()
 	}
 

diff --git a/internal/ownership/generator.go b/internal/ownership/generator.go
@@ -472,6 +472,17 @@ func (o *generator) processGeneration(message genMessage) creationError {
 }
 
 func (o *generator) createDevGeneration() {
+	if existing := o.discoverer.Generation(StartToken); existing != nil {
+		if existing.End != StartToken || existing.Version != 1 {
+			log.Panic().Msgf(
+				"There's existing token data in local db store but it's not in dev mode." +
+					" Consider cleaning the data directory before starting in dev mode.s")
+		}
+
+		log.Info().Msgf("Dev generation restored")
+		return
+	}
+
 	gen := &Generation{
 		Start:     StartToken,
 		End:       StartToken,
@@ -484,7 +495,9 @@ func (o *generator) createDevGeneration() {
 		Status:    StatusCommitted,
 		Parents:   []GenId{},
 	}
-	o.discoverer.RepairCommitted(gen)
+
+	err := o.discoverer.RepairCommitted(gen)
+	utils.PanicIfErr(err, "Could not create dev generation")
 }
 
 // Gets a value between base delay and max delay