mirror of
https://github.com/grafana/grafana.git
synced 2025-01-13 09:32:12 -06:00
44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
package cloudwatch
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatch"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatch/cloudwatchiface"
|
|
"github.com/grafana/grafana-plugin-sdk-go/experimental/errorsource"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/features"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/utils"
|
|
)
|
|
|
|
func (e *cloudWatchExecutor) executeRequest(ctx context.Context, client cloudwatchiface.CloudWatchAPI,
|
|
metricDataInput *cloudwatch.GetMetricDataInput) ([]*cloudwatch.GetMetricDataOutput, error) {
|
|
mdo := make([]*cloudwatch.GetMetricDataOutput, 0)
|
|
|
|
nextToken := ""
|
|
for {
|
|
if nextToken != "" {
|
|
metricDataInput.NextToken = aws.String(nextToken)
|
|
}
|
|
// GetMetricData EndTime is exclusive, so we round up to the next minute to get the last data point
|
|
if features.IsEnabled(ctx, features.FlagCloudWatchRoundUpEndTime) {
|
|
*metricDataInput.EndTime = metricDataInput.EndTime.Truncate(time.Minute).Add(time.Minute)
|
|
}
|
|
|
|
resp, err := client.GetMetricDataWithContext(ctx, metricDataInput)
|
|
if err != nil {
|
|
return mdo, errorsource.DownstreamError(err, false)
|
|
}
|
|
|
|
mdo = append(mdo, resp)
|
|
utils.QueriesTotalCounter.WithLabelValues(utils.GetMetricDataLabel).Add(float64(len(metricDataInput.MetricDataQueries)))
|
|
if resp.NextToken == nil || *resp.NextToken == "" {
|
|
break
|
|
}
|
|
nextToken = *resp.NextToken
|
|
}
|
|
|
|
return mdo, nil
|
|
}
|