count tokio tasks

src/json_rescue_v5_load.rs

@@ -10,7 +10,10 @@ use log::{error, info, trace, warn};
 use neo4rs::Graph;
 use std::sync::Arc;
 use std::{path::Path, thread::available_parallelism};
-use tokio::sync::{Mutex, Semaphore};
+use tokio::{
+    runtime::Handle,
+    sync::{Semaphore},
+};
 
 /// How many records to read from the archives before attempting insert
 // static LOAD_QUEUE_SIZE: usize = 1000;
@@ -73,19 +76,17 @@ pub async fn rip_concurrent_limited(
     info!("tgz archives found: {}", archives_count);
 
     let semaphore = Arc::new(Semaphore::new(threads)); // Semaphore to limit concurrency
-    let progress = Arc::new(Mutex::new(0u64));
     let mut tasks = vec![];
 
     for p in tgz_list.into_iter() {
         let pool = pool.clone(); // Clone pool for each task
         let semaphore = Arc::clone(&semaphore); // Clone semaphore for each task
-        let progress = Arc::clone(&progress); // Clone semaphore for each task
 
         let task = tokio::spawn(async move {
             let _permit = semaphore.acquire().await; // Acquire semaphore permit
-            let mut n = progress.lock().await;
-            info!("PROGRESS: {n}/{archives_count}");
-            *n += 1;
+            let metrics = Handle::current().metrics();
+            let n = metrics.spawned_tasks_count();
+            println!("Runtime has had {} tasks spawned", n);
             single_thread_decompress_extract(&p, &pool).await // Perform the task
         });