mirror of
https://github.com/fluxcd/flagger.git
synced 2026-04-15 06:57:34 +00:00
114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
package providers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatch"
|
|
|
|
flaggerv1 "github.com/weaveworks/flagger/pkg/apis/flagger/v1beta1"
|
|
)
|
|
|
|
const (
|
|
cloudWatchMaxRetries = 3
|
|
cloudWatchStartDeltaMultiplierOnMetricInterval = 10
|
|
)
|
|
|
|
type CloudWatchProvider struct {
|
|
client cloudWatchClient
|
|
startDelta time.Duration
|
|
}
|
|
|
|
// for the testing purpose
|
|
type cloudWatchClient interface {
|
|
GetMetricData(input *cloudwatch.GetMetricDataInput) (*cloudwatch.GetMetricDataOutput, error)
|
|
}
|
|
|
|
// NewCloudWatchProvider takes a metricInterval, a provider spec and the credentials map, and
|
|
// returns a cloudWatchProvider ready to execute queries against the AWS CloudWatch metrics
|
|
func NewCloudWatchProvider(metricInterval string, provider flaggerv1.MetricTemplateProvider) (*CloudWatchProvider, error) {
|
|
if provider.Region == "" {
|
|
return nil, fmt.Errorf("region not specified")
|
|
}
|
|
|
|
sess, err := session.NewSession(aws.NewConfig().
|
|
WithRegion(provider.Region).WithMaxRetries(cloudWatchMaxRetries))
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating aws session: %s", err.Error())
|
|
}
|
|
|
|
md, err := time.ParseDuration(metricInterval)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error parsing metric interval: %s", err.Error())
|
|
}
|
|
|
|
return &CloudWatchProvider{
|
|
client: cloudwatch.New(sess),
|
|
startDelta: cloudWatchStartDeltaMultiplierOnMetricInterval * md,
|
|
}, err
|
|
}
|
|
|
|
// RunQuery executes the aws cloud watch metrics query against GetMetricData endpoint
|
|
// and returns the the first result as float64
|
|
func (p *CloudWatchProvider) RunQuery(query string) (float64, error) {
|
|
var cq []*cloudwatch.MetricDataQuery
|
|
if err := json.Unmarshal([]byte(query), &cq); err != nil {
|
|
return 0, fmt.Errorf("error unmarshaling query: %s", err.Error())
|
|
}
|
|
|
|
end := time.Now()
|
|
start := end.Add(-p.startDelta)
|
|
res, err := p.client.GetMetricData(&cloudwatch.GetMetricDataInput{
|
|
EndTime: aws.Time(end),
|
|
MaxDatapoints: aws.Int64(20),
|
|
StartTime: aws.Time(start),
|
|
MetricDataQueries: cq,
|
|
})
|
|
|
|
if err != nil {
|
|
return 0, fmt.Errorf("error requesting cloudwatch: %s", err.Error())
|
|
}
|
|
|
|
mr := res.MetricDataResults
|
|
if len(mr) < 1 {
|
|
return 0, fmt.Errorf("invalid response: %s: %w", res.String(), ErrNoValuesFound)
|
|
}
|
|
|
|
vs := mr[0].Values
|
|
if len(vs) < 1 {
|
|
return 0, fmt.Errorf("invalid reponse %s: %w", res.String(), ErrNoValuesFound)
|
|
}
|
|
|
|
return aws.Float64Value(vs[0]), nil
|
|
}
|
|
|
|
// IsOnline calls GetMetricData endpoint with the empty query
|
|
// and returns an error if the returned status code is NOT http.StatusBadRequests.
|
|
// For example, if the flagger does not have permission to perform `cloudwatch:GetMetricData`,
|
|
// the returned status code would be http.StatusForbidden
|
|
func (p *CloudWatchProvider) IsOnline() (bool, error) {
|
|
_, err := p.client.GetMetricData(&cloudwatch.GetMetricDataInput{
|
|
EndTime: aws.Time(time.Time{}),
|
|
MetricDataQueries: []*cloudwatch.MetricDataQuery{},
|
|
StartTime: aws.Time(time.Time{}),
|
|
})
|
|
|
|
if err == nil {
|
|
return true, nil
|
|
}
|
|
|
|
ae, ok := err.(awserr.RequestFailure)
|
|
if !ok {
|
|
return false, fmt.Errorf("unexpected error: %v", err)
|
|
} else if ae.StatusCode() != http.StatusBadRequest {
|
|
return false, fmt.Errorf("unexpected status code: %v", ae)
|
|
}
|
|
return true, nil
|
|
}
|