Merged pull request redis#132 from banjiewen/banjiewen.

Added socket disconnection if commands are rejected due to AOF/RDB load.

Author: andymccurdy

redis/connection.py

@@ -97,6 +97,11 @@ def read_response(self, command_name, catch_errors):
         if byte == '-':
             if response.startswith('ERR '):
                 response = response[4:]
+            if response.startswith('LOADING '):
+                # If we're loading the dataset into memory, kill the socket
+                # so we re-initialize (and re-SELECT) next time.
+                self.disconnect()
+                response = response[8:]
             raise ResponseError(response)
         # single value
         elif byte == '+':