Class CRDialect

app/src/sqlite/dialect.ts:12–65

} from 'kysely'

class CRDialect extends SqliteDialect {
	database: () => Promise<CRDatabase>

	constructor(config: CRDialectConfig) {
		super(config as unknown as SqliteDialectConfig)
		this.database = async () =>
			typeof config.database === 'function'
				? await config.database()
				: config.database
	}

	createDriver() {
		const load = this.database
		const waiter = mutex()

		let db: CRDatabase
		let connection: DatabaseConnection

		return {
			async init() {
				db = await load()
				connection = {
					async executeQuery<O>(query: CompiledQuery) {
						return {
							rows: (await db.execO(query.sql, query.parameters)) as O[],
						}
					},
					async *streamQuery() {
						throw new Error("Sqlite driver doesn't support streaming")
					},
				}
			},
			async acquireConnection() {
				await waiter.lock()
				return connection
			},
			async beginTransaction(connection: DatabaseConnection) {
				await connection.executeQuery(CompiledQuery.raw('begin'))
			},
			async commitTransaction(connection: DatabaseConnection) {
				await connection.executeQuery(CompiledQuery.raw('commit'))
			},
			async rollbackTransaction(connection: DatabaseConnection) {
				await connection.executeQuery(CompiledQuery.raw('rollback'))
			},
			async releaseConnection() {
				waiter.unlock()
			},
			async destroy() {
				db.close()
			},
		}
	}
}

function mutex() {
	let promise: Promise<void> | undefined