Skip to content

Commit

Permalink
feat: Change default prefetch_count to 1 (assuming at most one datase…
Browse files Browse the repository at this point in the history
…t being processed)
  • Loading branch information
jpmckinney committed Aug 10, 2023
1 parent f28102f commit 3fb9c82
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 7 deletions.
4 changes: 1 addition & 3 deletions contracting_process/field_level/report_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,7 @@ def create(dataset_id):
report = {}
examples = {}

for order, definition in enumerate(definitions.items()):
path, quality_checks = definition

for order, (path, quality_checks) in enumerate(definitions.items()):
report[path] = {
"examples_filled": False,
"processing_order": order,
Expand Down
5 changes: 2 additions & 3 deletions pelican/util/services.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,17 +43,16 @@ def decode(body: bytes, content_type: str | None) -> Any:
YAPW_KWARGS = {
"url": settings.RABBIT_URL,
"exchange": settings.RABBIT_EXCHANGE_NAME,
"prefetch_count": 20,
"encode": encode,
"decode": decode,
}


def consume(*args: Any, **kwargs: Any) -> None:
def consume(*args: Any, prefetch_count=1, **kwargs: Any) -> None:
"""
Consume messages from RabbitMQ.
"""
client = AsyncConsumer(*args, **kwargs, **YAPW_KWARGS)
client = AsyncConsumer(*args, prefetch_count=prefetch_count, **kwargs, **YAPW_KWARGS)
client.start()


Expand Down
2 changes: 1 addition & 1 deletion workers/check/data_item.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def start():
Perform the field-level and compiled release-level checks.
"""
bootstrap()
consume(on_message_callback=callback, queue=consume_routing_key)
consume(on_message_callback=callback, queue=consume_routing_key, prefetch_count=20)


def callback(client_state, channel, method, properties, input_message):
Expand Down

0 comments on commit 3fb9c82

Please sign in to comment.