add a thread name env var as well

Author: jbr

src/rt/mod.rs

@@ -1,5 +1,6 @@
 //! The runtime.
 
+use std::env;
 use std::thread;
 
 use once_cell::sync::Lazy;
@@ -12,14 +13,20 @@ pub struct Runtime {}
 /// The global runtime.
 pub static RUNTIME: Lazy<Runtime> = Lazy::new(|| {
     // Create an executor thread pool.
-    let num_threads = std::env::var("ASYNC_STD_RT_THREADS")
-        .map(|env| env.parse().expect("ASYNC_STD_RT_THREADS must be a number"))
+
+    let thread_count = env::var("ASYNC_STD_THREAD_COUNT")
+        .map(|env| {
+            env.parse()
+                .expect("ASYNC_STD_THREAD_COUNT must be a number")
+        })
         .unwrap_or_else(|_| num_cpus::get())
         .max(1);
 
-    for _ in 0..num_threads {
+    let thread_name = env::var("ASYNC_STD_THREAD_NAME").unwrap_or("async-std/runtime".to_string());
+
+    for _ in 0..thread_count {
         thread::Builder::new()
-            .name("async-std/runtime".to_string())
+            .name(thread_name.clone())
             .spawn(|| smol::run(future::pending::<()>()))
             .expect("cannot start a runtime thread");
     }