db.d: close database on exit

soulfind-dev · Jan 3, 2025 · 4e8af0d · 4e8af0d
1 parent 262e79d
commit 4e8af0d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/db.d b/src/db.d
@@ -14,7 +14,7 @@ import soulfind.defines : blue, default_max_users, default_port, norm;
 import std.conv : to;
 import std.exception : ifThrown;
 import std.file : exists, isFile;
-import std.stdio : writefln;
+import std.stdio : writefln, writeln;
 import std.string : format, replace, toStringz;
 
 struct SdbUserStats
@@ -39,6 +39,7 @@ class Sdb
 
     this(string filename)
     {
+        debug(db) writefln!("DB: Using database: %s")(filename);
         open_db(filename);
 
         if (!exists(filename) || !isFile(filename))
@@ -72,12 +73,24 @@ class Sdb
         init_config();
     }
 
+    ~this()
+    {
+        debug(db) writeln("DB: Shutting down...");
+        close_db();
+    }
+
     @trusted
     private void open_db(string filename)
     {
         sqlite3_open(filename.toStringz(), &db);
     }
 
+    @trusted
+    private void close_db()
+    {
+        sqlite3_close(db);
+    }
+
     private void init_config()
     {
         const sql = format!(

diff --git a/src/setup/package.d b/src/setup/package.d
@@ -35,6 +35,7 @@ int run(string[] args)
 
     sdb = new Sdb(db_filename);
     try { main_menu(); } catch (StdioException) {}
+    sdb = null;
     return 0;
 }