diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/pgq/baseconsumer.py | 5 | ||||
| -rw-r--r-- | python/pgq/consumer.py | 3 |
2 files changed, 7 insertions, 1 deletions
diff --git a/python/pgq/baseconsumer.py b/python/pgq/baseconsumer.py index 9159106e..3ea1c6c6 100644 --- a/python/pgq/baseconsumer.py +++ b/python/pgq/baseconsumer.py @@ -285,6 +285,9 @@ class BaseConsumer(skytools.DBScript): def _launch_process_batch(self, db, batch_id, list): self.process_batch(db, batch_id, list) + def _make_event(self, queue_name, row): + return Event(queue_name, row) + def _load_batch_events_old(self, curs, batch_id): """Fetch all events for this batch.""" @@ -298,7 +301,7 @@ class BaseConsumer(skytools.DBScript): # map them to python objects ev_list = [] for r in rows: - ev = Event(self.queue_name, r) + ev = self._make_event(self.queue_name, r) ev_list.append(ev) return ev_list diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py index 294519d8..10cb0909 100644 --- a/python/pgq/consumer.py +++ b/python/pgq/consumer.py @@ -60,6 +60,9 @@ class Consumer(BaseConsumer): _batch_walker_class = RetriableBatchWalker + def _make_event(self, queue_name, row): + return RetriableWalkerEvent(self, queue_name, row) + def _flush_retry(self, curs, batch_id, list): """Tag retry events.""" |