Fix vt naming for set_vt

Author: jason810496

pgmq_sqlalchemy/queue.py

@@ -812,14 +812,12 @@ def read_with_poll(
             queue_name, vt, qty, max_poll_seconds, poll_interval_ms
         )
 
-    def _set_vt_sync(
-        self, queue_name: str, msg_id: int, vt_offset: int
-    ) -> Optional[Message]:
+    def _set_vt_sync(self, queue_name: str, msg_id: int, vt: int) -> Optional[Message]:
         """Set the visibility timeout for a message."""
         with self.session_maker() as session:
             row = session.execute(
-                text("select * from pgmq.set_vt(:queue_name,:msg_id,:vt_offset);"),
-                {"queue_name": queue_name, "msg_id": msg_id, "vt_offset": vt_offset},
+                text("select * from pgmq.set_vt(:queue_name,:msg_id,:vt);"),
+                {"queue_name": queue_name, "msg_id": msg_id, "vt": vt},
             ).fetchone()
             session.commit()
         if row is None:
@@ -829,17 +827,17 @@ def _set_vt_sync(
         )
 
     async def _set_vt_async(
-        self, queue_name: str, msg_id: int, vt_offset: int
+        self, queue_name: str, msg_id: int, vt: int
     ) -> Optional[Message]:
         """Set the visibility timeout for a message."""
         async with self.session_maker() as session:
             row = (
                 await session.execute(
-                    text("select * from pgmq.set_vt(:queue_name,:msg_id,:vt_offset);"),
+                    text("select * from pgmq.set_vt(:queue_name,:msg_id,:vt);"),
                     {
                         "queue_name": queue_name,
                         "msg_id": msg_id,
-                        "vt_offset": vt_offset,
+                        "vt": vt,
                     },
                 )
             ).fetchone()
@@ -851,7 +849,7 @@ async def _set_vt_async(
             msg_id=row[0], read_ct=row[1], enqueued_at=row[2], vt=row[3], message=row[4]
         )
 
-    def set_vt(self, queue_name: str, msg_id: int, vt_offset: int) -> Optional[Message]:
+    def set_vt(self, queue_name: str, msg_id: int, vt: int) -> Optional[Message]:
         """
         .. _set_vt_method: ref:`pgmq_sqlalchemy.PGMQueue.set_vt`
         .. |set_vt_method| replace:: :py:meth:`~pgmq_sqlalchemy.PGMQueue.set_vt`
@@ -861,7 +859,7 @@ def set_vt(self, queue_name: str, msg_id: int, vt_offset: int) -> Optional[Messa
         Args:
             queue_name (str): The name of the queue.
             msg_id (int): The message id.
-            vt_offset (int): The visibility timeout in seconds.
+            vt (int): The visibility timeout in seconds.
 
         Returns:
             |schema_message_class|_ or ``None`` if the message does not exist.
@@ -904,15 +902,15 @@ def consumer_with_backoff_retry(pgmq_client: PGMQueue, queue_name: str):
                     pgmq_client.set_vt(
                         queue_name=query_name,
                         msg_id=msg.msg_id,
-                        vt_offset=_exp_backoff_retry(msg)
+                        vt=_exp_backoff_retry(msg)
                     )
 
         """
         if self.is_async:
             return self.loop.run_until_complete(
-                self._set_vt_async(queue_name, msg_id, vt_offset)
+                self._set_vt_async(queue_name, msg_id, vt)
             )
-        return self._set_vt_sync(queue_name, msg_id, vt_offset)
+        return self._set_vt_sync(queue_name, msg_id, vt)
 
     def _pop_sync(self, queue_name: str) -> Optional[Message]:
         with self.session_maker() as session: