c4181/whisper-worker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

limit prefect to 3

Browse source
This commit is contained in:
Christopher Moyer 2024-12-17 23:03:55 -05:00
parent 2c03d5a9bd
commit 165521f11d
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/main.rs
View file

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