tsdb: Add MustIndex() function to Head

grafana · Dec 24, 2024 · 460b7be · 460b7be
1 parent b500527
commit 460b7be
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tsdb/head_read.go b/tsdb/head_read.go
@@ -37,6 +37,10 @@ func (h *Head) Index() (IndexReader, error) {
 	return h.indexRange(math.MinInt64, math.MaxInt64), nil
 }
 
+func (h *Head) MustIndex() IndexReader {
+	return h.indexRange(math.MinInt64, math.MaxInt64)
+}
+
 func (h *Head) indexRange(mint, maxt int64) *headIndexReader {
 	if hmin := h.MinTime(); hmin > mint {
 		mint = hmin

diff --git a/tsdb/head_read_test.go b/tsdb/head_read_test.go
@@ -428,6 +428,9 @@ func TestHeadIndexReader_PostingsForLabelMatching(t *testing.T) {
 
 		ir, err := h.Index()
 		require.NoError(t, err)
+
+		irMust := h.MustIndex()
+		require.Equal(t, irMust, ir)
 		return ir
 	})
 }