import asyncio

from taskiq import InMemoryBroker

from taskiq_pipelines import EMPTY_PARAM_NAME, Pipeline, PipelineMiddleware

broker = InMemoryBroker()
broker.add_middlewares(PipelineMiddleware())


@broker.task
async def add_one(value: int) -> int:
    return value + 1


@broker.task
async def do_nothing() -> None:
    print("I'm doing nothing.")


async def main() -> None:
    # Never forget to call startup in the beginning.
    await broker.startup()

    pipe = (
        Pipeline(broker, add_one).call_next(do_nothing, param_name=EMPTY_PARAM_NAME)
    )
    task = await pipe.kiq(2)

    await asyncio.sleep(2)


if __name__ == "__main__":
    asyncio.run(main())