Add support for Unix sockets. Fixes #14.

.travis.yml

@@ -19,4 +19,4 @@ install:
   - ln -s `pwd` $HOME/gopath/src/gopkg.in/pg.v2
 
 before_script:
-  - psql -c 'CREATE DATABASE test;' -U postgres
+  - psql -c 'CREATE DATABASE test' -U postgres

conn.go

@@ -15,15 +15,15 @@ import (
 var zeroTime = time.Time{}
 
 func dial(opt *Options) (net.Conn, error) {
-	return net.DialTimeout(
-		"tcp", net.JoinHostPort(opt.getHost(), opt.getPort()), opt.getDialTimeout())
+	return net.DialTimeout(opt.getNetwork(), opt.getAddr(), opt.getDialTimeout())
 }
 
 type conn struct {
-	opt    *Options
-	cn     net.Conn
-	br     *bufio.Reader
-	buf    *buffer
+	opt *Options
+	cn  net.Conn
+	br  *bufio.Reader
+	buf *buffer
+
 	inUse  bool
 	usedAt time.Time
 

db.go

@@ -11,6 +11,7 @@ import (
 const defaultBackoff = 100 * time.Millisecond
 
 type Options struct {
+	Network  string
 	Host     string
 	Port     string
 	User     string
@@ -28,6 +29,13 @@ type Options struct {
 	IdleCheckFrequency time.Duration
 }
 
+func (opt *Options) getNetwork() string {
+	if opt == nil || opt.Network == "" {
+		return "tcp"
+	}
+	return opt.Network
+}
+
 func (opt *Options) getHost() string {
 	if opt == nil || opt.Host == "" {
 		return "localhost"
@@ -42,6 +50,15 @@ func (opt *Options) getPort() string {
 	return opt.Port
 }
 
+func (opt *Options) getAddr() string {
+	switch opt.getNetwork() {
+	case "tcp":
+		return net.JoinHostPort(opt.getHost(), opt.getPort())
+	default:
+		return opt.getHost()
+	}
+}
+
 func (opt *Options) getUser() string {
 	if opt == nil || opt.User == "" {
 		return ""

main_test.go

@@ -20,6 +20,19 @@ import (
 	"gopkg.in/pg.v2/pgutil"
 )
 
+func TestUnixSocket(t *testing.T) {
+	db := pg.Connect(&pg.Options{
+		Network:  "unix",
+		Host:     "/var/run/postgresql/.s.PGSQL.5432",
+		User:     "postgres",
+		Database: "test",
+	})
+	_, err := db.Exec("SELECT 1")
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
 func Test(t *testing.T) { TestingT(t) }
 
 var _ = Suite(&DBTest{})