Add user agent to AWS requests (#1890)

* Add user agent to AWS requests

Signed-off-by: Clare Liguori <liguori@amazon.com>

* Fix import spacing

Signed-off-by: Clare Liguori <liguori@amazon.com>
This commit is contained in:
Clare Liguori 2022-07-18 15:07:34 -07:00 committed by GitHub
parent 1a4e2725ec
commit 3195217f12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 1 deletions

View File

@ -16,7 +16,10 @@ package aws
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/dapr/kit/logger"
)
func GetClient(accessKey string, secretKey string, sessionToken string, region string, endpoint string) (*session.Session, error) {
@ -34,10 +37,19 @@ func GetClient(accessKey string, secretKey string, sessionToken string, region s
awsConfig = awsConfig.WithEndpoint(endpoint)
}
awsSession, err := session.NewSession(awsConfig)
awsSession, err := session.NewSessionWithOptions(session.Options{
Config: *awsConfig,
SharedConfigState: session.SharedConfigEnable,
})
if err != nil {
return nil, err
}
userAgentHandler := request.NamedHandler{
Name: "UserAgentHandler",
Fn: request.MakeAddToUserAgentHandler("dapr", logger.DaprVersion),
}
awsSession.Handlers.Build.PushBackNamed(userAgentHandler)
return awsSession, nil
}