From a7e2f175b4797172237eb8ce1990c4b4e210eb6c Mon Sep 17 00:00:00 2001
From: NoiseByNorthwest <noisebynorthwest@gmail.com>
Date: Wed, 26 Jun 2013 11:55:04 +0200
Subject: [PATCH] Adds connection manager (= global connection pool) to avoid
 duplicate connections to the same host when dealing with multiple nsqphp
 objects

---
 src/nsqphp/Connection/ConnectionManager.php | 19 +++++++++++++++++
 src/nsqphp/nsqphp.php                       | 23 +++++++++++++--------
 2 files changed, 33 insertions(+), 9 deletions(-)
 create mode 100644 src/nsqphp/Connection/ConnectionManager.php

diff --git a/src/nsqphp/Connection/ConnectionManager.php b/src/nsqphp/Connection/ConnectionManager.php
new file mode 100644
index 0000000..4640ac2
--- /dev/null
+++ b/src/nsqphp/Connection/ConnectionManager.php
@@ -0,0 +1,19 @@
+<?php
+
+namespace nsqphp\Connection;
+
+class ConnectionManager extends ConnectionPool
+{
+    private function __construct()
+    {
+    }
+
+    public static function getInstance()
+    {
+        static $instance;
+        if (!$instance) {
+            $instance = new self;
+        }
+        return $instance;
+    }
+}
diff --git a/src/nsqphp/nsqphp.php b/src/nsqphp/nsqphp.php
index 59043a4..9ed9c14 100644
--- a/src/nsqphp/nsqphp.php
+++ b/src/nsqphp/nsqphp.php
@@ -219,21 +219,26 @@ public function publishTo($hosts, $cl = NULL)
         if (!is_array($hosts)) {
             $hosts = explode(',', $hosts);
         }
+        $cm = Connection\ConnectionManager::getInstance();
         foreach ($hosts as $h) {
             if (strpos($h, ':') === FALSE) {
                 $h .= ':4150';
             }
             
             $parts = explode(':', $h);
-            $conn = new Connection\Connection(
-                    $parts[0],
-                    isset($parts[1]) ? $parts[1] : NULL,
-                    $this->connectionTimeout,
-                    $this->readWriteTimeout,
-                    $this->readWaitTimeout,
-                    FALSE,      // blocking
-                    array($this, 'connectionCallback')
-                    );
+            $conn = $cm->find($h);
+            if (!$conn) {
+                $conn = new Connection\Connection(
+                        $parts[0],
+                        isset($parts[1]) ? $parts[1] : NULL,
+                        $this->connectionTimeout,
+                        $this->readWriteTimeout,
+                        $this->readWaitTimeout,
+                        FALSE,      // blocking
+                        array($this, 'connectionCallback')
+                        );
+                $cm->add($conn);
+            }
             $this->pubConnectionPool->add($conn);
         }