1 files changed, 14 insertions, 0 deletions

diff --git a/deep_test/deep_test.cpp b/deep_test/deep_test.cpp
index da467f3..c330cbe 100644
--- a/deep_test/deep_test.cpp
+++ b/deep_test/deep_test.cpp
@@ -47,6 +47,7 @@ void MyDeepFactory::deleteDeepObjectInternal(lua_State* const L_, DeepPrelude* c
 [[nodiscard]] static int deep_gc(lua_State* L)
 {
     MyDeepUserdata* const _self{ static_cast<MyDeepUserdata*>(MyDeepFactory::Instance.toDeep(L, 1)) };
+    luaL_argcheck(L, 1, !_self->inUse.load(), "being collected while in use!");
     return 0;
 }
 
@@ -76,6 +77,18 @@ void MyDeepFactory::deleteDeepObjectInternal(lua_State* const L_, DeepPrelude* c
 
 // #################################################################################################
 
+[[nodiscard]] static int deep_invoke(lua_State* L)
+{
+    MyDeepUserdata* const _self{ static_cast<MyDeepUserdata*>(MyDeepFactory::Instance.toDeep(L, 1)) };
+    luaL_argcheck(L, 2, lua_gettop(L) >= 2, "need something to call");
+    _self->inUse.fetch_add(1, std::memory_order_seq_cst);
+    lua_call(L, lua_gettop(L) - 2, LUA_MULTRET);
+    _self->inUse.fetch_sub(1, std::memory_order_seq_cst);
+    return 1;
+}
+
+// #################################################################################################
+
 [[nodiscard]] static int deep_set(lua_State* const L_)
 {
     MyDeepUserdata* const _self{ static_cast<MyDeepUserdata*>(MyDeepFactory::Instance.toDeep(L_, 1)) };
@@ -105,6 +118,7 @@ static luaL_Reg const deep_mt[] = {
     { "__gc", deep_gc },
     { "__tostring", deep_tostring },
     { "getuv", deep_getuv },
+    { "invoke", deep_invoke },
     { "set", deep_set },
     { "setuv", deep_setuv },
     { nullptr, nullptr }