Enable shell for both mysql and postgres

apppackio · Jan 24, 2021 · 6af6cff · 6af6cff
1 parent cae1a20
commit 6af6cff
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/app/app.go b/app/app.go
@@ -75,6 +75,7 @@ type Settings struct {
 		ShellTaskFamily    string `json:"shell_task_family"`
 		S3Bucket           string `json:"s3_bucket"`
 		DumpLoadTaskFamily string `json:"dumpload_task_family"`
+		Engine             string `json:"engine"`
 	} `json:"dbutils"`
 	CodebuildProject struct {
 		Name string `json:"name"`

diff --git a/cmd/db.go b/cmd/db.go
@@ -89,6 +89,14 @@ var dbShellCmd = &cobra.Command{
 		checkErr(err)
 		err = a.LoadSettings()
 		checkErr(err)
+		var exec string
+		if a.Settings.DBUtils.Engine == "mysql" {
+			exec = "mysql"
+		} else if a.Settings.DBUtils.Engine == "postgres" {
+			exec = "psql"
+		} else {
+			checkErr(fmt.Errorf("unknoown database engine %s", a.Settings.DBUtils.Engine))
+		}
 		taskOutput, err := a.StartTask(&a.Settings.DBUtils.ShellTaskFamily, app.ShellBackgroundCommand, false)
 		checkErr(err)
 		shellTask := taskOutput.Tasks[0]
@@ -97,7 +105,7 @@ var dbShellCmd = &cobra.Command{
 		err = a.WaitForTaskRunning(shellTask)
 		checkErr(err)
 		Spinner.Stop()
-		err = a.ConnectToTask(shellTask, aws.String("entrypoint.sh psql"))
+		err = a.ConnectToTask(shellTask, aws.String(fmt.Sprintf("entrypoint.sh %s", exec)))
 		checkErr(err)
 	},
 }