feat: collectGarbage method

opentibiabr · Sep 20, 2023 · 2c79115 · 2c79115
1 parent adb6978
commit 2c79115
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/lua/scripts/lua_environment.cpp b/src/lua/scripts/lua_environment.cpp
@@ -182,3 +182,19 @@ void LuaEnvironment::executeTimerEvent(uint32_t eventIndex) {
 		luaL_unref(luaState, LUA_REGISTRYINDEX, parameter);
 	}
 }
+
+void LuaEnvironment::collectGarbage() const {
+	// prevents recursive collects
+	static bool collecting = false;
+	if (!collecting) {
+		collecting = true;
+
+		// we must collect two times because __gc metamethod
+		// is called on uservalues only the second time
+		for (int i = -1; ++i < 2;) {
+			lua_gc(luaState, LUA_GCCOLLECT, 0);
+		}
+
+		collecting = false;
+	}
+}
diff --git a/src/lua/scripts/lua_environment.hpp b/src/lua/scripts/lua_environment.hpp
@@ -82,6 +82,8 @@ class LuaEnvironment : public LuaScriptInterface {
 		return shuttingDown;
 	}
 
+	void collectGarbage() const;
+
 private:
 	void executeTimerEvent(uint32_t eventIndex);