Merge pull request #4 from tavallaie/main

Transfer `pgmq` package ownership and fix delayed message bug

Author: tavallaie

README.md

@@ -5,13 +5,13 @@
 Install with `pip` from pypi.org:
 
 ```bash
-pip install pgmq-py
+pip install pgmq
 ```
 
 To use the async version, install with the optional dependencies:
 
 ```bash
-pip install pgmq-py[async]
+pip install pgmq[async]
 ```
 
 Dependencies:
@@ -41,7 +41,7 @@ export PG_DATABASE=test_db
 Initialize a connection to Postgres using environment variables:
 
 ```python
-from pgmq_py import PGMQueue, Message
+from pgmq import PGMQueue, Message
 
 queue = PGMQueue()
 ```
@@ -51,7 +51,7 @@ queue = PGMQueue()
 Initialization for the async version requires an explicit call of the initializer:
 
 ```python
-from pgmq_py.async_queue import PGMQueue
+from pgmq.async_queue import PGMQueue
 
 async def main():
     queue = PGMQueue()
@@ -63,7 +63,7 @@ Then, the interface is exactly the same as the sync version.
 ### Initialize a connection to Postgres without environment variables
 
 ```python
-from pgmq_py import PGMQueue, Message
+from pgmq import PGMQueue, Message
 
 queue = PGMQueue(
     host="0.0.0.0",
@@ -260,13 +260,13 @@ queue = PGMQueue(
 
 # Using Transactions
 
-To perform multiple operations within a single transaction, use the `@transaction` decorator from the `pgmq_py.decorators` module. 
+To perform multiple operations within a single transaction, use the `@transaction` decorator from the `pgmq.decorators` module. 
 This ensures that all operations within the function are executed within the same transaction and are either committed together or rolled back if an error occurs.
 
 First, import the transaction decorator:
 
 ```python
-from pgmq_py.decorators import transaction
+from pgmq.decorators import transaction
 ```
 
 ### Example: Transactional Operation

benches/bench.py

@@ -6,7 +6,7 @@
 from typing import Optional
 import pandas as pd
 from sqlalchemy import create_engine, text
-from pgmq_py import PGMQueue
+from pgmq import PGMQueue
 from urllib.parse import urlparse
 import typer
 from benches.ops import consume, produce, queue_depth

benches/locustfile.py

@@ -1,5 +1,5 @@
 import time
-from pgmq_py import PGMQueue
+from pgmq import PGMQueue
 from locust import task, User
 
 

benches/ops.py

@@ -1,6 +1,5 @@
 import json
 import logging
-import multiprocessing
 import os
 import time
 import psycopg
@@ -183,7 +182,7 @@ def consume(
 def queue_depth(
     queue_name: str,
     connection_info: dict,
-    kill_flag: multiprocessing.Value,
+    kill_flag,
     duration_seconds: int,
 ):
     username = connection_info["username"]

benches/stats.py

@@ -7,7 +7,7 @@
 from sqlalchemy import create_engine, text
 import numpy as np
 from benches.log import write_event_log
-from pgmq_py import PGMQueue
+from pgmq import PGMQueue
 
 
 def stack_events(

example/example_app_async.py

@@ -1,6 +1,6 @@
 import asyncio
-from pgmq_py.async_queue import PGMQueue
-from pgmq_py.decorators import async_transaction as transaction
+from pgmq.async_queue import PGMQueue
+from pgmq.decorators import async_transaction as transaction
 
 
 async def main():

example/example_app_sync.py

@@ -1,5 +1,5 @@
-from pgmq_py.queue import PGMQueue
-from pgmq_py.decorators import transaction
+from pgmq.queue import PGMQueue
+from pgmq.decorators import transaction
 
 queue = PGMQueue(
     host="localhost",

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
-name = "pgmq-py"
-version = "0.1.0"
+name = "pgmq"
+version = "1.0.0"
 description = "Python client for the PGMQ Postgres extension."
 readme = "README.md"
 license = "Apache-2.0"

src/pgmq/__init__.py

@@ -0,0 +1,4 @@
+from pgmq.queue import Message, PGMQueue  # type: ignore
+from pgmq.decorators import transaction, async_transaction
+
+__all__ = ["Message", "PGMQueue", "transaction", "async_transaction"]

src/pgmq_py/async_queue.py renamed to src/pgmq/async_queue.py

@@ -9,8 +9,8 @@
 
 from orjson import dumps, loads
 
-from pgmq_py.messages import Message, QueueMetrics
-from pgmq_py.decorators import async_transaction as transaction
+from pgmq.messages import Message, QueueMetrics
+from pgmq.decorators import async_transaction as transaction
 
 
 @dataclass