pgq: Consumer class also need _make_event, for _load_batch_events_old()

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."""