import logging
import time

from pyinfra.config import get_config
from pyinfra.payload_processing.processor import make_payload_processor
from pyinfra.queue.queue_manager import QueueManager

logging.basicConfig()
logger = logging.getLogger()
logger.setLevel(logging.INFO)


def json_processor_mock(data: dict):
    time.sleep(5)
    return [{"result1": "result1"}, {"result2": "result2"}]


def main():
    logger.info("Start consuming...")
    queue_manager = QueueManager(get_config())
    queue_manager.start_consuming(make_payload_processor(json_processor_mock))


if __name__ == "__main__":
    main()