active-matches: Put MATCHES_PER_FILE as env variable

OpenSource-Deadlock-Tools · Oct 10, 2024 · 0449d6c · 0449d6c
1 parent 612a63d
commit 0449d6c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/active-matches-scraper/docker-compose.yaml b/active-matches-scraper/docker-compose.yaml
@@ -6,6 +6,7 @@ services:
     env_file: ../.env
     environment:
       CACHE_FOLDER: /tmp
+      MATCHES_PER_FILE: 10000
     volumes:
     - tmp:/tmp
     networks:

diff --git a/active-matches-scraper/src/main.rs b/active-matches-scraper/src/main.rs
@@ -12,7 +12,8 @@ use tokio::time::sleep;
 mod rmq;
 mod s3;
 
-const MATCHES_PER_FILE: usize = 10_000;
+static MATCHES_PER_FILE: LazyLock<usize> =
+    LazyLock::new(|| std::env::var("MATCHES_PER_FILE").unwrap().parse().unwrap());
 
 static CACHE_FOLDER: LazyLock<String> =
     LazyLock::new(|| std::env::var("CACHE_FOLDER").unwrap_or("./tmp".to_string()));
@@ -35,7 +36,7 @@ async fn main() {
 
     let mut file_writer;
     loop {
-        while match_count < MATCHES_PER_FILE {
+        while match_count < *MATCHES_PER_FILE {
             let start = std::time::Instant::now();
             file_writer = match OpenOptions::new()
                 .write(true)