Update telemetry/metrics.go

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
berachain · Dec 6, 2023 · 52d7b96 · 52d7b96
1 parent a8cc689
commit 52d7b96
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/telemetry/metrics.go b/telemetry/metrics.go
@@ -113,10 +113,15 @@ func New(cfg Config) (_ *Metrics, rerr error) {
 		memSink := metrics.NewInmemSink(10*time.Second, time.Minute)
 		sink = memSink
 		inMemSig := metrics.DefaultInmemSignal(memSink)
-		defer func() {
-			if rerr != nil {
-				inMemSig.Stop()
-			}
+		case "mem":
+		  memSink := metrics.NewInmemSink(10*time.Second, time.Minute)
+		  sink = memSink
+		  inMemSig := metrics.DefaultInmemSignal(memSink)
+		  defer func() {
+		    if rerr != nil {
+		      inMemSig.Stop()
+		    }
+		  }()
 		}()
 	case "statsd":
 		var err error