close metrics

celestiaorg · Oct 31, 2023 · 7d65a7b · 7d65a7b
1 parent c5aec2d
commit 7d65a7b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 40 deletions.
diff --git a/metrics.go b/metrics.go
diff --git a/store/metrics.go b/store/metrics.go
@@ -111,5 +111,9 @@ func (m *metrics) observe(ctx context.Context, f func(context.Context)) {
 }
 
 func (m *metrics) Close() error {
+	if m == nil {
+		return nil
+	}
+
 	return m.headHeightReg.Unregister()
 }
diff --git a/store/store.go b/store/store.go
@@ -168,7 +168,7 @@ func (s *Store[H]) Stop(ctx context.Context) error {
 	// cleanup caches
 	s.cache.Purge()
 	s.heightIndex.cache.Purge()
-	return nil
+	return s.metrics.Close()
 }
 
 func (s *Store[H]) Height() uint64 {