import os
import asyncio

from app.services.common.service_base_async import ServiceBaseAsync


PRODUCE_BURST = int(os.getenv("PRODUCE_BURST", "10"))
PRODUCE_ONCE  = os.getenv("PRODUCE_ONCE", "true").lower() == "true"
EVENT_TYPE    = os.getenv("EVENT_TYPE", "db-mongo")

PROCESS_SEC = 10


async def produce(svc: ServiceBaseAsync):
    await asyncio.sleep(PROCESS_SEC)
    print(f"Parser Payment Producer produced {len([1,2])} events to '{svc.produce_key}'")


async def handle_from_parser(svc: ServiceBaseAsync, job):
    await asyncio.sleep(PROCESS_SEC)
    print("Parser Payment Consumer from parser:", job)
    await svc.ack_current()
    return


async def handle_from_mail(svc: ServiceBaseAsync, job):
    await asyncio.sleep(PROCESS_SEC)
    print("Parser Payment Consumer from mail:", job)
    await svc.ack_current()
    return


async def consume_default(svc: ServiceBaseAsync, job):
    await asyncio.sleep(PROCESS_SEC)
    print("Parser Payment Consumer default:", job)
    await svc.ack_current()
    return


if __name__ == "__main__":

    svc = ServiceBaseAsync(produce_fn=produce, consume_fn=consume_default, handlers={"parser.publish": handle_from_parser, "mail.publish": handle_from_mail})
    asyncio.run(svc.run())