Fixes wrong cancellation

pkg/phlare/phlare.go

@@ -314,7 +314,7 @@ func (f *Phlare) setupModuleManager() error {
 		API:            {Server},
 		Agent:          {API},
 		Distributor:    {Overrides, Ring, API, UsageReport},
-		Querier:        {API, MemberlistKV, Ring, UsageReport},
+		Querier:        {Overrides, API, MemberlistKV, Ring, UsageReport},
 		QueryFrontend:  {OverridesExporter, API, MemberlistKV, UsageReport},
 		QueryScheduler: {Overrides, API, MemberlistKV, UsageReport},
 		Ingester:       {Overrides, API, MemberlistKV, Storage, UsageReport},

pkg/phlaredb/block_querier.go

@@ -1105,7 +1105,8 @@ func (r *parquetReader[M, P]) open(ctx context.Context, bucketReader phlareobj.B
 		}
 		r.size = attrs.Size
 	}
-	ra, err := bucketReader.ReaderAt(ctx, filePath)
+	// the same reader is used to serve all requests, so we pass context.Background() here
+	ra, err := bucketReader.ReaderAt(context.Background(), filePath)
 	if err != nil {
 		return errors.Wrapf(err, "create reader '%s'", filePath)
 	}

pkg/phlaredb/phlaredb.go

@@ -252,19 +252,19 @@ func (f *PhlareDB) Series(ctx context.Context, req *connect.Request[ingestv1.Ser
 func (f *PhlareDB) MergeProfilesStacktraces(ctx context.Context, stream *connect.BidiStream[ingestv1.MergeProfilesStacktracesRequest, ingestv1.MergeProfilesStacktracesResponse]) error {
 	f.headLock.RLock()
 	defer f.headLock.RUnlock()
-	return MergeProfilesStacktraces(ctx, stream, f.blockQuerier.Queriers().ForTimeRange)
+	return MergeProfilesStacktraces(ctx, stream, f.queriers().ForTimeRange)
 }
 
 func (f *PhlareDB) MergeProfilesLabels(ctx context.Context, stream *connect.BidiStream[ingestv1.MergeProfilesLabelsRequest, ingestv1.MergeProfilesLabelsResponse]) error {
 	f.headLock.RLock()
 	defer f.headLock.RUnlock()
-	return MergeProfilesLabels(ctx, stream, f.blockQuerier.Queriers().ForTimeRange)
+	return MergeProfilesLabels(ctx, stream, f.queriers().ForTimeRange)
 }
 
 func (f *PhlareDB) MergeProfilesPprof(ctx context.Context, stream *connect.BidiStream[ingestv1.MergeProfilesPprofRequest, ingestv1.MergeProfilesPprofResponse]) error {
 	f.headLock.RLock()
 	defer f.headLock.RUnlock()
-	return MergeProfilesPprof(ctx, stream, f.blockQuerier.Queriers().ForTimeRange)
+	return MergeProfilesPprof(ctx, stream, f.queriers().ForTimeRange)
 }
 
 type BidiServerMerge[Res any, Req any] interface {

pkg/phlaredb/query/iters.go

@@ -933,7 +933,6 @@ func (r *RowNumberIterator[T]) Err() error {
 
 func (r *RowNumberIterator[T]) Seek(to RowNumberWithDefinitionLevel) bool {
 	for CompareRowNumbers(0, r.current.RowNumber, to.RowNumber) == -1 {
-
 		if !r.Next() {
 			return false
 		}