Bug:NATS PullSubscriber works not clear with max_workers > 1 #2563

import uuid

import asyncio
from nats.js.api import RetentionPolicy

from typing import AsyncGenerator

from fast_depends import Depends
from faststream.nats import JStream, NatsBroker, NatsMessage, PullSub

stream = JStream(
    "test",
    subjects=["test.>"],
    retention=RetentionPolicy.WORK_QUEUE,
)

broker = NatsBroker()


async def progress_sender(message: NatsMessage) -> AsyncGenerator[None]:  # noqa: RUF029, RUF100
    async def in_progress_task() -> None:
        while True:
            await asyncio.sleep(5.0)
            await message.in_progress()

    task = asyncio.create_task(in_progress_task())
    yield
    task.cancel()


@broker.subscriber(
    "test.processing",
    durable="test-worker",
    pull_sub=PullSub(batch_size=2, batch=True),
    stream=stream,
    dependencies=[Depends(progress_sender)],
    max_workers=2,
)
async def test(msg: str) -> None:
    print(f"Received message {msg}")
    await asyncio.sleep(30)


async def main() -> None:
    await broker.start()
    await asyncio.gather(
        *[broker.publish(f"{uuid.uuid4()}", "test.processing") for _ in range(4)]
    )

    while True:
        await asyncio.sleep(1)


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