fix: reduce peak memory usage (#308)

hchunhui · Feb 6, 2024 · 7f3eca2 · 7f3eca2
1 parent cf2c572
commit 7f3eca2
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 0 deletions.
diff --git a/src/lib/lua.cc b/src/lib/lua.cc
@@ -264,6 +264,10 @@ std::shared_ptr<LuaObj> Lua::getglobal(const std::string &v) {
   return o;
 }
 
+void Lua::gc() {
+  lua_gc(L_, LUA_GCCOLLECT, 0);
+}
+
 LuaObj::LuaObj(lua_State *L, int i) : L_(L) {
   lua_pushvalue(L, i);
   id_ = luaL_ref(L, LUA_REGISTRYINDEX);

diff --git a/src/lib/lua.h b/src/lib/lua.h
@@ -34,6 +34,8 @@ class Lua {
 
   std::shared_ptr<LuaObj> newthreadx(lua_State *L, int nargs);
 
+  void gc();
+
   template <typename ... I>
   std::shared_ptr<LuaObj> newthread(I ... input);
 

diff --git a/src/lua_gears.cc b/src/lua_gears.cc
@@ -23,6 +23,10 @@ bool LuaTranslation::Next() {
   }
 }
 
+LuaTranslation::~LuaTranslation() {
+  lua_->gc();
+}
+
 static std::vector<std::string> split_string(const std::string& str, const std::string& delimiter) {
     std::vector<std::string> result;
     size_t pos = 0;

diff --git a/src/lua_gears.h b/src/lua_gears.h
@@ -24,6 +24,8 @@ class LuaTranslation : public Translation {
     return c_;
   }
 
+  virtual ~LuaTranslation();
+
 private:
   Lua *lua_;
   an<Candidate> c_;