Add async methods to module trace (#2)

* Add async methods.

* Format.

Author: lgxbslgx

src/trace.rs

@@ -1,6 +1,7 @@
 //! Chrome trace output.
 
 use std::fs::File;
+use std::future::Future;
 use std::io::{BufWriter, Write};
 use std::time::Instant;
 
@@ -56,6 +57,14 @@ impl Trace {
         result
     }
 
+    async fn async_scope<T>(&mut self, name: &str, f: impl Future<Output = T>) -> T {
+        let start = Instant::now();
+        let result = f.await;
+        let end = Instant::now();
+        self.write_complete(name, 0, start, end);
+        result
+    }
+
     /*
     These functions were useful when developing, but are currently unused.
 
@@ -121,6 +130,18 @@ pub fn scope<T>(name: &'static str, f: impl FnOnce() -> T) -> T {
     }
 }
 
+#[inline]
+#[allow(static_mut_refs)]
+pub async fn async_scope<T>(name: &'static str, f: impl Future<Output = T>) -> T {
+    // Safety: accessing global mut, not threadsafe.
+    unsafe {
+        match &mut TRACE {
+            None => f.await,
+            Some(t) => t.async_scope(name, f).await,
+        }
+    }
+}
+
 pub fn close() {
     if_enabled(|t| t.close());
 }