diff options
Diffstat (limited to 'python/pgq/consumer.py')
| -rw-r--r-- | python/pgq/consumer.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py index 19fbf5ef..3d78105f 100644 --- a/python/pgq/consumer.py +++ b/python/pgq/consumer.py @@ -31,12 +31,13 @@ class RetriableBatchWalker(BaseBatchWalker): """BatchWalker that returns RetriableEvents """ - _event_class = RetriableWalkerEvent - def __init__(self, curs, batch_id, queue_name, fetch_size = 300, consumer_filter = None): super(RetriableBatchWalker, self).__init__(self, curs, batch_id, queue_name, fetch_size, consumer_filter) self.status_map = {} + def _make_event(self, queue_name, row): + return RetriableWalkerEvent(self, queue_name, row) + def tag_event_done(self, event): if event.id in self.status_map: del self.status_map[event.id] |