pgq.consumer: rename next_batch_custom result fields

fields names are out-of-sync with get_batch_info()..

diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py
index 23eaa4f9331f354f1abd8f6e4c6de9e23ac18c4f..b566280710f82da07bcf970cb9e7e6dea9a816a0 100644 (file)
--- a/python/pgq/consumer.py
+++ b/python/pgq/consumer.py
@@ -294,7 +294,14 @@ class Consumer(skytools.DBScript):
     def _load_next_batch(self, curs):
         """Allocate next batch. (internal)"""
 
-        q = "select * from pgq.next_batch_custom(%s, %s, %s, %s, %s)"
+        q = """select batch_id,
+                      prev_tick_id,
+                      cur_tick_id as tick_id,
+                      cur_tick_time as batch_end,
+                      prev_tick_time as batch_start,
+                      prev_tick_event_seq as seq_start,
+                      cur_tick_event_seq as seq_end
+                 from pgq.next_batch_custom(%s, %s, %s, %s, %s)"""
         curs.execute(q, [self.queue_name, self.consumer_name,
                          self.pgq_min_lag, self.pgq_min_count, self.pgq_min_interval])
         self.batch_info = curs.fetchone()