Skip to content
This repository has been archived by the owner on Aug 4, 2020. It is now read-only.

Commit

Permalink
Enable failover on functions that previously lacked it
Browse files Browse the repository at this point in the history
These include:
 * get_count
 * multiget_count
 * multiget_slice
  • Loading branch information
thobbs committed Jan 25, 2011
1 parent e6a9a24 commit 4965d50
Showing 1 changed file with 16 additions and 3 deletions.
19 changes: 16 additions & 3 deletions pycassa/pool.py
Original file line number Diff line number Diff line change
Expand Up @@ -317,6 +317,7 @@ def __init__(self, pool, max_retries, *args, **kwargs):
super(ConnectionWrapper, self).__init__(*args, **kwargs)
self._pool._notify_on_connect(self)

# For testing purposes only
self._should_fail = False
self._original_meth = self.send_batch_mutate

Expand Down Expand Up @@ -433,11 +434,23 @@ def _fail_once(self, *args, **kwargs):
@_retry
def get(self, *args, **kwargs):
pass

@_retry
def get_slice(self, *args, **kwargs):
pass


@_retry
def multiget_slice(self, *args, **kwargs):
pass

@_retry
def get_count(self, *args, **kwargs):
pass

@_retry
def multiget_count(self, *args, **kwargs):
pass

@_retry
def get_range_slices(self, *args, **kwargs):
pass
Expand Down Expand Up @@ -470,7 +483,7 @@ def describe_keyspace(self, *args, **kwargs):
def get_keyspace_description(self, keyspace=None, use_dict_for_col_metadata=False):
"""
Describes the given keyspace.
If `use_dict_for_col_metadata` is ``True``, the column metadata will be stored
as a dictionary instead of a list
Expand Down

0 comments on commit 4965d50

Please sign in to comment.