Merge pull request #2 from fising/master

Validate the connection when it is returned.

src/ConnectionPool.php

@@ -148,6 +148,10 @@ public function borrow()
      */
     public function return($connection): bool
     {
+        if (!$this->connector->validate($connection)) {
+            throw new \RuntimeException('Connection of unexpected type returned.');
+        }
+
         if (!$this->initialized) {
             throw new \RuntimeException('Please initialize the connection pool first, call $pool->init().');
         }

src/Connectors/ConnectorInterface.php

@@ -32,4 +32,12 @@ public function isConnected($connection): bool;
      * @return mixed
      */
     public function reset($connection, array $config);
+
+    /**
+     * Validate the connection
+     *
+     * @param mixed $connection
+     * @return bool
+     */
+    public function validate($connection): bool;
 }

src/Connectors/CoroutineMySQLConnector.php

@@ -31,4 +31,9 @@ public function reset($connection, array $config)
     {
 
     }
+
+    public function validate($connection): bool
+    {
+        return $connection instanceof MySQL;
+    }
 }

src/Connectors/CoroutinePostgreSQLConnector.php

@@ -34,4 +34,9 @@ public function reset($connection, array $config)
     {
         /**@var PostgreSQL $connection */
     }
+
+    public function validate($connection): bool
+    {
+        return $connection instanceof PostgreSQL;
+    }
 }

src/Connectors/CoroutineRedisConnector.php

@@ -45,4 +45,9 @@ public function reset($connection, array $config)
         }
         $connection->setDefer(false);
     }
+
+    public function validate($connection): bool
+    {
+        return $connection instanceof Redis;
+    }
 }

src/Connectors/PhpRedisConnector.php

@@ -45,4 +45,9 @@ public function reset($connection, array $config)
             $connection->select($config['database']);
         }
     }
+
+    public function validate($connection): bool
+    {
+        return $connection instanceof \Redis;
+    }
 }