umemcached uses disconnect()

mozilla · Aug 21, 2012 · d4bf33b · d4bf33b
1 parent 271699b
commit d4bf33b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/memcachepool/cache.py b/memcachepool/cache.py
@@ -69,6 +69,9 @@ def delete(self, key, version=None):
             conn.delete(key)
 
     def get_many(self, keys, version=None):
+        if keys == {}:
+            return {}
+
         new_keys = map(lambda x: self.make_key(x, version=version), keys)
         with self._pool.reserve() as conn:
             ret = conn.get_multi(new_keys)

diff --git a/memcachepool/pool.py b/memcachepool/pool.py
@@ -52,7 +52,7 @@ def _checkout_connection(self):
                     return ts, client
                 # Otherwise, the connection is stale.
                 # Close it, push an empty slot onto the queue, and retry.
-                client.disconnect_all()
+                client.disconnect()
                 self.clients.put(EMPTY_SLOT)
                 continue