i have big dumb

Author: Absolucy

Cargo.lock

@@ -69,6 +69,7 @@ caith = { version = "4.2.4", optional = true }
 uuid = { version = "1.17", optional = true, features = ["v4", "v7", "fast-rng"] }
 cuid2 = { version = "0.1.4", optional = true }
 dhat = "0.3"
+ctor = "0.4"
 
 [features]
 default = [

Cargo.toml

@@ -6,14 +6,25 @@ static ALLOC: dhat::Alloc = dhat::Alloc;
 
 static PROFILER: LazyLock<Mutex<Option<Profiler>>> = LazyLock::new(Mutex::default);
 
-pub fn start_dhat(path: String) {
+byond_fn!(fn start_dhat(path) {
     let mut profiler = PROFILER.lock().unwrap();
     if let Some(old_profiler) = profiler.take() {
         std::mem::drop(old_profiler);
     }
     *profiler = Some(Profiler::builder().file_name(path).build());
-}
+    Some("")
+});
+
+byond_fn!(
+    fn stop_dhat() {
+        std::mem::drop(PROFILER.lock().unwrap().take());
+        Some("")
+    }
+);
 
-pub fn stop_dhat() {
-    std::mem::drop(PROFILER.lock().unwrap().take());
+#[ctor::dtor]
+fn ensure_dhat_stopped() {
+    if let Ok(mut profiler) = PROFILER.try_lock() {
+        std::mem::drop(profiler.take());
+    };
 }