Add support for aurora postgres and mysql in proxy command

common-fate · Sep 25, 2024 · 4763afd · 4763afd
1 parent 18daa6e
commit 4763afd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/granted/rds/rds.go b/pkg/granted/rds/rds.go
@@ -439,11 +439,11 @@ var proxyCommand = cli.Command{
 			var connectionString, cliString, port string
 			yellow := color.New(color.FgYellow)
 			switch commandData.GrantOutput.Database.Engine {
-			case "postgres":
+			case "postgres", "aurora-postgresql":
 				port = grab.If(overridePort != 0, strconv.Itoa(overridePort), "5432")
 				connectionString = yellow.Sprintf("postgresql://%s:[email protected]:%s/%s?sslmode=disable", commandData.GrantOutput.User.Username, port, commandData.GrantOutput.Database.Database)
 				cliString = yellow.Sprintf(`psql "postgresql://%s:[email protected]:%s/%s?sslmode=disable"`, commandData.GrantOutput.User.Username, port, commandData.GrantOutput.Database.Database)
-			case "mysql":
+			case "mysql", "aurora-mysql":
 				port = grab.If(overridePort != 0, strconv.Itoa(overridePort), "3306")
 				connectionString = yellow.Sprintf("%s:password@tcp(127.0.0.1:%s)/%s", commandData.GrantOutput.User.Username, port, commandData.GrantOutput.Database.Database)
 				cliString = yellow.Sprintf(`mysql -u %s -p'password' -h 127.0.0.1 -P %s %s`, commandData.GrantOutput.User.Username, port, commandData.GrantOutput.Database.Database)