Add some debug logging

Author: Dov Alperin

pkg/flypg/node.go

@@ -170,18 +170,18 @@ func (n *Node) PostInit() error {
 		return fmt.Errorf("failed to query current primary: %s", err)
 	}
 
-	conn, err := n.NewLocalConnection(context.TODO())
-	if err != nil {
-		return err
-	}
-
 	switch primaryIP {
 	case "":
 		// Check if we can be a primary
 		if !n.ValidPrimary() {
 			return fmt.Errorf("no primary to follow and can't configure self as primary because primary region is '%s' and we are in '%s'", n.Region, os.Getenv("PRIMARY_REGION"))
 		}
 
+		conn, err := n.NewLocalConnection(context.TODO())
+		if err != nil {
+			return err
+		}
+
 		// Initialize ourselves as the primary.
 		if err := n.createRequiredUsers(conn); err != nil {
 			return fmt.Errorf("failed to create required users: %s", err)
@@ -214,10 +214,16 @@ func (n *Node) PostInit() error {
 		// If we are here, we are a standby or a demoted primary who needs
 		// to be reconfigured as a standby.
 
+		conn, err := n.NewRepLocalConnection(context.TODO())
+		if err != nil {
+			return err
+		}
+
 		role, err := n.currentRole(context.TODO(), conn)
 		if err != nil {
 			return err
 		}
+		fmt.Printf("Reconfiguring a %s node as healthy\n", role)
 
 		if role == "primary" {
 			fmt.Println("Unregistering primary")
@@ -252,7 +258,12 @@ func (n *Node) PostInit() error {
 
 func (n *Node) NewLocalConnection(ctx context.Context) (*pgx.Conn, error) {
 	host := net.JoinHostPort(n.PrivateIP, strconv.Itoa(n.PGPort))
-	return openConnection(ctx, host, n.OperatorCredentials)
+	return openConnection(ctx, host, "postgres", n.OperatorCredentials)
+}
+
+func (n *Node) NewRepLocalConnection(ctx context.Context) (*pgx.Conn, error) {
+	host := net.JoinHostPort(n.PrivateIP, strconv.Itoa(n.PGPort))
+	return openConnection(ctx, host, "repmgr", n.ManagerCredentials)
 }
 
 func (n *Node) createRequiredUsers(conn *pgx.Conn) error {
@@ -389,11 +400,11 @@ func (n *Node) setDefaultHBA() error {
 	return nil
 }
 
-func openConnection(ctx context.Context, host string, creds Credentials) (*pgx.Conn, error) {
+func openConnection(ctx context.Context, host string, database string, creds Credentials) (*pgx.Conn, error) {
 	ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
 	defer cancel()
 
-	url := fmt.Sprintf("postgres://%s/postgres", host)
+	url := fmt.Sprintf("postgres://%s/%s", host, database)
 	conf, err := pgx.ParseConfig(url)
 	if err != nil {
 		return nil, err