/
connection.go
43 lines (36 loc) · 1.13 KB
/
connection.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
// Copyright (c) 2017, The dcrdata developers
// See LICENSE for details.
package dcrpg
import (
"database/sql"
"fmt"
"strings"
_ "github.com/lib/pq" // Start the PostgreSQL driver
)
// Connect opens a connection to a PostgreSQL database. The caller is
// responsible for calling Close() on the returned db when finished using it.
// The input host may be an IP address for TCP connection, or an absolute path
// to a UNIX domain socket. An empty string should be provided for UNIX sockets.
func Connect(host, port, user, pass, dbname string) (*sql.DB, error) {
var psqlInfo string
if pass == "" {
psqlInfo = fmt.Sprintf("host=%s user=%s "+
"dbname=%s sslmode=disable",
host, user, dbname)
} else {
psqlInfo = fmt.Sprintf("host=%s user=%s "+
"password=%s dbname=%s sslmode=disable",
host, user, pass, dbname)
}
// Only add port arg fot TCP connection since UNIX domain sockets (specified
// by a "/" prefix) do not have a port.
if !strings.HasPrefix(host, "/") {
psqlInfo += fmt.Sprintf(" port=%s", port)
}
db, err := sql.Open("postgres", psqlInfo)
if err != nil {
return nil, err
}
err = db.Ping()
return db, err
}