Skip to content

feat: add awsiamrds db auth driver #12566

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 21 commits into from
Mar 20, 2024
Prev Previous commit
Next Next commit
fix function calls
  • Loading branch information
f0ssel committed Mar 19, 2024
commit eba4c6b04e87a6b3f39792d9f90d7e8431fccdbf
2 changes: 1 addition & 1 deletion cli/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -674,7 +674,7 @@ func (r *RootCmd) Server(newAPI func(context.Context, *coderd.Options) (*coderd.
}

if codersdk.PostgresAuth(vals.PostgresAuth) == codersdk.PostgresAuthAWSRDSIAM {
sqlDriver, err = awsrdsiam.Register(sqlDriver)
sqlDriver, err = awsrdsiam.Register(inv.Context(), sqlDriver)
if err != nil {
return xerrors.Errorf("register aws rds iam auth: %w", err)
}
Expand Down
2 changes: 1 addition & 1 deletion cli/server_createadminuser.go
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ func (r *RootCmd) newCreateAdminUserCommand() *serpent.Command {

sqlDriver := "postgres"
if codersdk.PostgresAuth(newUserPgAuth) == codersdk.PostgresAuthAWSRDSIAM {
sqlDriver, err = awsrdsiam.Register(sqlDriver)
sqlDriver, err = awsrdsiam.Register(inv.Context(), sqlDriver)
if err != nil {
return xerrors.Errorf("register aws rds iam auth: %w", err)
}
Expand Down
2 changes: 1 addition & 1 deletion coderd/database/awsrdsiam/driver_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ func TestDriver(t *testing.T) {
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
defer cancel()

sqlDriver, err := awsrdsiam.Register("postgres")
sqlDriver, err := awsrdsiam.Register(ctx, "postgres")
require.NoError(t, err)

db, err := cli.ConnectToPostgres(ctx, slogtest.Make(t, nil), sqlDriver, url)
Expand Down
6 changes: 3 additions & 3 deletions enterprise/cli/server_dbcrypt.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ func (*RootCmd) dbcryptRotateCmd() *serpent.Command {

sqlDriver := "postgres"
if codersdk.PostgresAuth(flags.PostgresAuth) == codersdk.PostgresAuthAWSRDSIAM {
sqlDriver, err = awsrdsiam.Register(sqlDriver)
sqlDriver, err = awsrdsiam.Register(inv.Context(), sqlDriver)
if err != nil {
return xerrors.Errorf("register aws rds iam auth: %w", err)
}
Expand Down Expand Up @@ -157,7 +157,7 @@ func (*RootCmd) dbcryptDecryptCmd() *serpent.Command {

sqlDriver := "postgres"
if codersdk.PostgresAuth(flags.PostgresAuth) == codersdk.PostgresAuthAWSRDSIAM {
sqlDriver, err = awsrdsiam.Register(sqlDriver)
sqlDriver, err = awsrdsiam.Register(inv.Context(), sqlDriver)
if err != nil {
return xerrors.Errorf("register aws rds iam auth: %w", err)
}
Expand Down Expand Up @@ -213,7 +213,7 @@ Are you sure you want to continue?`
var err error
sqlDriver := "postgres"
if codersdk.PostgresAuth(flags.PostgresAuth) == codersdk.PostgresAuthAWSRDSIAM {
sqlDriver, err = awsrdsiam.Register(sqlDriver)
sqlDriver, err = awsrdsiam.Register(inv.Context(), sqlDriver)
if err != nil {
return xerrors.Errorf("register aws rds iam auth: %w", err)
}
Expand Down