diff options
author | Marko Kreen | 2013-06-20 17:39:26 +0000 |
---|---|---|
committer | Marko Kreen | 2013-06-20 17:39:26 +0000 |
commit | 53151b6eb4413e2af98ce3bf1e73008ad2ff3710 (patch) | |
tree | 0228e6e8cb036c46716a45fe6c8f43da9c6ac335 | |
parent | 9e72e088a5007d55e8b3f5e982e3c52c855feffc (diff) |
pgq: Consumer class also need _make_event, for _load_batch_events_old()
-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.""" |