Merge pull request #6329 from fpetkovski/expose-tsdb-infos

Expose info for each TSDB

cmd/thanos/query.go

@@ -782,12 +782,12 @@ func runQuery(
 			info.WithStoreInfoFunc(func() *infopb.StoreInfo {
 				if httpProbe.IsReady() {
 					mint, maxt := proxy.TimeRange()
-
 					return &infopb.StoreInfo{
 						MinTime:                      mint,
 						MaxTime:                      maxt,
 						SupportsSharding:             true,
 						SupportsWithoutReplicaLabels: true,
+						TsdbInfos:                    proxy.TSDBInfos(),
 					}
 				}
 				return nil

cmd/thanos/receive.go

@@ -343,6 +343,7 @@ func runReceive(
 						MaxTime:                      maxTime,
 						SupportsSharding:             true,
 						SupportsWithoutReplicaLabels: true,
+						TsdbInfos:                    proxy.TSDBInfos(),
 					}
 				}
 				return nil

cmd/thanos/rule.go

@@ -632,6 +632,7 @@ func runRule(
 						MaxTime:                      maxt,
 						SupportsSharding:             true,
 						SupportsWithoutReplicaLabels: true,
+						TsdbInfos:                    tsdbStore.TSDBInfos(),
 					}
 				}
 				return nil

cmd/thanos/sidecar.go

@@ -272,6 +272,7 @@ func runSidecar(
 						MaxTime:                      maxt,
 						SupportsSharding:             true,
 						SupportsWithoutReplicaLabels: true,
+						TsdbInfos:                    promStore.TSDBInfos(),
 					}
 				}
 				return nil

cmd/thanos/store.go

@@ -438,6 +438,7 @@ func runStore(
 					MaxTime:                      maxt,
 					SupportsSharding:             true,
 					SupportsWithoutReplicaLabels: true,
+					TsdbInfos:                    bs.TSDBInfos(),
 				}
 			}
 			return nil

pkg/info/infopb/custom.pb.go

@@ -0,0 +1,40 @@
+package infopb
+
+import (
+	"math"
+
+	"github.com/prometheus/prometheus/model/labels"
+
+	"github.com/thanos-io/thanos/pkg/store/labelpb"
+)
+
+func NewTSDBInfo(mint, maxt int64, lbls []labelpb.ZLabel) TSDBInfo {
+	return TSDBInfo{
+		Labels: labelpb.ZLabelSet{
+			Labels: lbls,
+		},
+		MinTime: mint,
+		MaxTime: maxt,
+	}
+}
+
+type TSDBInfos []TSDBInfo
+
+func (infos TSDBInfos) MaxT() int64 {
+	var maxt int64 = math.MinInt64
+	for _, info := range infos {
+		if info.MaxTime > maxt {
+			maxt = info.MaxTime
+		}
+	}
+	return maxt
+}
+
+func (infos TSDBInfos) LabelSets() []labels.Labels {
+	lsets := make([]labels.Labels, 0, len(infos))
+	for _, info := range infos {
+		lsets = append(lsets, labelpb.ZLabelsToPromLabels(info.Labels.Labels))
+
+	}
+	return lsets
+}