PostgreSQL integration for Taskiq with support for asyncpg, psqlpy and aiopg drivers.
Depend on your preferred PostgreSQL driver, you can install this library:
# with asyncpg
pip install taskiq-postgres[asyncpg]
# with psqlpy
pip install taskiq-postgres[psqlpy]
# with aiopg
pip install taskiq-postgres[aiopg]Simple example of usage with asyncpg:
# broker.py
import asyncio
from taskiq_pg.asyncpg import AsyncpgResultBackend, AsyncpgBroker
result_backend = AsyncpgResultBackend(
dsn="postgres://postgres:postgres@localhost:5432/postgres",
)
broker = AsyncpgBroker(
dsn="postgres://postgres:postgres@localhost:5432/postgres",
).with_result_backend(result_backend)
@broker.task
async def best_task_ever() -> None:
"""Solve all problems in the world."""
await asyncio.sleep(5.5)
print("All problems are solved!")
async def main():
await broker.startup()
task = await best_task_ever.kiq()
print(await task.wait_result())
await broker.shutdown()
if __name__ == "__main__":
asyncio.run(main())Your experience with other drivers will be pretty similar. Just change the import statement and that's it.
There are too many libraries for PostgreSQL and Taskiq integration. Although they have different view on interface and different functionality. To address this issue I created this library with a common interface for most popular PostgreSQL drivers that handle similarity across functionality of result backends and brokers.