diff options
author | Petr Jelinek | 2013-01-24 13:04:31 +0000 |
---|---|---|
committer | Petr Jelinek | 2013-01-24 13:04:31 +0000 |
commit | c5155488cac0c73774aa4e2788f34f37e28818c2 (patch) | |
tree | 196433412e6bfcd7c4c79429ada2d249f510ed3b /python/pgq/consumer.py | |
parent | 37d41a17cd2227fd73d2e8aa11ecb70b06948d32 (diff) |
Initialize Event properly in BaseConsumer
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] |