Simplify consumer code and add pyproject.toml for example

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

Author: Copilot

examples/fastapi_pub_sub/README.md

@@ -30,7 +30,14 @@ docker run -d --name postgres -e POSTGRES_PASSWORD=postgres -p 5432:5432 quay.io
 
 ## Installation
 
-Install required dependencies using uv:
+Install required dependencies using uv with the example's pyproject.toml:
+
+```bash
+cd examples/fastapi_pub_sub
+uv pip install -e .
+```
+
+Or install dependencies directly:
 
 ```bash
 uv pip install fastapi uvicorn psycopg2-binary asyncpg pgmq-sqlalchemy

examples/fastapi_pub_sub/consumer.py

@@ -67,8 +67,7 @@ async def consume_messages(pgmq: PGMQueue, batch_size: int = 10, vt: int = 30):
     while True:
         try:
             # Read a batch of messages using pgmq instance method
-            async with pgmq.session_maker() as session:
-                messages = await pgmq.read_batch_async(QUEUE_NAME, vt=vt, batch_size=batch_size, session=session, commit=True)
+            messages = await pgmq.read_batch(QUEUE_NAME, vt=vt, batch_size=batch_size)
 
             if not messages:
                 logger.debug("No messages available, waiting...")
@@ -89,8 +88,7 @@ async def consume_messages(pgmq: PGMQueue, batch_size: int = 10, vt: int = 30):
             # Delete successfully processed messages using pgmq instance method
             for (msg_id, _), result in zip(tasks, results):
                 if isinstance(result, bool) and result:
-                    async with pgmq.session_maker() as session:
-                        deleted = await pgmq.delete_async(QUEUE_NAME, msg_id, session=session, commit=True)
+                    deleted = await pgmq.delete(QUEUE_NAME, msg_id)
                     if deleted:
                         logger.info(f"Deleted message {msg_id}")
                 elif isinstance(result, Exception):

examples/fastapi_pub_sub/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "fastapi-pub-sub-example"
+version = "0.1.0"
+description = "FastAPI pub/sub example using pgmq-sqlalchemy"
+readme = "README.md"
+requires-python = ">=3.9"
+dependencies = [
+    "fastapi>=0.104.0",
+    "uvicorn>=0.24.0",
+    "psycopg2-binary>=2.9.9",
+    "asyncpg>=0.29.0",
+    "pgmq-sqlalchemy>=0.1.2",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"