Spaces:
Sleeping
Sleeping
| 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 |