all tests pass now except pub/sub. connection_pool's get_connection now always received the command name for the next command. still need to pass keys.

Author: andymccurdy

redis/client.py

@@ -191,19 +191,6 @@ def __init__(self, host='localhost', port=6379,
                 encoding_errors=errors
                 )
 
-    #### Legacy accessors of connection information ####
-    def _get_host(self):
-        return self.connection.host
-    host = property(_get_host)
-
-    def _get_port(self):
-        return self.connection.port
-    port = property(_get_port)
-
-    def _get_db(self):
-        return self.connection.db
-    db = property(_get_db)
-
     def pipeline(self, transaction=True):
         """
         Return a new pipeline object that can queue multiple commands for
@@ -230,9 +217,9 @@ def lock(self, name, timeout=None, sleep=0.1):
 
     #### COMMAND EXECUTION AND PROTOCOL PARSING ####
     def execute_command(self, *args, **options):
-        connection = self.connection_pool.get_connection()
+        command_name = args[0]
+        connection = self.connection_pool.get_connection(command_name)
         try:
-            command_name = args[0]
             subscription_command = command_name in self.SUBSCRIPTION_COMMANDS
             if self.subscribed and not subscription_command:
                 raise PubSubError("Cannot issue commands other than SUBSCRIBE "
@@ -1121,6 +1108,9 @@ def hvals(self, name):
         "Return the list of values within hash ``name``"
         return self.execute_command('HVALS', name)
 
+    def pubsub(self):
+        return PubSub(self.connection_pool)
+
 
     # channels
     def psubscribe(self, patterns):
@@ -1236,7 +1226,7 @@ def execute_command(self, *args, **options):
         return self
 
     def _execute_transaction(self, commands):
-        connection = self.connection_pool.get_connection()
+        connection = self.connection_pool.get_connection('MULTI')
         try:
             all_cmds = ''.join(starmap(connection.pack_command,
                                        [args for args, options in commands]))
@@ -1272,7 +1262,7 @@ def _execute_transaction(self, commands):
 
     def _execute_pipeline(self, commands):
         # build up all commands into a single request to increase network perf
-        connection = self.connection_pool.get_connection()
+        connection = self.connection_pool.get_connection('MULTI')
         try:
             all_cmds = ''.join(starmap(connection.pack_command,
                                        [args for args, options in commands]))

redis/connection.py

@@ -244,16 +244,25 @@ def __init__(self, connection_class=Connection, **kwargs):
         self.connection_class = connection_class
         self.kwargs = kwargs
         self._connection = None
+        self._in_use = False
 
-    def get_connection(self, *args, **kwargs):
+    def copy(self):
+        "Return a new instance of this class with the same parameters"
+        return self.__class__(self.connection_class, **self.kwargs)
+
+    def get_connection(self, command_name, *keys):
         "Get a connection from the pool"
+        if self._in_use:
+            raise ConnectionError("Connection already in-use")
         if not self._connection:
             self._connection = self.connection_class(**self.kwargs)
+        self._in_use = True
         return self._connection
 
     def release(self, connection):
         "Releases the connection back to the pool"
-        pass
+        assert self._connection == connection
+        self._in_use = False
 
     def disconnect(self):
         "Disconnects all connections in the pool"