add get_all_duration and merge_duration to SG query hints (#6730)

Signed-off-by: Ben Ye <[email protected]>
thanos-io · Sep 19, 2023 · 18f8035 · 18f8035
1 parent 9524124
commit 18f8035
Show file tree

Hide file tree

Showing 4 changed files with 174 additions and 64 deletions.
diff --git a/pkg/store/bucket.go b/pkg/store/bucket.go
@@ -3647,6 +3647,8 @@ func (s queryStats) toHints() *hintspb.QueryStats {
 		MergedSeriesCount:      int64(s.mergedSeriesCount),
 		MergedChunksCount:      int64(s.mergedChunksCount),
 		DataDownloadedSizeSum:  int64(s.DataDownloadedSizeSum),
+		GetAllDuration:         s.GetAllDuration,
+		MergeDuration:          s.MergeDuration,
 	}
 }
 

diff --git a/pkg/store/hintspb/custom.go b/pkg/store/hintspb/custom.go
@@ -47,4 +47,7 @@ func (m *QueryStats) Merge(other *QueryStats) {
 	m.ChunksFetchedSizeSum += other.ChunksFetchedSizeSum
 	m.ChunksTouched += other.ChunksTouched
 	m.ChunksTouchedSizeSum += other.ChunksTouchedSizeSum
+
+	m.GetAllDuration += other.GetAllDuration
+	m.MergeDuration += other.MergeDuration
 }
diff --git a/pkg/store/hintspb/hints.pb.go b/pkg/store/hintspb/hints.pb.go
diff --git a/pkg/store/hintspb/hints.proto b/pkg/store/hintspb/hints.proto
@@ -5,6 +5,7 @@ syntax = "proto3";
 package hintspb;
 
 import "gogoproto/gogo.proto";
+import "google/protobuf/duration.proto";
 import "store/storepb/types.proto";
 
 option go_package = "hintspb";
@@ -90,4 +91,6 @@ message QueryStats {
     int64 chunks_fetch_count = 19;
 
     int64 data_downloaded_size_sum = 20;
+    google.protobuf.Duration get_all_duration = 21 [(gogoproto.stdduration) = true, (gogoproto.nullable) = false];
+    google.protobuf.Duration merge_duration = 22 [(gogoproto.stdduration) = true, (gogoproto.nullable) = false];
 }