From 0591572d8ed918db05a15ae8ebb59dba2934dd1f Mon Sep 17 00:00:00 2001
From: Randall McPherson <rlmcpherson@gmail.com>
Date: Fri, 16 May 2014 10:23:44 -0400
Subject: [PATCH] Make pool clients safe for concurrent access with mutex.

---
 client/pool.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/client/pool.go b/client/pool.go
index d5db727..23efb74 100644
--- a/client/pool.go
+++ b/client/pool.go
@@ -18,7 +18,8 @@ var (
 
 type poolClient struct {
 	*Client
-	Rate int
+	Rate  int
+	mutex sync.Mutex
 }
 
 type SelectionHandler func(map[string]*poolClient, string) string
@@ -95,6 +96,8 @@ func (pool *Pool) Remove(addr string) {
 func (pool *Pool) Do(funcname string, data []byte,
 	flag byte, h ResponseHandler) (addr, handle string, err error) {
 	client := pool.selectServer()
+	client.Lock()
+	defer client.Unlock()
 	handle, err = client.Do(funcname, data, flag, h)
 	addr = client.addr
 	return
@@ -103,6 +106,8 @@ func (pool *Pool) Do(funcname string, data []byte,
 func (pool *Pool) DoBg(funcname string, data []byte,
 	flag byte) (addr, handle string, err error) {
 	client := pool.selectServer()
+	client.Lock()
+	defer client.Unlock()
 	handle, err = client.DoBg(funcname, data, flag)
 	addr = client.addr
 	return
@@ -112,6 +117,8 @@ func (pool *Pool) DoBg(funcname string, data []byte,
 // !!!Not fully tested.!!!
 func (pool *Pool) Status(addr, handle string) (status *Status, err error) {
 	if client, ok := pool.clients[addr]; ok {
+		client.Lock()
+		defer client.Unlock()
 		status, err = client.Status(handle)
 	} else {
 		err = ErrNotFound
@@ -131,6 +138,8 @@ func (pool *Pool) Echo(addr string, data []byte) (echo []byte, err error) {
 			return
 		}
 	}
+	client.Lock()
+	defer client.Unlock()
 	echo, err = client.Echo(data)
 	return
 }