summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarko Kreen2013-06-20 17:39:26 +0000
committerMarko Kreen2013-06-20 17:39:26 +0000
commit53151b6eb4413e2af98ce3bf1e73008ad2ff3710 (patch)
tree0228e6e8cb036c46716a45fe6c8f43da9c6ac335
parent9e72e088a5007d55e8b3f5e982e3c52c855feffc (diff)
pgq: Consumer class also need _make_event, for _load_batch_events_old()
-rw-r--r--python/pgq/baseconsumer.py5
-rw-r--r--python/pgq/consumer.py3
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."""