42 lines
804 B
Go
42 lines
804 B
Go
package cluster
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
_ "github.com/lib/pq"
|
|
"strings"
|
|
)
|
|
|
|
func (c *Cluster) pgConnectionString() string {
|
|
hostname := fmt.Sprintf("%s.%s.svc.cluster.local", (*c.cluster).Metadata.Name, (*c.cluster).Metadata.Namespace)
|
|
password := c.pgUsers[superuserName].password
|
|
|
|
return fmt.Sprintf("host='%s' dbname=postgres sslmode=require user='%s' password='%s'",
|
|
hostname,
|
|
superuserName,
|
|
strings.Replace(password, "$", "\\$", -1))
|
|
}
|
|
|
|
func (c *Cluster) initDbConn() error {
|
|
//TODO: concurrent safe?
|
|
if c.pgDb == nil {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
if c.pgDb == nil {
|
|
conn, err := sql.Open("postgres", c.pgConnectionString())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = conn.Ping()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.pgDb = conn
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|