Chore: Use interfaces where possible (#26392)

Signed-off-by: Arve Knudsen <arve.knudsen@gmail.com>
2020-07-23 08:17:20 +02:00
parent 3a8854b1b9
commit 415638cdda
9 changed files with 23 additions and 17 deletions
@@ -142,7 +142,7 @@ func getCredentials(dsInfo *datasourceInfo) (*credentials.Credentials, error) {
 	return creds, nil
 }

-func webIdentityProvider(sess *session.Session) credentials.Provider {
+func webIdentityProvider(sess client.ConfigProvider) credentials.Provider {
 	svc := newSTSService(sess)

 	roleARN := os.Getenv("AWS_ROLE_ARN")
@@ -171,7 +171,7 @@ func ecsCredProvider(sess *session.Session, uri string) credentials.Provider {
 		func(p *endpointcreds.Provider) { p.ExpiryWindow = 5 * time.Minute })
 }

-func ec2RoleProvider(sess *session.Session) credentials.Provider {
+func ec2RoleProvider(sess client.ConfigProvider) credentials.Provider {
 	return &ec2rolecreds.EC2RoleProvider{Client: newEC2Metadata(sess), ExpiryWindow: 5 * time.Minute}
 }