Skip to content

Commit

Permalink
Merge pull request #167 from abhinavdahiya/aws_useragent
Browse files Browse the repository at this point in the history
client: add cluster-api-provider-aws to UserAgent for AWS API calls
  • Loading branch information
openshift-merge-robot committed Feb 26, 2019
2 parents 46f4852 + 8338c4f commit a3e8045
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions pkg/client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ import (

"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
Expand Down Expand Up @@ -154,6 +155,7 @@ func NewClient(ctrlRuntimeClient client.Client, secretName, namespace, region st
if err != nil {
return nil, err
}
s.Handlers.Build.PushBackNamed(addProviderVersionToUserAgent)

return &awsClient{
ec2Client: ec2.New(s),
Expand All @@ -178,10 +180,18 @@ func NewClientFromKeys(accessKey, secretAccessKey, region string) (Client, error
if err != nil {
return nil, err
}
s.Handlers.Build.PushBackNamed(addProviderVersionToUserAgent)

return &awsClient{
ec2Client: ec2.New(s),
elbClient: elb.New(s),
elbv2Client: elbv2.New(s),
}, nil
}

// addProviderVersionToUserAgent is a named handler that will add cluster-api-provider-aws
// version information to requests made by the AWS SDK.
var addProviderVersionToUserAgent = request.NamedHandler{
Name: "openshift.io/cluster-api-provider-aws",
Fn: request.MakeAddToUserAgentHandler("openshift.io cluster-api-provider-aws", "dummy"),
}

0 comments on commit a3e8045

Please sign in to comment.