Skip to content

Commit a3e8045

Browse files
Merge pull request #167 from abhinavdahiya/aws_useragent
client: add cluster-api-provider-aws to UserAgent for AWS API calls
2 parents 46f4852 + 8338c4f commit a3e8045

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pkg/client/client.go

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import (
2525

2626
"github.com/aws/aws-sdk-go/aws"
2727
"github.com/aws/aws-sdk-go/aws/credentials"
28+
"github.com/aws/aws-sdk-go/aws/request"
2829
"github.com/aws/aws-sdk-go/aws/session"
2930
"github.com/aws/aws-sdk-go/service/ec2"
3031
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
@@ -154,6 +155,7 @@ func NewClient(ctrlRuntimeClient client.Client, secretName, namespace, region st
154155
if err != nil {
155156
return nil, err
156157
}
158+
s.Handlers.Build.PushBackNamed(addProviderVersionToUserAgent)
157159

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

182185
return &awsClient{
183186
ec2Client: ec2.New(s),
184187
elbClient: elb.New(s),
185188
elbv2Client: elbv2.New(s),
186189
}, nil
187190
}
191+
192+
// addProviderVersionToUserAgent is a named handler that will add cluster-api-provider-aws
193+
// version information to requests made by the AWS SDK.
194+
var addProviderVersionToUserAgent = request.NamedHandler{
195+
Name: "openshift.io/cluster-api-provider-aws",
196+
Fn: request.MakeAddToUserAgentHandler("openshift.io cluster-api-provider-aws", "dummy"),
197+
}

0 commit comments

Comments
 (0)