40 lines
987 B
Python
40 lines
987 B
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
import dramatiq
|
|
|
|
from mev_inspect.queue.broker import connect_broker
|
|
from mev_inspect.queue.middleware import (
|
|
AsyncMiddleware,
|
|
DbMiddleware,
|
|
InspectorMiddleware,
|
|
)
|
|
from mev_inspect.queue.tasks import (
|
|
HIGH_PRIORITY,
|
|
HIGH_PRIORITY_QUEUE,
|
|
LOW_PRIORITY,
|
|
LOW_PRIORITY_QUEUE,
|
|
backfill_export_task,
|
|
inspect_many_blocks_task,
|
|
realtime_export_task,
|
|
)
|
|
|
|
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
|
|
broker = connect_broker()
|
|
broker.add_middleware(DbMiddleware())
|
|
broker.add_middleware(AsyncMiddleware())
|
|
broker.add_middleware(InspectorMiddleware(os.environ["RPC_URL"]))
|
|
dramatiq.set_broker(broker)
|
|
|
|
dramatiq.actor(
|
|
inspect_many_blocks_task, queue_name=LOW_PRIORITY_QUEUE, priority=LOW_PRIORITY
|
|
)
|
|
dramatiq.actor(
|
|
backfill_export_task, queue_name=LOW_PRIORITY_QUEUE, priority=LOW_PRIORITY_QUEUE
|
|
)
|
|
dramatiq.actor(
|
|
realtime_export_task, queue_name=HIGH_PRIORITY_QUEUE, priority=HIGH_PRIORITY
|
|
)
|