|
1 |
| -import type { DatabaseIntrospector, Dialect, DialectAdapter, Driver, Kysely, QueryCompiler } from 'kysely' |
2 |
| -import type { BunWorkerDialectConfig } from './type' |
3 |
| -import { SqliteAdapter, SqliteIntrospector, SqliteQueryCompiler } from 'kysely' |
4 |
| -import { BunWorkerDriver } from './driver' |
| 1 | +import type { BunWorkerDialectConfig, InitData } from './type' |
| 2 | +import { GenericSqliteWorkerDialect } from 'kysely-generic-sqlite/worker' |
| 3 | +import { createNodeMitt } from 'kysely-generic-sqlite/worker-helper-node' |
| 4 | +import { handleWebWorker } from 'kysely-generic-sqlite/worker-helper-web' |
5 | 5 |
|
6 |
| -export type { BunWorkerDialectConfig, Promisable } from './type' |
7 |
| -export { createOnMessageCallback } from './worker/utils' |
| 6 | +export * from './type' |
| 7 | +export * from './worker/utils' |
8 | 8 |
|
9 |
| -export class BunWorkerDialect implements Dialect { |
| 9 | +export class BunWorkerDialect extends GenericSqliteWorkerDialect<globalThis.Worker, InitData> { |
10 | 10 | /**
|
11 | 11 | * dialect for `bun:sqlite`, run sql in worker
|
12 | 12 | */
|
13 |
| - constructor( |
14 |
| - private config?: BunWorkerDialectConfig, |
15 |
| - ) { } |
16 |
| - |
17 |
| - createDriver(): Driver { |
18 |
| - return new BunWorkerDriver(this.config) |
19 |
| - } |
20 |
| - |
21 |
| - createQueryCompiler(): QueryCompiler { |
22 |
| - return new SqliteQueryCompiler() |
23 |
| - } |
24 |
| - |
25 |
| - createAdapter(): DialectAdapter { |
26 |
| - return new SqliteAdapter() |
27 |
| - } |
28 |
| - |
29 |
| - createIntrospector(db: Kysely<any>): DatabaseIntrospector { |
30 |
| - return new SqliteIntrospector(db) |
| 13 | + constructor(config?: BunWorkerDialectConfig) { |
| 14 | + const { |
| 15 | + url: fileName = ':memory:', |
| 16 | + cacheStatment: cache = false, |
| 17 | + onCreateConnection, |
| 18 | + worker = new Worker( |
| 19 | + new URL('./worker', import.meta.url), |
| 20 | + { type: 'module' }, |
| 21 | + ), |
| 22 | + } = config || {} |
| 23 | + super( |
| 24 | + () => ({ |
| 25 | + data: { cache, fileName }, |
| 26 | + mitt: createNodeMitt(), |
| 27 | + handle: handleWebWorker, |
| 28 | + worker, |
| 29 | + }), |
| 30 | + onCreateConnection, |
| 31 | + ) |
31 | 32 | }
|
32 | 33 | }
|
0 commit comments