Class BunSQLiteKVStore

packages/db/src/kv-store/storage/bun-sqlite.ts:36–237

};

export class BunSQLiteKVStore implements KVStore {
  db: Database;
  private statements: SQLiteKVState['statements'];
  private transactions: SQLiteKVState['transactions'];
  private walGuard?: NodeJS.Timer;

  constructor(databasePath: string, options?: SQLiteKVStoreOptions);
  constructor(database: Database, options?: SQLiteKVStoreOptions);
  constructor(arg0: string | Database, options: SQLiteKVStoreOptions = {}) {
    if (typeof arg0 === 'string') {
      this.db = new Database(arg0, { create: true });
    } else {
      this.db = arg0;
    }
    const parsedOptions = parseSqliteKvStoreOptions(options);
    this.db.exec(parsedOptions.pragma);
    this.createTable();
    this.statements = {
      get: this.db.query(STATEMENTS.get),
      set: this.db.query(STATEMENTS.set),
      delete: this.db.query(STATEMENTS.delete),
      deleteRange: this.db.query(STATEMENTS.deleteRange),
      scan: this.db.query(STATEMENTS.scan),
      scanValues: this.db.query(STATEMENTS.scanValues),
      count: this.db.query(STATEMENTS.count),
      countRange: this.db.query(STATEMENTS.countRange),
      truncate: this.db.query(STATEMENTS.truncate),
    };
    this.transactions = {
      write: this.db.transaction(
        (sets: Iterable<[Tuple, any]>, deletes: Iterable<Tuple>) => {
          for (const key of deletes) {
            const encodedKey = encodeTuple(key);
            this.freeStatement(this.statements.delete).run(encodedKey);
          }
          for (const [key, value] of sets) {
            const encodedKey = encodeTuple(key);
            const encodedValue = JSON.stringify(value);
            this.freeStatement(this.statements.set).run(
              encodedKey,
              encodedValue
            );
          }
        }
      ),
    };
    this.walGuard = this.startWalGuard(parsedOptions);
  }

  private startWalGuard(options: Required<SQLiteKVStoreOptions>) {
    if (this.walGuard) {
      clearInterval(this.walGuard);
    }
    const dbPath = this.db.filename;
    const walFile = `${dbPath}-wal`;
    const walCheck = setInterval(() => {
      walSizeGuard(this.db, walFile, {
        restartMax: options.checkpointRestart,