Use port from AppBinding (#461)

/cherry-pick Signed-off-by: Emruz Hossain <[email protected]> Ref: stashed/stash#1272
stashed · Dec 17, 2020 · b418cef · b418cef
1 parent 33294d8
commit b418cef
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pkg/backup.go b/pkg/backup.go
@@ -221,6 +221,10 @@ func (opt *postgresOptions) backupPostgreSQL(targetRef api_v1beta1.TargetRef) (*
 			"-h", appBinding.Spec.ClientConfig.Service.Name,
 		},
 	}
+	// if port is specified, append port in the arguments
+	if appBinding.Spec.ClientConfig.Service.Port != 0 {
+		opt.backupOptions.StdinPipeCommand.Args = append(opt.backupOptions.StdinPipeCommand.Args, fmt.Sprintf("--port=%d", appBinding.Spec.ClientConfig.Service.Port))
+	}
 	for _, arg := range strings.Fields(opt.pgArgs) {
 		opt.backupOptions.StdinPipeCommand.Args = append(opt.backupOptions.StdinPipeCommand.Args, arg)
 	}

diff --git a/pkg/restore.go b/pkg/restore.go
@@ -18,6 +18,7 @@ package pkg
 
 import (
 	"context"
+	"fmt"
 	"path/filepath"
 	"strings"
 
@@ -182,6 +183,10 @@ func (opt *postgresOptions) restorePostgreSQL(targetRef api_v1beta1.TargetRef) (
 			"-h", appBinding.Spec.ClientConfig.Service.Name,
 		},
 	}
+	// if port is specified, append port in the arguments
+	if appBinding.Spec.ClientConfig.Service.Port != 0 {
+		opt.dumpOptions.StdoutPipeCommand.Args = append(opt.dumpOptions.StdoutPipeCommand.Args, fmt.Sprintf("--port=%d", appBinding.Spec.ClientConfig.Service.Port))
+	}
 	for _, arg := range strings.Fields(opt.pgArgs) {
 		opt.dumpOptions.StdoutPipeCommand.Args = append(opt.dumpOptions.StdoutPipeCommand.Args, arg)
 	}