make db host pluggable

This commit is contained in:
Seán C McCord 2018-07-18 02:40:27 -04:00
parent 3434c3100e
commit 426e7415d7

View file

@ -19,6 +19,7 @@ func Connect() error {
return nil return nil
} }
var host = "localhost"
var username = "nobody" var username = "nobody"
var password string var password string
@ -30,10 +31,17 @@ func Connect() error {
username = os.Getenv("COCKROACH_USER") username = os.Getenv("COCKROACH_USER")
password = os.Getenv("COCKROACH_PASS") password = os.Getenv("COCKROACH_PASS")
} }
if os.Getenv("COCKROACH_HOST") != "" {
host = os.Getenv("COCKROACH_HOST")
}
dsn := fmt.Sprintf("postgresql://%s@localhost:26257/cycore?sslmode=disabled", username) dsn := fmt.Sprintf("postgresql://%s@%s:26257/cycore?sslmode=disabled", username, host)
if os.Getenv("KUBERNETES_SERVICE_HOST") != "" { if os.Getenv("KUBERNETES_SERVICE_HOST") != "" {
dsn = fmt.Sprintf("postgresql://%s:%s@cockroachdb-public.db.svc.cluster.local:26257/cycore?sslmode=verify-full&sslrootcert=/var/run/secrets/kubernetes.io/serviceaccount/ca.crt&sslcert=/cockroach-certs/cert&sslkey=/cockroach-certs/key", username, password) if host == "localhost" {
host = "cockroachdb-public.db.svc.cluster.local"
}
dsn = fmt.Sprintf("postgresql://%s:%s@%s:26257/cycore?sslmode=verify-full&sslrootcert=/var/run/secrets/kubernetes.io/serviceaccount/ca.crt&sslcert=/cockroach-certs/cert&sslkey=/cockroach-certs/key", username, password, host)
} }
if os.Getenv("DSN") != "" { if os.Getenv("DSN") != "" {
dsn = os.Getenv("DSN") dsn = os.Getenv("DSN")