feat: added autoCleanupOnClose and autoCloseOnExit on db opttions

README.md

@@ -121,6 +121,8 @@ Creates a new **MiftahDB** instance.
     - `mmapSize`: The memory-mapped file size (default: `30000000000`).
     - `lockingMode`: Determines the database locking mode (default: `"NORMAL"`).
     - `autoVacuumMode`: Configures the auto-vacuum behavior (default: `"OFF"`).
+    - `autoCleanupOnClose`: Automatically cleans up expired keys when the database is closed (default: `false`).
+    - `autoCloseOnExit`: Automatically closes the database when the process exits (default: `true`).
 
 #### Example Usage
 

src/base.ts

@@ -23,6 +23,7 @@ export abstract class BaseMiftahDB implements IMiftahDB {
   protected declare db: Database;
   protected statements: Record<string, Statement>;
   private nameSpacePrefix: string | null = null;
+  private autoCleanupOnClose: boolean;
 
   constructor(path = ":memory:", options: DBOptions = defaultDBOptions) {
     this.initDatabase(path);
@@ -43,8 +44,10 @@ export abstract class BaseMiftahDB implements IMiftahDB {
     this.db.exec(SQL_STATEMENTS.CREATE_INDEX);
 
     this.statements = this.prepareStatements();
+    this.autoCleanupOnClose = options.autoCleanupOnClose ?? false;
 
-    executeOnExit(() => this.close());
+    const autoCloseOnExit = options.autoCloseOnExit ?? true;
+    if (autoCloseOnExit) executeOnExit(() => this.close());
   }
 
   protected prepareStatements(): Record<string, Statement> {
@@ -69,6 +72,11 @@ export abstract class BaseMiftahDB implements IMiftahDB {
 
   protected abstract initDatabase(path: string | ":memory:"): void;
 
+  protected beforeClose(): void {
+    this.vacuum();
+    if (this.autoCleanupOnClose) this.cleanup();
+  }
+
   private addNamespacePrefix(k: string): string {
     return this.nameSpacePrefix ? `${this.nameSpacePrefix}:${k}` : k;
   }
@@ -316,7 +324,7 @@ export abstract class BaseMiftahDB implements IMiftahDB {
 
   @SafeExecution
   close(): Result<boolean> {
-    this.vacuum();
+    this.beforeClose();
 
     this.db.exec("PRAGMA wal_checkpoint(TRUNCATE)");
 

src/bun.ts

@@ -56,7 +56,8 @@ export class MiftahDB extends BaseMiftahDB {
   // Overridden due to difference implementation in `bun:sqlite` and `better-sqlite3`
   @SafeExecution
   override close(): Result<boolean> {
-    this.vacuum();
+    this.beforeClose();
+
     this.db.exec("PRAGMA wal_checkpoint(TRUNCATE)");
 
     // @ts-expect-error `deserialize` exists in `bun:sqlite` but not in `better-sqlite3`.

src/types.ts

@@ -467,5 +467,6 @@ export const defaultDBOptions = {
   mmapSize: 30000000000,
   lockingMode: "NORMAL" as "NORMAL" | "EXCLUSIVE",
   autoVacuumMode: "OFF" as "OFF" | "FULL" | "INCREMENTAL",
-  cleanUpOnClose: false as boolean,
+  autoCleanupOnClose: false as boolean,
+  autoCloseOnExit: true as boolean,
 } as const;