11package awsrdsiam
22
33import (
4+ "context"
45 "database/sql"
56 "database/sql/driver"
67 "fmt"
78 "net/url"
89
9- "github.com/aws/aws-sdk-go/aws/session"
10- "github.com/aws/aws-sdk-go/service/rds/rdsutils"
10+ "github.com/aws/aws-sdk-go-v2/aws"
11+ "github.com/aws/aws-sdk-go-v2/config"
12+ "github.com/aws/aws-sdk-go-v2/feature/rds/auth"
1113 "golang.org/x/xerrors"
1214)
1315
1416type awsRdsIamDriver struct {
1517 parent driver.Driver
16- sess * session. Session
18+ cfg aws. Config
1719}
1820
1921var _ driver.Driver = & awsRdsIamDriver {}
2022
2123// Register initializes and registers our aws rds iam wrapped database driver.
22- func Register (parentName string ) (string , error ) {
23- sess , err := session . NewSession ( )
24+ func Register (ctx context. Context , parentName string ) (string , error ) {
25+ cfg , err := config . LoadDefaultConfig ( ctx )
2426 if err != nil {
25- return "" , xerrors . Errorf ( "creating aws session: %w" , err )
27+ return "" , err
2628 }
2729
2830 db , err := sql .Open (parentName , "" )
@@ -31,25 +33,25 @@ func Register(parentName string) (string, error) {
3133 }
3234
3335 // create a new aws rds iam driver
34- d := newDriver (db .Driver (), sess )
36+ d := newDriver (db .Driver (), cfg )
3537 name := fmt .Sprintf ("%s-awsrdsiam" , parentName )
3638 sql .Register (fmt .Sprintf ("%s-awsrdsiam" , parentName ), d )
3739
3840 return name , nil
3941}
4042
4143// newDriver will create a new *AwsRdsIamDriver using the environment aws session.
42- func newDriver (parentDriver driver.Driver , sess * session. Session ) * awsRdsIamDriver {
44+ func newDriver (parentDriver driver.Driver , cfg aws. Config ) * awsRdsIamDriver {
4345 return & awsRdsIamDriver {
4446 parent : parentDriver ,
45- sess : sess ,
47+ cfg : cfg ,
4648 }
4749}
4850
4951// Open creates a new connection to the database using the provided name.
5052func (d * awsRdsIamDriver ) Open (name string ) (driver.Conn , error ) {
5153 // set password with signed aws authentication token for the rds instance
52- nURL , err := getAuthenticatedURL (d .sess , name )
54+ nURL , err := getAuthenticatedURL (d .cfg , name )
5355 if err != nil {
5456 return nil , xerrors .Errorf ("assigning authentication token to url: %w" , err )
5557 }
@@ -63,15 +65,15 @@ func (d *awsRdsIamDriver) Open(name string) (driver.Conn, error) {
6365 return conn , nil
6466}
6567
66- func getAuthenticatedURL (sess * session. Session , dbURL string ) (string , error ) {
68+ func getAuthenticatedURL (cfg aws. Config , dbURL string ) (string , error ) {
6769 nURL , err := url .Parse (dbURL )
6870 if err != nil {
6971 return "" , xerrors .Errorf ("parsing dbURL: %w" , err )
7072 }
7173
7274 // generate a new rds session auth tokenized URL
7375 rdsEndpoint := fmt .Sprintf ("%s:%s" , nURL .Hostname (), nURL .Port ())
74- token , err := rdsutils .BuildAuthToken (rdsEndpoint , * sess . Config . Region , nURL .User .Username (), sess . Config .Credentials )
76+ token , err := auth .BuildAuthToken (context . Background (), rdsEndpoint , cfg . Region , nURL .User .Username (), cfg .Credentials )
7577 if err != nil {
7678 return "" , xerrors .Errorf ("building rds auth token: %w" , err )
7779 }
0 commit comments