Fix ambiguous column references in delete and archive SQL statements (#27)

* Initial plan

* Fix ambiguous column references in delete_batch and archive_batch

Co-authored-by: jason810496 <[email protected]>

* Add aliases to delete and archive statements to avoid ambiguity

Co-authored-by: jason810496 <[email protected]>

---------

Co-authored-by: copilot-swe-agent[bot] <[email protected]>
Co-authored-by: jason810496 <[email protected]>

Author: Copilot

pgmq_sqlalchemy/operation.py

@@ -213,7 +213,7 @@ def _get_delete_statement(
         queue_name: str, msg_id: int
     ) -> Tuple[str, Dict[str, Any]]:
         """Get statement and params for delete."""
-        return "select pgmq.delete(:queue_name, :msg_id);", {
+        return "select pgmq.delete(:queue_name, :msg_id) as deleted;", {
             "queue_name": queue_name,
             "msg_id": msg_id,
         }
@@ -224,7 +224,7 @@ def _get_delete_batch_statement(
     ) -> Tuple[str, Dict[str, Any]]:
         """Get statement and params for delete_batch."""
         return (
-            "select msg_id from unnest(CAST(:msg_ids AS bigint[])) as msg_id where pgmq.delete(:queue_name, msg_id);",
+            "select t.msg_id from unnest(CAST(:msg_ids AS bigint[])) as t(msg_id) where pgmq.delete(:queue_name, t.msg_id);",
             {"queue_name": queue_name, "msg_ids": msg_ids},
         )
 
@@ -233,7 +233,7 @@ def _get_archive_statement(
         queue_name: str, msg_id: int
     ) -> Tuple[str, Dict[str, Any]]:
         """Get statement and params for archive."""
-        return "select pgmq.archive(:queue_name, :msg_id);", {
+        return "select pgmq.archive(:queue_name, :msg_id) as archived;", {
             "queue_name": queue_name,
             "msg_id": msg_id,
         }
@@ -244,7 +244,7 @@ def _get_archive_batch_statement(
     ) -> Tuple[str, Dict[str, Any]]:
         """Get statement and params for archive_batch."""
         return (
-            "select msg_id from unnest(CAST(:msg_ids AS bigint[])) as msg_id where pgmq.archive(:queue_name, msg_id);",
+            "select t.msg_id from unnest(CAST(:msg_ids AS bigint[])) as t(msg_id) where pgmq.archive(:queue_name, t.msg_id);",
             {"queue_name": queue_name, "msg_ids": msg_ids},
         )