Reset metrics on config change (#437)

* Reset metrics on config change * reset metrics logic as a method of Storage * Fix line length linting issues
arriven · Mar 26, 2022 · dd94570 · dd94570
1 parent b444097
commit dd94570
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/runner/runner.go b/src/runner/runner.go
@@ -104,6 +104,8 @@ func (r *Runner) Run(ctx context.Context, logger *zap.Logger) {
 
 			lastKnownConfig = rawConfig
 
+			metrics.Default.ResetAll()
+
 			if cancel != nil {
 				cancel()
 			}

diff --git a/src/utils/metrics/metrics.go b/src/utils/metrics/metrics.go
@@ -76,6 +76,12 @@ func (ms *Storage) Read(name string) uint64 {
 	return sum
 }
 
+func (ms *Storage) ResetAll() {
+	for k := range ms.trackers {
+		ms.trackers[k] = &metricTracker{}
+	}
+}
+
 // NewWriter creates a writer for accumulated writes to the storage
 func (ms *Storage) NewWriter(name, jobID string) *Writer {
 	return &Writer{ms: ms, jobID: jobID, name: name, value: 0}