diff --git a/loader.py b/loader.py index 5a17347..cbf122b 100644 --- a/loader.py +++ b/loader.py @@ -198,6 +198,7 @@ def main(): logging.info("connected to rabbitmq") ch = conn.channel() + ch.basic_qos(prefetch_count=2 * args.max_batch_size) ch.queue_declare(args.rabbitmq_queue, durable=True) ch.queue_bind(args.rabbitmq_queue, args.rabbitmq_exchange, "#")