taskiq_nats_broker_async_task_quickstart.py

1import asyncio
2from taskiq_nats import NatsBroker
3from taskiq import Context, TaskiqDepends
4
5# Any broker can be used here. 
6# For this example, we'll use NatsBroker.
7broker = NatsBroker("nats://localhost:4222")
8
9@broker.task
10async def add_one(value: int) -> int:
11    return value + 1
12
13async def main():
14    # Start the broker
15    await broker.startup()
16
17    # Send the task to the worker
18    task = await add_one.kiq(1)
19    
20    # Wait for the result (requires a result backend)
21    result = await task.wait_result()
22    print(f"Task result: {result.return_value}")
23
24    # Shutdown the broker
25    await broker.shutdown()
26
27if __name__ == "__main__":
28    asyncio.run(main())