Print tokio task trace in s3_object_apply_fns.rs.

azriel91 · Oct 11, 2024 · 849b314 · 849b314
1 parent d601942
commit 849b314
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/.cargo/config.toml b/.cargo/config.toml
@@ -50,3 +50,6 @@ envman_build_release = [
 
 [env]
 CLICOLOR_FORCE = "1"
+
+[target.'cfg(not(target_arch = "wasm32"))']
+rustflags = ["--cfg", "tokio_unstable", "--cfg", "tokio_taskdump"]
diff --git a/examples/envman/src/items/peace_aws_s3_object/s3_object_apply_fns.rs b/examples/envman/src/items/peace_aws_s3_object/s3_object_apply_fns.rs
@@ -163,6 +163,22 @@ where
                                 })?;
                             base64::engine::general_purpose::STANDARD.encode(bytes)
                         };
+                        #[cfg(not(target_arch = "wasm32"))]
+                        {
+                            let handle = tokio::runtime::Handle::current();
+                            if let Ok(dump) = tokio::time::timeout(
+                                tokio::time::Duration::from_secs(2),
+                                handle.dump(),
+                            )
+                            .await
+                            {
+                                for (i, task) in dump.tasks().iter().enumerate() {
+                                    let trace = task.trace();
+                                    eprintln!("==== Task {i}: ===");
+                                    eprintln!("{trace}\n");
+                                }
+                            }
+                        }
                         let put_object_output = client
                             .put_object()
                             .bucket(bucket_name)