UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

49 lines 1.46 kB
import { entityKind } from "../entity.js"; import { DefaultLogger } from "../logger.js"; import { createTableRelationsHelpers, extractTablesRelationalConfig } from "../relations.js"; import { BaseSQLiteDatabase } from "../sqlite-core/db.js"; import { SQLiteAsyncDialect } from "../sqlite-core/dialect.js"; import { SQLiteD1Session } from "./session.js"; class DrizzleD1Database extends BaseSQLiteDatabase { static [entityKind] = "D1Database"; async batch(batch) { return this.session.batch(batch); } } function drizzle(client, config = {}) { const dialect = new SQLiteAsyncDialect({ casing: config.casing }); let logger; if (config.logger === true) { logger = new DefaultLogger(); } else if (config.logger !== false) { logger = config.logger; } let schema; if (config.schema) { const tablesConfig = extractTablesRelationalConfig( config.schema, createTableRelationsHelpers ); schema = { fullSchema: config.schema, schema: tablesConfig.tables, tableNamesMap: tablesConfig.tableNamesMap }; } const session = new SQLiteD1Session(client, dialect, schema, { logger, cache: config.cache }); const db = new DrizzleD1Database("async", dialect, session, schema); db.$client = client; db.$cache = config.cache; if (db.$cache) { db.$cache["invalidate"] = config.cache?.onMutate; } return db; } export { DrizzleD1Database, drizzle }; //# sourceMappingURL=driver.js.map