add rabbitmq test

demml · Sep 24, 2024 · 0b3dda0 · 0b3dda0
1 parent de42531
commit 0b3dda0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/test-lints.yml b/.github/workflows/test-lints.yml
@@ -56,7 +56,7 @@ jobs:
         if: always()
         run: docker compose -f "docker-compose.yml" down
 
-  test-kafka:
+  test-kafka-rabbitmq:
     needs: test-os
     name: test kafka
 

diff --git a/src/main.rs b/src/main.rs
@@ -49,7 +49,9 @@ async fn start_main_server() -> Result<(), anyhow::Error> {
         startup_kafka(pool.clone()).await?;
     }
 
-    startup_rabbitmq(pool.clone()).await?;
+    if std::env::var("RABBITMQ_ADDR").is_ok() {
+        startup_rabbitmq(pool.clone()).await?;
+    }
 
     // run drift background task
     let num_scheduler_workers = std::env::var("NUM_SCOUTER_SCHEDULER_WORKERS")

diff --git a/src/rabbitmq/startup.rs b/src/rabbitmq/startup.rs
@@ -8,7 +8,7 @@ pub async fn startup_rabbitmq(pool: Pool<Postgres>) -> Result<(), anyhow::Error>
     info!("Starting RabbitMQ consumer");
 
     let num_rabbits = std::env::var("NUM_SCOUTER_RABBITMQ_CONSUMERS")
-        .unwrap_or_else(|_| "1".to_string())
+        .unwrap_or_else(|_| "3".to_string())
         .parse::<usize>()
         .map_err(|e| lapin::Error::from(std::io::Error::new(std::io::ErrorKind::Other, e)))?;