panic less often

celestiaorg · Jul 23, 2024 · 1a0ed39 · 1a0ed39
1 parent 9cb41db
commit 1a0ed39
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/store/store.go b/store/store.go
@@ -180,6 +180,10 @@ func (s *Store[H]) Height() uint64 {
 
 	head, err := s.Head(ctx)
 	if err != nil {
+		if errors.Is(err, context.Canceled) ||
+			errors.Is(err, datastore.ErrNotFound) {
+			return 0
+		}
 		panic(err)
 	}
 	return head.Height()