priority_worker.py

import asyncio

import nats.errors
from nats import connect
from nats.aio.msg import Msg

PRIORITY = "high"


async def main():
    nc = await connect()
    js = nc.jetstream()

    hp_sub = await js.subscribe(f"worker.task.{PRIORITY}.>")
    default_sub = await js.subscribe(f"worker.task.>")

    while True:
        msg: Msg | None = None
        try:
            msg = await hp_sub.next_msg()
            task_id = msg.subject
        except nats.errors.TimeoutError:
            try:
                msg = await default_sub.next_msg()
            except nats.errors.TimeoutError:
                continue
        # handle(msg)


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