From f64761880bf025e96e4248bec797c957cc340cf5 Mon Sep 17 00:00:00 2001
From: Vladimir Mihailenco <vladimir.webdev@gmail.com>
Date: Sat, 2 Feb 2013 12:50:43 +0200
Subject: [PATCH] Add NewUnixClient to connect to the unix sockets.

---
 redis.go      | 10 ++++++++++
 redis_test.go | 25 +++++++++++++++++++++++--
 2 files changed, 33 insertions(+), 2 deletions(-)

diff --git a/redis.go b/redis.go
index 2cd0aa7..2059907 100644
--- a/redis.go
+++ b/redis.go
@@ -32,6 +32,12 @@ func TLSConnector(addr string, tlsConfig *tls.Config) OpenConnFunc {
 	}
 }
 
+func UnixConnector(addr string) OpenConnFunc {
+	return func() (net.Conn, error) {
+		return net.DialTimeout("unix", addr, 3*time.Second)
+	}
+}
+
 func AuthSelectFunc(password string, db int64) InitConnFunc {
 	if password == "" && db < 0 {
 		return nil
@@ -182,3 +188,7 @@ func NewTLSClient(addr string, tlsConfig *tls.Config, password string, db int64)
 		AuthSelectFunc(password, db),
 	)
 }
+
+func NewUnixClient(addr string, password string, db int64) *Client {
+	return NewClient(UnixConnector(addr), nil, AuthSelectFunc(password, db))
+}
diff --git a/redis_test.go b/redis_test.go
index f8d9bdb..1cf0acf 100644
--- a/redis_test.go
+++ b/redis_test.go
@@ -52,9 +52,30 @@ func (t *RedisShutdownTest) TestShutdown(c *C) {
 
 //------------------------------------------------------------------------------
 
+type RedisConnectorTest struct{}
+
+var _ = Suite(&RedisConnectorTest{})
+
+func (t *RedisConnectorTest) TestTCPConnector(c *C) {
+	client := redis.NewTCPClient(":6379", "", -1)
+	ping := client.Ping()
+	c.Check(ping.Err(), IsNil)
+	c.Check(ping.Val(), Equals, "PONG")
+}
+
+func (t *RedisConnectorTest) TestUnixConnector(c *C) {
+	client := redis.NewUnixClient("/tmp/redis.sock", "", -1)
+	ping := client.Ping()
+	c.Check(ping.Err(), IsNil)
+	c.Check(ping.Val(), Equals, "PONG")
+}
+
+//------------------------------------------------------------------------------
+
 type RedisConnPoolTest struct {
 	openedConnCount, closedConnCount, initedConnCount int64
-	client                                            *redis.Client
+
+	client *redis.Client
 }
 
 var _ = Suite(&RedisConnPoolTest{})
@@ -2734,7 +2755,7 @@ func (t *RedisTest) TestScriptingScriptFlush(c *C) {
 
 func (t *RedisTest) TestScriptingScriptKill(c *C) {
 	scriptKill := t.client.ScriptKill()
-	c.Assert(scriptKill.Err(), ErrorMatches, "ERR No scripts in execution right now.")
+	c.Assert(scriptKill.Err(), ErrorMatches, ".*No scripts in execution right now.")
 	c.Assert(scriptKill.Val(), Equals, "")
 }