From 165521f11d03854277c808c1cd1c5402f20e3038 Mon Sep 17 00:00:00 2001
From: Christopher Moyer <cm4181@gmail.com>
Date: Tue, 17 Dec 2024 23:03:55 -0500
Subject: [PATCH] limit prefect to 3

---
 src/main.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index d25032e..c24ba1e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -9,7 +9,7 @@ use std::sync::{Arc, Mutex};
 use chrono::{DateTime, Utc};
 use lapin::{Connection, ConnectionProperties, Consumer};
 use lapin::message::{Delivery, DeliveryResult};
-use lapin::options::{BasicAckOptions, BasicConsumeOptions};
+use lapin::options::{BasicAckOptions, BasicConsumeOptions, BasicQosOptions};
 use lapin::types::FieldTable;
 use serde::{Deserialize, Serialize};
 use log::{error, info, trace, warn};
@@ -18,7 +18,7 @@ use meilisearch_sdk::task_info::TaskInfo;
 use meilisearch_sdk::tasks::Task;
 use uuid::Uuid;
 
-#[tokio::main(flavor = "multi_thread")]
+#[tokio::main]
 async fn main() {
     let ascii = r#"
 __      ___         _              _ __                          __      __                 _
@@ -44,6 +44,7 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
         .with_reactor(tokio_reactor_trait::Tokio);
     let connection = Connection::connect(&cfg.rabbit_mq_config.connection_string, options).await.unwrap();
     let channel = connection.create_channel().await.unwrap();
+    channel.basic_qos(3, BasicQosOptions::default()).await.unwrap();
     let consumer = channel.basic_consume("transcribe", "whisper-worker", BasicConsumeOptions::default(), FieldTable::default()).await.unwrap();
 
     let processing_lock = Arc::new(Mutex::new(()));