lua: database: Add __close method

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/lua/database.c b/src/lua/database.c
index ee86ddad97e4bb29267765cc75bac52b971b0c99..f5306f984c6d8d45451a078a92f25de5d1f564c2 100644 (file)
--- a/src/lua/database.c
+++ b/src/lua/database.c
@@ -71,13 +71,24 @@ static int Database_open(lua_State* L) {
        return 1;
 }
 
-static int Database_gc(lua_State* L) {
-       Database* self = luaL_checkdatabase(L, 0);
+static int Database_close(lua_State* L) {
+       Database* self = luaL_checkdatabase(L, 1);
 
-       if (self->db)
+       // Free database
+       if (self->db) {
                loc_database_unref(self->db);
+               self->db = NULL;
+       }
+
+       return 0;
+}
+
+static int Database_gc(lua_State* L) {
+       Database* self = luaL_checkdatabase(L, 1);
 
+       // Free the object
        free(self);
+
        return 0;
 }
 
@@ -105,6 +116,7 @@ static int Database_lookup(lua_State* L) {
 static const struct luaL_Reg database_functions[] = {
        { "open", Database_open },
        { "lookup", Database_lookup },
+       { "__close", Database_close },
        { "__gc", Database_gc },
        { NULL, NULL },
 };