Class SQLiteSyncStorageTransaction

packages/sync-core/src/lib/SQLiteSyncStorage.ts:500–626

* Transaction implementation for SQLiteSyncStorage. * Provides access to documents, tombstones, and metadata within a transaction. * * @internal

 * @internal
 */
class SQLiteSyncStorageTransaction<R extends UnknownRecord> implements TLSyncStorageTransaction<R> {
	private _clock: number
	private _closed = false
	private _didIncrementClock: boolean = false

	constructor(
		private storage: SQLiteSyncStorage<R>,
		private stmts: SQLiteSyncStorage<R>['stmts']
	) {
		this._clock = this.storage.getClock()
	}

	/** @internal */
	close() {
		this._closed = true
	}

	private assertNotClosed() {
		assert(!this._closed, 'Transaction has ended, iterator cannot be consumed')
	}

	getClock(): number {
		return this._clock
	}

	private getNextClock(): number {
		if (!this._didIncrementClock) {
			this._didIncrementClock = true
			this.stmts.incrementDocumentClock.run()
			this._clock = this.storage.getClock()
		}
		return this._clock
	}

	get(id: string): R | undefined {
		this.assertNotClosed()
		const row = this.stmts.getDocument.all(id)[0]
		if (!row) return undefined
		return decodeState<R>(row.state)
	}

	set(id: string, record: R): void {
		this.assertNotClosed()
		assert(id === record.id, `Record id mismatch: key does not match record.id`)
		const clock = this.getNextClock()
		// Automatically clear tombstone if it exists
		this.stmts.deleteTombstone.run(id)
		this.stmts.insertDocument.run(id, encodeState(record), clock)
	}

	delete(id: string): void {
		this.assertNotClosed()
		// Only create a tombstone if the record actually exists
		const exists = this.stmts.documentExists.all(id)[0]
		if (!exists) return
		const clock = this.getNextClock()
		this.stmts.deleteDocument.run(id)
		this.stmts.insertTombstone.run(id, clock)