From ec37a448d667ddac7d4c15eb4465f49de23afd0b Mon Sep 17 00:00:00 2001
From: Francois Hardrouyere <hello@franfran.dev>
Date: Fri, 5 Jul 2024 22:49:57 +0200
Subject: [PATCH] parse `TOKIO_CONSOLE_BUFFER_CAPACITY` and set on default env

---
 console-subscriber/src/builder.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/console-subscriber/src/builder.rs b/console-subscriber/src/builder.rs
index 85406c8ec..25eadcf21 100644
--- a/console-subscriber/src/builder.rs
+++ b/console-subscriber/src/builder.rs
@@ -333,6 +333,10 @@ impl Builder {
             self.recording_path = Some(path.into());
         }
 
+        if let Some(capacity) = usize_from_env("TOKIO_CONSOLE_BUFFER_CAPACITY") {
+            self.event_buffer_capacity = capacity;
+        }
+
         self
     }
 
@@ -765,3 +769,14 @@ fn duration_from_env(var_name: &str) -> Option<Duration> {
         ),
     }
 }
+
+fn usize_from_env(var_name: &str) -> Option<usize> {
+    let var = std::env::var(var_name).ok()?;
+    match var.parse::<usize>() {
+        Ok(num) => Some(num),
+        Err(e) => panic!(
+            "failed to parse a usize from `{}={:?}`: {}",
+            var_name, var, e
+        ),
+    }
+}