tsdbstore: Optimized response framing if iterator finished.

Signed-off-by: Bartlomiej Plotka <[email protected]>
thanos-io · Aug 26, 2020 · 0a69bad · 0a69bad
1 parent a6b1b6c
commit 0a69bad
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/store/tsdb.go b/pkg/store/tsdb.go
@@ -173,12 +173,14 @@ func (s *TSDBStore) Series(r *storepb.SeriesRequest, srv storepb.Store_SeriesSer
 			if frameBytesLeft > 0 && isNext {
 				continue
 			}
-
 			if err := srv.Send(storepb.NewSeriesResponse(&storepb.Series{Labels: seriesLabels.Labels, Chunks: seriesChunks})); err != nil {
 				return status.Error(codes.Aborted, err.Error())
 			}
-			frameBytesLeft = bytesLeftForChunks
-			seriesChunks = make([]storepb.AggrChunk, 0, len(seriesChunks))
+
+			if isNext {
+				frameBytesLeft = bytesLeftForChunks
+				seriesChunks = make([]storepb.AggrChunk, 0, len(seriesChunks))
+			}
 		}
 		if err := chIter.Err(); err != nil {
 			return status.Error(codes.Internal, errors.Wrap(err, "chunk iter").Error())