From a5ae251fa3d62bb023dfbfc8ef692b28ff7a9a16 Mon Sep 17 00:00:00 2001
From: Filip Tibell <filip.tibell@gmail.com>
Date: Thu, 1 Feb 2024 11:18:30 +0100
Subject: [PATCH] Add missing track method to trait

---
 lib/traits.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/lib/traits.rs b/lib/traits.rs
index e724caf..d15f926 100644
--- a/lib/traits.rs
+++ b/lib/traits.rs
@@ -94,6 +94,13 @@ pub trait LuaRuntimeExt<'lua> {
         args: impl IntoLuaMulti<'lua>,
     ) -> LuaResult<ThreadId>;
 
+    /**
+        Registers the given thread to be tracked within the current runtime.
+
+        Must be called before waiting for a thread to complete or getting its result.
+    */
+    fn track_thread(&'lua self, id: ThreadId);
+
     /**
         Gets the result of the given thread.
 
@@ -221,6 +228,13 @@ impl<'lua> LuaRuntimeExt<'lua> for Lua {
         queue.push_item(self, thread, args)
     }
 
+    fn track_thread(&'lua self, id: ThreadId) {
+        let map = self
+            .app_data_ref::<ThreadResultMap>()
+            .expect("lua threads can only be tracked within a runtime");
+        map.track(id);
+    }
+
     fn get_thread_result(&'lua self, id: ThreadId) -> Option<LuaResult<LuaMultiValue<'lua>>> {
         let map = self
             .app_data_ref::<ThreadResultMap>()