Add priorities to queue tasks
This commit is contained in:
17
cli.py
17
cli.py
@@ -12,7 +12,12 @@ from mev_inspect.db import get_inspect_session, get_trace_session
|
||||
from mev_inspect.inspector import MEVInspector
|
||||
from mev_inspect.prices import fetch_prices, fetch_prices_range
|
||||
from mev_inspect.queue.broker import connect_broker
|
||||
from mev_inspect.queue.tasks import export_block_task, inspect_many_blocks_task
|
||||
from mev_inspect.queue.tasks import (
|
||||
BACKFILL_INSPECT_MANY_BLOCKS_PRIORITY,
|
||||
LIVE_EXPORT_BLOCK_PRIORITY,
|
||||
export_block_task,
|
||||
inspect_many_blocks_task,
|
||||
)
|
||||
from mev_inspect.s3_export import export_block
|
||||
|
||||
RPC_URL_ENV = "RPC_URL"
|
||||
@@ -102,7 +107,11 @@ async def inspect_many_blocks_command(
|
||||
@click.argument("batch_size", type=int, default=10)
|
||||
def enqueue_many_blocks_command(start_block: int, end_block: int, batch_size: int):
|
||||
broker = connect_broker()
|
||||
inspect_many_blocks_actor = dramatiq.actor(inspect_many_blocks_task, broker=broker)
|
||||
inspect_many_blocks_actor = dramatiq.actor(
|
||||
inspect_many_blocks_task,
|
||||
broker=broker,
|
||||
priority=BACKFILL_INSPECT_MANY_BLOCKS_PRIORITY,
|
||||
)
|
||||
|
||||
if start_block < end_block:
|
||||
after_block = start_block
|
||||
@@ -137,7 +146,9 @@ def fetch_all_prices():
|
||||
@click.argument("block_number", type=int)
|
||||
def enqueue_s3_export(block_number: int):
|
||||
broker = connect_broker()
|
||||
export_actor = dramatiq.actor(export_block_task, broker=broker)
|
||||
export_actor = dramatiq.actor(
|
||||
export_block_task, broker=broker, priority=LIVE_EXPORT_BLOCK_PRIORITY
|
||||
)
|
||||
logger.info(f"Sending block {block_number} export to queue")
|
||||
export_actor.send(block_number)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user