Added ability to choose user to log in as
This commit is contained in:
@@ -41,10 +41,10 @@ func Connect(options map[string]string) (c *conn, err error) {
|
|||||||
var database string
|
var database string
|
||||||
|
|
||||||
msg := newStartupMessage()
|
msg := newStartupMessage()
|
||||||
|
msg.options["user"], _ = options["user"]
|
||||||
if database, present = options["database"]; present {
|
if database, present = options["database"]; present {
|
||||||
msg.options["database"] = database
|
msg.options["database"] = database
|
||||||
}
|
}
|
||||||
msg.options["user"] = "jack"
|
|
||||||
c.txStartupMessage(msg)
|
c.txStartupMessage(msg)
|
||||||
|
|
||||||
for {
|
for {
|
||||||
|
|||||||
+2
-2
@@ -5,7 +5,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func TestConnect(t *testing.T) {
|
func TestConnect(t *testing.T) {
|
||||||
conn, err := Connect(map[string]string{"socket": "/private/tmp/.s.PGSQL.5432", "database": "pgx_test"})
|
conn, err := Connect(map[string]string{"socket": "/private/tmp/.s.PGSQL.5432", "user": "pgx", "database": "pgx_test"})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("Unable to establish connection")
|
t.Fatal("Unable to establish connection")
|
||||||
}
|
}
|
||||||
@@ -35,7 +35,7 @@ func TestConnect(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestQuery(t *testing.T) {
|
func TestQuery(t *testing.T) {
|
||||||
conn, err := Connect(map[string]string{"socket": "/private/tmp/.s.PGSQL.5432"})
|
conn, err := Connect(map[string]string{"socket": "/private/tmp/.s.PGSQL.5432", "user": "pgx", "database": "pgx_test"})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("Unable to establish connection")
|
t.Fatal("Unable to establish connection")
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user