directus/packages/data/src/engine.ts

import type { ReadableStream } from 'node:stream/web';
import type { AbstractQuery } from './types/abstract-query.js';
import type { DataDriver } from './types/driver.js';

export class DataEngine {
	#stores: Map<string, DataDriver>;

	constructor() {
		this.#stores = new Map();
	}

	/** Registers a new data store for use in queries */
	async registerStore(name: string, driver: DataDriver) {
		await driver.register?.();
		this.#stores.set(name, driver);
	}

	/** Access the driver of a given store. Errors if it hasn't been registered */
	store(name: string): DataDriver {
		const store = this.#stores.get(name);

		if (!store) {
			throw new Error(`Store "${name}" doesn't exist.`);
		}

		return store;
	}

	/** Execute a root abstract query */
	async query(query: AbstractQuery): Promise<ReadableStream<Record<string, unknown>>> {
		return this.store(query.store).query(query);
	}

	/** Gracefully shutdown connected drivers */
	async destroy() {
		await Promise.all(Array.from(this.#stores.values()).map((driver) => driver.destroy?.()));
	}
}