func query(
endpoints *datastructs.ApplicationStatuses,
metricName string,
aggregatorGen tsdb.AggregatorGenerator,
start, end float64,
options *QueryOptions) (result *tsdb.TaggedTimeSeriesSet, err error) {
if options == nil {
options = &QueryOptions{}
}
apps, store := endpoints.AllWithStore()
var taggedTimeSeriesSlice []tsdb.TaggedTimeSeries
var metricNameFound bool
if options.GroupByHostName && options.GroupByAppName {
for i := range apps {
if options.isIncluded(apps[i].EndpointId.HostName(), apps[i].Name) {
timeSeries, ok := store.TsdbTimeSeries(
metricName,
apps[i].EndpointId,
start,
end)
if ok {
metricNameFound = true
var aggregator tsdb.Aggregator
aggregator, err = aggregatorGen(start, end)
if err != nil {
return
}
aggregator.Add(timeSeries)
aggregatedTimeSeries := aggregator.Aggregate()
if len(aggregatedTimeSeries) != 0 {
taggedTimeSeriesSlice = append(
taggedTimeSeriesSlice, tsdb.TaggedTimeSeries{
Tags: tsdb.TagSet{
HostName: apps[i].EndpointId.HostName(),
AppName: apps[i].Name,
},
Values: aggregatedTimeSeries,
})
}
}
}
}
} else {
aggregatorMap := make(map[tsdb.TagSet]tsdb.Aggregator)
for i := range apps {
if options.isIncluded(apps[i].EndpointId.HostName(), apps[i].Name) {
timeSeries, ok := store.TsdbTimeSeries(
metricName,
apps[i].EndpointId,
start,
end)
if ok {
metricNameFound = true
var tagSet tsdb.TagSet
if options.GroupByHostName {
tagSet.HostName = apps[i].EndpointId.HostName()
}
if options.GroupByAppName {
tagSet.AppName = apps[i].Name
}
aggregator := aggregatorMap[tagSet]
if aggregator == nil {
aggregator, err = aggregatorGen(start, end)
if err != nil {
return
}
aggregatorMap[tagSet] = aggregator
}
aggregator.Add(timeSeries)
}
}
}
if metricNameFound {
for k, v := range aggregatorMap {
aggregatedTimeSeries := v.Aggregate()
if len(aggregatedTimeSeries) != 0 {
taggedTimeSeriesSlice = append(
taggedTimeSeriesSlice, tsdb.TaggedTimeSeries{
Tags: k,
Values: aggregatedTimeSeries,
})
}
}
}
}
if metricNameFound {
return &tsdb.TaggedTimeSeriesSet{
MetricName: metricName,
Data: taggedTimeSeriesSlice,
GroupedByHostName: options.GroupByHostName,
GroupedByAppName: options.GroupByAppName,
}, nil
}
return nil, ErrNoSuchMetric
}
