@fragno-dev/db 0.0.1 → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.turbo/turbo-build.log +137 -13
- package/.turbo/turbo-test.log +36 -0
- package/CHANGELOG.md +7 -0
- package/dist/adapters/adapters.d.ts +18 -0
- package/dist/adapters/adapters.d.ts.map +1 -0
- package/dist/adapters/drizzle/drizzle-adapter.d.ts +21 -0
- package/dist/adapters/drizzle/drizzle-adapter.d.ts.map +1 -0
- package/dist/adapters/drizzle/drizzle-adapter.js +62 -0
- package/dist/adapters/drizzle/drizzle-adapter.js.map +1 -0
- package/dist/adapters/drizzle/drizzle-query.d.ts +17 -0
- package/dist/adapters/drizzle/drizzle-query.d.ts.map +1 -0
- package/dist/adapters/drizzle/drizzle-query.js +139 -0
- package/dist/adapters/drizzle/drizzle-query.js.map +1 -0
- package/dist/adapters/drizzle/drizzle-uow-compiler.d.ts +9 -0
- package/dist/adapters/drizzle/drizzle-uow-compiler.d.ts.map +1 -0
- package/dist/adapters/drizzle/drizzle-uow-compiler.js +300 -0
- package/dist/adapters/drizzle/drizzle-uow-compiler.js.map +1 -0
- package/dist/adapters/drizzle/drizzle-uow-decoder.js +82 -0
- package/dist/adapters/drizzle/drizzle-uow-decoder.js.map +1 -0
- package/dist/adapters/drizzle/drizzle-uow-executor.js +125 -0
- package/dist/adapters/drizzle/drizzle-uow-executor.js.map +1 -0
- package/dist/adapters/drizzle/generate.js +273 -0
- package/dist/adapters/drizzle/generate.js.map +1 -0
- package/dist/adapters/drizzle/join-column-utils.js +28 -0
- package/dist/adapters/drizzle/join-column-utils.js.map +1 -0
- package/dist/adapters/drizzle/shared.js +11 -0
- package/dist/adapters/drizzle/shared.js.map +1 -0
- package/dist/adapters/kysely/kysely-adapter.d.ts +23 -0
- package/dist/adapters/kysely/kysely-adapter.d.ts.map +1 -0
- package/dist/adapters/kysely/kysely-adapter.js +119 -0
- package/dist/adapters/kysely/kysely-adapter.js.map +1 -0
- package/dist/adapters/kysely/kysely-query-builder.js +306 -0
- package/dist/adapters/kysely/kysely-query-builder.js.map +1 -0
- package/dist/adapters/kysely/kysely-query-compiler.js +67 -0
- package/dist/adapters/kysely/kysely-query-compiler.js.map +1 -0
- package/dist/adapters/kysely/kysely-query.js +158 -0
- package/dist/adapters/kysely/kysely-query.js.map +1 -0
- package/dist/adapters/kysely/kysely-uow-compiler.js +139 -0
- package/dist/adapters/kysely/kysely-uow-compiler.js.map +1 -0
- package/dist/adapters/kysely/kysely-uow-executor.js +89 -0
- package/dist/adapters/kysely/kysely-uow-executor.js.map +1 -0
- package/dist/adapters/kysely/migration/execute.js +176 -0
- package/dist/adapters/kysely/migration/execute.js.map +1 -0
- package/dist/fragment.d.ts +54 -0
- package/dist/fragment.d.ts.map +1 -0
- package/dist/fragment.js +92 -0
- package/dist/fragment.js.map +1 -0
- package/dist/id.d.ts +2 -0
- package/dist/migration-engine/auto-from-schema.js +116 -0
- package/dist/migration-engine/auto-from-schema.js.map +1 -0
- package/dist/migration-engine/create.d.ts +41 -0
- package/dist/migration-engine/create.d.ts.map +1 -0
- package/dist/migration-engine/create.js +58 -0
- package/dist/migration-engine/create.js.map +1 -0
- package/dist/migration-engine/shared.d.ts +90 -0
- package/dist/migration-engine/shared.d.ts.map +1 -0
- package/dist/migration-engine/shared.js +8 -0
- package/dist/migration-engine/shared.js.map +1 -0
- package/dist/mod.d.ts +55 -2
- package/dist/mod.d.ts.map +1 -1
- package/dist/mod.js +111 -2
- package/dist/mod.js.map +1 -1
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/column-builder.js +108 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/column-builder.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/column.js +55 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/column.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/entity.js +18 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/entity.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/columns/common.js +183 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/columns/common.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/columns/enum.js +58 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/columns/enum.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/foreign-keys.js +68 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/foreign-keys.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/unique-constraint.js +56 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/unique-constraint.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/utils/array.js +65 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/pg-core/utils/array.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/conditions.js +81 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/conditions.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/select.js +13 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/select.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/functions/aggregate.js +10 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/functions/aggregate.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/sql.js +372 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/sql.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/subquery.js +23 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/subquery.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/table.js +62 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/table.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/table.utils.js +6 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/table.utils.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/tracing-utils.js +8 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/tracing-utils.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/tracing.js +8 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/tracing.js.map +1 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/view-common.js +6 -0
- package/dist/node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/view-common.js.map +1 -0
- package/dist/query/condition-builder.d.ts +41 -0
- package/dist/query/condition-builder.d.ts.map +1 -0
- package/dist/query/condition-builder.js +93 -0
- package/dist/query/condition-builder.js.map +1 -0
- package/dist/query/cursor.d.ts +88 -0
- package/dist/query/cursor.d.ts.map +1 -0
- package/dist/query/cursor.js +103 -0
- package/dist/query/cursor.js.map +1 -0
- package/dist/query/orm/orm.d.ts +18 -0
- package/dist/query/orm/orm.d.ts.map +1 -0
- package/dist/query/orm/orm.js +48 -0
- package/dist/query/orm/orm.js.map +1 -0
- package/dist/query/query.d.ts +79 -0
- package/dist/query/query.d.ts.map +1 -0
- package/dist/query/query.js +1 -0
- package/dist/query/result-transform.js +155 -0
- package/dist/query/result-transform.js.map +1 -0
- package/dist/query/unit-of-work.d.ts +435 -0
- package/dist/query/unit-of-work.d.ts.map +1 -0
- package/dist/query/unit-of-work.js +549 -0
- package/dist/query/unit-of-work.js.map +1 -0
- package/dist/schema/create.d.ts +273 -116
- package/dist/schema/create.d.ts.map +1 -1
- package/dist/schema/create.js +410 -222
- package/dist/schema/create.js.map +1 -1
- package/dist/schema/serialize.js +101 -0
- package/dist/schema/serialize.js.map +1 -0
- package/dist/schema-generator/schema-generator.d.ts +15 -0
- package/dist/schema-generator/schema-generator.d.ts.map +1 -0
- package/dist/shared/providers.d.ts +6 -0
- package/dist/shared/providers.d.ts.map +1 -0
- package/dist/util/import-generator.js +26 -0
- package/dist/util/import-generator.js.map +1 -0
- package/dist/util/parse.js +15 -0
- package/dist/util/parse.js.map +1 -0
- package/dist/util/types.d.ts +8 -0
- package/dist/util/types.d.ts.map +1 -0
- package/package.json +63 -2
- package/src/adapters/adapters.ts +22 -0
- package/src/adapters/drizzle/drizzle-adapter-pglite.test.ts +433 -0
- package/src/adapters/drizzle/drizzle-adapter.test.ts +122 -0
- package/src/adapters/drizzle/drizzle-adapter.ts +118 -0
- package/src/adapters/drizzle/drizzle-query.ts +234 -0
- package/src/adapters/drizzle/drizzle-uow-compiler.test.ts +1084 -0
- package/src/adapters/drizzle/drizzle-uow-compiler.ts +546 -0
- package/src/adapters/drizzle/drizzle-uow-decoder.ts +165 -0
- package/src/adapters/drizzle/drizzle-uow-executor.ts +213 -0
- package/src/adapters/drizzle/generate.test.ts +643 -0
- package/src/adapters/drizzle/generate.ts +481 -0
- package/src/adapters/drizzle/join-column-utils.test.ts +79 -0
- package/src/adapters/drizzle/join-column-utils.ts +39 -0
- package/src/adapters/drizzle/migrate-drizzle.test.ts +226 -0
- package/src/adapters/drizzle/shared.ts +22 -0
- package/src/adapters/drizzle/test-utils.ts +56 -0
- package/src/adapters/kysely/kysely-adapter-pglite.test.ts +789 -0
- package/src/adapters/kysely/kysely-adapter.ts +196 -0
- package/src/adapters/kysely/kysely-query-builder.test.ts +1344 -0
- package/src/adapters/kysely/kysely-query-builder.ts +611 -0
- package/src/adapters/kysely/kysely-query-compiler.ts +124 -0
- package/src/adapters/kysely/kysely-query.ts +254 -0
- package/src/adapters/kysely/kysely-uow-compiler.test.ts +916 -0
- package/src/adapters/kysely/kysely-uow-compiler.ts +271 -0
- package/src/adapters/kysely/kysely-uow-executor.ts +149 -0
- package/src/adapters/kysely/kysely-uow-joins.test.ts +811 -0
- package/src/adapters/kysely/migration/execute-mysql.test.ts +1173 -0
- package/src/adapters/kysely/migration/execute-postgres.test.ts +2657 -0
- package/src/adapters/kysely/migration/execute.ts +382 -0
- package/src/adapters/kysely/migration/kysely-migrator.test.ts +197 -0
- package/src/fragment.test.ts +287 -0
- package/src/fragment.ts +198 -0
- package/src/migration-engine/auto-from-schema.test.ts +118 -58
- package/src/migration-engine/auto-from-schema.ts +103 -32
- package/src/migration-engine/create.test.ts +34 -46
- package/src/migration-engine/create.ts +41 -26
- package/src/migration-engine/shared.ts +26 -6
- package/src/mod.ts +197 -1
- package/src/query/condition-builder.test.ts +379 -0
- package/src/query/condition-builder.ts +294 -0
- package/src/query/cursor.test.ts +296 -0
- package/src/query/cursor.ts +147 -0
- package/src/query/orm/orm.ts +92 -0
- package/src/query/query-type.test.ts +429 -0
- package/src/query/query.ts +200 -0
- package/src/query/result-transform.test.ts +795 -0
- package/src/query/result-transform.ts +247 -0
- package/src/query/unit-of-work-types.test.ts +192 -0
- package/src/query/unit-of-work.test.ts +947 -0
- package/src/query/unit-of-work.ts +1199 -0
- package/src/schema/create.test.ts +653 -110
- package/src/schema/create.ts +708 -337
- package/src/schema/serialize.test.ts +559 -0
- package/src/schema/serialize.ts +359 -0
- package/src/schema-generator/schema-generator.ts +12 -0
- package/src/shared/config.ts +0 -8
- package/src/util/import-generator.ts +28 -0
- package/src/util/parse.ts +16 -0
- package/src/util/types.ts +4 -0
- package/tsconfig.json +1 -1
- package/tsdown.config.ts +11 -1
- package/vitest.config.ts +3 -0
- /package/dist/{cuid.js → id.js} +0 -0
- /package/src/{cuid.ts → id.ts} +0 -0
|
@@ -0,0 +1,300 @@
|
|
|
1
|
+
import { Column } from "../../schema/create.js";
|
|
2
|
+
import { serialize } from "../../schema/serialize.js";
|
|
3
|
+
import { decodeCursor, serializeCursorValues } from "../../query/cursor.js";
|
|
4
|
+
import { buildCondition } from "../../query/condition-builder.js";
|
|
5
|
+
import { sql } from "../../node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/sql.js";
|
|
6
|
+
import { and, eq, gt, gte, inArray, isNotNull, isNull, like, lt, lte, ne, not, notInArray, notLike, or } from "../../node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/conditions.js";
|
|
7
|
+
import { asc, desc } from "../../node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/expressions/select.js";
|
|
8
|
+
import { count } from "../../node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/functions/aggregate.js";
|
|
9
|
+
import { parseDrizzle } from "./shared.js";
|
|
10
|
+
import { ReferenceSubquery, encodeValues } from "../../query/result-transform.js";
|
|
11
|
+
import { getOrderedJoinColumns } from "./join-column-utils.js";
|
|
12
|
+
|
|
13
|
+
//#region src/adapters/drizzle/drizzle-uow-compiler.ts
|
|
14
|
+
/**
|
|
15
|
+
* Create a Drizzle-specific Unit of Work compiler
|
|
16
|
+
*
|
|
17
|
+
* This compiler translates UOW operations into Drizzle query functions
|
|
18
|
+
* that can be executed as a batch/transaction.
|
|
19
|
+
*
|
|
20
|
+
* @param schema - The database schema
|
|
21
|
+
* @param config - Drizzle configuration
|
|
22
|
+
* @param onQuery - Optional callback to receive compiled queries for logging/debugging
|
|
23
|
+
* @returns A UOWCompiler instance for Drizzle
|
|
24
|
+
*/
|
|
25
|
+
function createDrizzleUOWCompiler(schema, config, onQuery) {
|
|
26
|
+
const [db, drizzleTables] = parseDrizzle(config.db);
|
|
27
|
+
const { provider } = config;
|
|
28
|
+
/**
|
|
29
|
+
* Convert a Fragno table to a Drizzle table
|
|
30
|
+
* @throws Error if table is not found in Drizzle schema
|
|
31
|
+
*/
|
|
32
|
+
function toDrizzleTable(table) {
|
|
33
|
+
const tableName = table.ormName;
|
|
34
|
+
const out = drizzleTables[tableName];
|
|
35
|
+
if (out) return out;
|
|
36
|
+
throw new Error(`[Drizzle] Unknown table name ${tableName}, is it included in your Drizzle schema?`);
|
|
37
|
+
}
|
|
38
|
+
/**
|
|
39
|
+
* Convert a Fragno column to a Drizzle column
|
|
40
|
+
* @throws Error if column is not found in Drizzle table
|
|
41
|
+
*/
|
|
42
|
+
function toDrizzleColumn(col) {
|
|
43
|
+
const fragnoTable = schema.tables[col.tableName];
|
|
44
|
+
if (!fragnoTable) throw new Error(`[Drizzle] Unknown table ${col.tableName} for column ${col.ormName}.`);
|
|
45
|
+
const out = toDrizzleTable(fragnoTable)[col.ormName];
|
|
46
|
+
if (out) return out;
|
|
47
|
+
throw new Error(`[Drizzle] Unknown column name ${col.ormName} in ${fragnoTable.ormName}.`);
|
|
48
|
+
}
|
|
49
|
+
/**
|
|
50
|
+
* Build a WHERE clause from a condition using Drizzle's query builder
|
|
51
|
+
*/
|
|
52
|
+
function buildWhere(condition) {
|
|
53
|
+
if (condition.type === "compare") {
|
|
54
|
+
const left = toDrizzleColumn(condition.a);
|
|
55
|
+
const op = condition.operator;
|
|
56
|
+
let right = condition.b;
|
|
57
|
+
if (right instanceof Column) right = toDrizzleColumn(right);
|
|
58
|
+
else right = serialize(right, condition.a, provider);
|
|
59
|
+
switch (op) {
|
|
60
|
+
case "=": return eq(left, right);
|
|
61
|
+
case "!=": return ne(left, right);
|
|
62
|
+
case ">": return gt(left, right);
|
|
63
|
+
case ">=": return gte(left, right);
|
|
64
|
+
case "<": return lt(left, right);
|
|
65
|
+
case "<=": return lte(left, right);
|
|
66
|
+
case "in": return inArray(left, right);
|
|
67
|
+
case "not in": return notInArray(left, right);
|
|
68
|
+
case "is": return right === null ? isNull(left) : eq(left, right);
|
|
69
|
+
case "is not": return right === null ? isNotNull(left) : ne(left, right);
|
|
70
|
+
case "contains":
|
|
71
|
+
right = typeof right === "string" ? `%${right}%` : sql`concat('%', ${right}, '%')`;
|
|
72
|
+
return like(left, right);
|
|
73
|
+
case "not contains":
|
|
74
|
+
right = typeof right === "string" ? `%${right}%` : sql`concat('%', ${right}, '%')`;
|
|
75
|
+
return notLike(left, right);
|
|
76
|
+
case "ends with":
|
|
77
|
+
right = typeof right === "string" ? `%${right}` : sql`concat('%', ${right})`;
|
|
78
|
+
return like(left, right);
|
|
79
|
+
case "not ends with":
|
|
80
|
+
right = typeof right === "string" ? `%${right}` : sql`concat('%', ${right})`;
|
|
81
|
+
return notLike(left, right);
|
|
82
|
+
case "starts with":
|
|
83
|
+
right = typeof right === "string" ? `${right}%` : sql`concat(${right}, '%')`;
|
|
84
|
+
return like(left, right);
|
|
85
|
+
case "not starts with":
|
|
86
|
+
right = typeof right === "string" ? `${right}%` : sql`concat(${right}, '%')`;
|
|
87
|
+
return notLike(left, right);
|
|
88
|
+
default: throw new Error(`Unsupported operator: ${op}`);
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
if (condition.type === "and") return and(...condition.items.map((item) => buildWhere(item)));
|
|
92
|
+
if (condition.type === "not") {
|
|
93
|
+
const result = buildWhere(condition.item);
|
|
94
|
+
if (!result) return;
|
|
95
|
+
return not(result);
|
|
96
|
+
}
|
|
97
|
+
return or(...condition.items.map((item) => buildWhere(item)));
|
|
98
|
+
}
|
|
99
|
+
/**
|
|
100
|
+
* Process reference subqueries in encoded values, converting them to Drizzle SQL subqueries
|
|
101
|
+
*/
|
|
102
|
+
function processReferenceSubqueries(values) {
|
|
103
|
+
const processed = {};
|
|
104
|
+
for (const [key, value] of Object.entries(values)) if (value instanceof ReferenceSubquery) {
|
|
105
|
+
const refTable = value.referencedTable;
|
|
106
|
+
const externalId = value.externalIdValue;
|
|
107
|
+
const internalIdCol = refTable.getInternalIdColumn();
|
|
108
|
+
const idCol = refTable.getIdColumn();
|
|
109
|
+
const drizzleRefTable = toDrizzleTable(refTable);
|
|
110
|
+
const drizzleIdCol = toDrizzleColumn(idCol);
|
|
111
|
+
processed[key] = db.select({ value: drizzleRefTable[internalIdCol.ormName] }).from(drizzleRefTable).where(eq(drizzleIdCol, externalId)).limit(1);
|
|
112
|
+
} else processed[key] = value;
|
|
113
|
+
return processed;
|
|
114
|
+
}
|
|
115
|
+
/**
|
|
116
|
+
* Get table from schema by name
|
|
117
|
+
* @throws Error if table is not found in schema
|
|
118
|
+
*/
|
|
119
|
+
function getTable(name) {
|
|
120
|
+
const table = schema.tables[name];
|
|
121
|
+
if (!table) throw new Error(`Invalid table name ${name}.`);
|
|
122
|
+
return table;
|
|
123
|
+
}
|
|
124
|
+
/**
|
|
125
|
+
* Get the version to check for a given ID and checkVersion flag.
|
|
126
|
+
* @returns The version to check or undefined if no check is required.
|
|
127
|
+
* @throws Error if the ID is a string and checkVersion is true.
|
|
128
|
+
*/
|
|
129
|
+
function getVersionToCheck(id, checkVersion) {
|
|
130
|
+
if (!checkVersion) return;
|
|
131
|
+
if (typeof id === "string") throw new Error(`Cannot use checkVersion with a string ID. Version checking requires a FragnoId with version information.`);
|
|
132
|
+
return id.version;
|
|
133
|
+
}
|
|
134
|
+
/**
|
|
135
|
+
* Process joins recursively to support nested joins with orderBy and limit
|
|
136
|
+
*/
|
|
137
|
+
function processJoins(joins) {
|
|
138
|
+
const result = {};
|
|
139
|
+
for (const join of joins) {
|
|
140
|
+
const { options, relation } = join;
|
|
141
|
+
if (!options) continue;
|
|
142
|
+
const targetTable = relation.table;
|
|
143
|
+
const joinName = relation.name;
|
|
144
|
+
const orderedColumns = getOrderedJoinColumns(targetTable, options.select === void 0 ? true : options.select);
|
|
145
|
+
const joinColumns = {};
|
|
146
|
+
for (const colName of orderedColumns) joinColumns[colName] = true;
|
|
147
|
+
let joinOrderBy;
|
|
148
|
+
if (options.orderBy && options.orderBy.length > 0) joinOrderBy = options.orderBy.map(([col, direction]) => {
|
|
149
|
+
const drizzleCol = toDrizzleColumn(col);
|
|
150
|
+
return direction === "asc" ? asc(drizzleCol) : desc(drizzleCol);
|
|
151
|
+
});
|
|
152
|
+
let joinWhere;
|
|
153
|
+
if (options.where) joinWhere = buildWhere(options.where);
|
|
154
|
+
const joinConfig = {
|
|
155
|
+
columns: joinColumns,
|
|
156
|
+
orderBy: joinOrderBy,
|
|
157
|
+
limit: options.limit,
|
|
158
|
+
where: joinWhere
|
|
159
|
+
};
|
|
160
|
+
if (options.join && options.join.length > 0) joinConfig.with = processJoins(options.join);
|
|
161
|
+
result[joinName] = joinConfig;
|
|
162
|
+
}
|
|
163
|
+
return result;
|
|
164
|
+
}
|
|
165
|
+
return {
|
|
166
|
+
compileRetrievalOperation(op) {
|
|
167
|
+
switch (op.type) {
|
|
168
|
+
case "count": {
|
|
169
|
+
let whereClause;
|
|
170
|
+
if (op.options.where) {
|
|
171
|
+
const condition = buildCondition(op.table.columns, op.options.where);
|
|
172
|
+
if (condition === false) return null;
|
|
173
|
+
if (condition !== true) whereClause = buildWhere(condition);
|
|
174
|
+
}
|
|
175
|
+
const drizzleTable = toDrizzleTable(op.table);
|
|
176
|
+
const query = db.select({ count: count() }).from(drizzleTable);
|
|
177
|
+
const compiledQuery = whereClause ? query.where(whereClause).toSQL() : query.toSQL();
|
|
178
|
+
onQuery?.(compiledQuery);
|
|
179
|
+
return compiledQuery;
|
|
180
|
+
}
|
|
181
|
+
case "find": {
|
|
182
|
+
const { useIndex: _useIndex, orderByIndex, joins, after, before, pageSize,...findOptions } = op.options;
|
|
183
|
+
let indexColumns = [];
|
|
184
|
+
let orderDirection = "asc";
|
|
185
|
+
if (orderByIndex) {
|
|
186
|
+
const index = op.table.indexes[orderByIndex.indexName];
|
|
187
|
+
orderDirection = orderByIndex.direction;
|
|
188
|
+
if (!index) if (orderByIndex.indexName === "_primary") indexColumns = [op.table.getIdColumn()];
|
|
189
|
+
else throw new Error(`Index "${orderByIndex.indexName}" not found on table "${op.table.name}"`);
|
|
190
|
+
else indexColumns = index.columns;
|
|
191
|
+
}
|
|
192
|
+
let orderBy;
|
|
193
|
+
if (indexColumns.length > 0) orderBy = indexColumns.map((col) => {
|
|
194
|
+
const drizzleCol = toDrizzleColumn(col);
|
|
195
|
+
return orderDirection === "asc" ? asc(drizzleCol) : desc(drizzleCol);
|
|
196
|
+
});
|
|
197
|
+
const columns = {};
|
|
198
|
+
const select = findOptions.select;
|
|
199
|
+
if (select === true || select === void 0) for (const col of Object.values(op.table.columns)) columns[col.ormName] = true;
|
|
200
|
+
else {
|
|
201
|
+
for (const k of select) columns[op.table.columns[k].ormName] = true;
|
|
202
|
+
for (const col of Object.values(op.table.columns)) if (col.isHidden && !columns[col.ormName]) columns[col.ormName] = true;
|
|
203
|
+
}
|
|
204
|
+
const whereClauses = [];
|
|
205
|
+
if (findOptions.where) {
|
|
206
|
+
const condition = buildCondition(op.table.columns, findOptions.where);
|
|
207
|
+
if (condition === false) return null;
|
|
208
|
+
if (condition !== true) {
|
|
209
|
+
const clause = buildWhere(condition);
|
|
210
|
+
if (clause) whereClauses.push(clause);
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
if ((after || before) && indexColumns.length > 0) {
|
|
214
|
+
const serializedValues = serializeCursorValues(decodeCursor(after || before), indexColumns, provider);
|
|
215
|
+
const isAfter = !!after;
|
|
216
|
+
const useGreaterThan = isAfter && orderDirection === "asc" || !isAfter && orderDirection === "desc";
|
|
217
|
+
if (indexColumns.length === 1) {
|
|
218
|
+
const col = toDrizzleColumn(indexColumns[0]);
|
|
219
|
+
const val = serializedValues[indexColumns[0].ormName];
|
|
220
|
+
whereClauses.push(useGreaterThan ? gt(col, val) : lt(col, val));
|
|
221
|
+
} else {
|
|
222
|
+
const drizzleCols = indexColumns.map((c) => toDrizzleColumn(c));
|
|
223
|
+
const vals = indexColumns.map((c) => serializedValues[c.ormName]);
|
|
224
|
+
const operator = useGreaterThan ? ">" : "<";
|
|
225
|
+
const colsSQL = sql.join(drizzleCols, sql.raw(", "));
|
|
226
|
+
const valsSQL = sql.join(vals.map((v) => sql`${v}`), sql.raw(", "));
|
|
227
|
+
whereClauses.push(sql`(${colsSQL}) ${sql.raw(operator)} (${valsSQL})`);
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
const queryConfig = {
|
|
231
|
+
columns,
|
|
232
|
+
limit: pageSize,
|
|
233
|
+
where: whereClauses.length > 0 ? and(...whereClauses) : void 0,
|
|
234
|
+
orderBy,
|
|
235
|
+
with: {}
|
|
236
|
+
};
|
|
237
|
+
if (joins) queryConfig.with = processJoins(joins);
|
|
238
|
+
const compiledQuery = db.query[op.table.ormName].findMany(queryConfig).toSQL();
|
|
239
|
+
onQuery?.(compiledQuery);
|
|
240
|
+
return compiledQuery;
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
},
|
|
244
|
+
compileMutationOperation(op) {
|
|
245
|
+
switch (op.type) {
|
|
246
|
+
case "create": {
|
|
247
|
+
const table = getTable(op.table);
|
|
248
|
+
const drizzleTable = toDrizzleTable(table);
|
|
249
|
+
const values = processReferenceSubqueries(encodeValues(op.values, table, true, provider));
|
|
250
|
+
const compiledQuery = db.insert(drizzleTable).values(values).toSQL();
|
|
251
|
+
onQuery?.(compiledQuery);
|
|
252
|
+
return {
|
|
253
|
+
query: compiledQuery,
|
|
254
|
+
expectedAffectedRows: null
|
|
255
|
+
};
|
|
256
|
+
}
|
|
257
|
+
case "update": {
|
|
258
|
+
const table = getTable(op.table);
|
|
259
|
+
const idColumn = table.getIdColumn();
|
|
260
|
+
const versionColumn = table.getVersionColumn();
|
|
261
|
+
const drizzleTable = toDrizzleTable(table);
|
|
262
|
+
const externalId = typeof op.id === "string" ? op.id : op.id.externalId;
|
|
263
|
+
const versionToCheck = getVersionToCheck(op.id, op.checkVersion);
|
|
264
|
+
const condition = versionToCheck !== void 0 ? buildCondition(table.columns, (eb) => eb.and(eb(idColumn.ormName, "=", externalId), eb(versionColumn.ormName, "=", versionToCheck))) : buildCondition(table.columns, (eb) => eb(idColumn.ormName, "=", externalId));
|
|
265
|
+
if (condition === false) return null;
|
|
266
|
+
const whereClause = condition === true ? void 0 : buildWhere(condition);
|
|
267
|
+
const setValues = processReferenceSubqueries(encodeValues(op.set, table, false, provider));
|
|
268
|
+
setValues[versionColumn.ormName] = sql.raw(`COALESCE(${versionColumn.ormName}, 0) + 1`);
|
|
269
|
+
const compiledQuery = db.update(drizzleTable).set(setValues).where(whereClause).toSQL();
|
|
270
|
+
onQuery?.(compiledQuery);
|
|
271
|
+
return {
|
|
272
|
+
query: compiledQuery,
|
|
273
|
+
expectedAffectedRows: op.checkVersion ? 1 : null
|
|
274
|
+
};
|
|
275
|
+
}
|
|
276
|
+
case "delete": {
|
|
277
|
+
const table = getTable(op.table);
|
|
278
|
+
const idColumn = table.getIdColumn();
|
|
279
|
+
const versionColumn = table.getVersionColumn();
|
|
280
|
+
const drizzleTable = toDrizzleTable(table);
|
|
281
|
+
const externalId = typeof op.id === "string" ? op.id : op.id.externalId;
|
|
282
|
+
const versionToCheck = getVersionToCheck(op.id, op.checkVersion);
|
|
283
|
+
const condition = versionToCheck !== void 0 ? buildCondition(table.columns, (eb) => eb.and(eb(idColumn.ormName, "=", externalId), eb(versionColumn.ormName, "=", versionToCheck))) : buildCondition(table.columns, (eb) => eb(idColumn.ormName, "=", externalId));
|
|
284
|
+
if (condition === false) return null;
|
|
285
|
+
const whereClause = condition === true ? void 0 : buildWhere(condition);
|
|
286
|
+
const compiledQuery = db.delete(drizzleTable).where(whereClause).toSQL();
|
|
287
|
+
onQuery?.(compiledQuery);
|
|
288
|
+
return {
|
|
289
|
+
query: compiledQuery,
|
|
290
|
+
expectedAffectedRows: op.checkVersion ? 1 : null
|
|
291
|
+
};
|
|
292
|
+
}
|
|
293
|
+
}
|
|
294
|
+
}
|
|
295
|
+
};
|
|
296
|
+
}
|
|
297
|
+
|
|
298
|
+
//#endregion
|
|
299
|
+
export { createDrizzleUOWCompiler };
|
|
300
|
+
//# sourceMappingURL=drizzle-uow-compiler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"drizzle-uow-compiler.js","names":["Drizzle.eq","Drizzle.ne","Drizzle.gt","Drizzle.gte","Drizzle.lt","Drizzle.lte","Drizzle.inArray","Drizzle.notInArray","Drizzle.isNull","Drizzle.isNotNull","Drizzle.like","Drizzle.notLike","Drizzle.and","Drizzle.not","Drizzle.or","processed: Record<string, unknown>","result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>>","joinColumns: Record<string, boolean>","joinOrderBy: Drizzle.SQL[] | undefined","Drizzle.asc","Drizzle.desc","joinWhere: Drizzle.SQL | undefined","joinConfig: Drizzle.DBQueryConfig<\"many\", boolean>","whereClause: Drizzle.SQL | undefined","Drizzle.count","indexColumns: AnyColumn[]","orderDirection: \"asc\" | \"desc\"","orderBy: Drizzle.SQL[] | undefined","columns: Record<string, boolean>","whereClauses: Drizzle.SQL[]","queryConfig: Drizzle.DBQueryConfig<\"many\", boolean>"],"sources":["../../../src/adapters/drizzle/drizzle-uow-compiler.ts"],"sourcesContent":["import * as Drizzle from \"drizzle-orm\";\nimport type { AnyColumn, AnySchema, AnyTable, FragnoId } from \"../../schema/create\";\nimport { Column } from \"../../schema/create\";\nimport type {\n CompiledMutation,\n MutationOperation,\n RetrievalOperation,\n UOWCompiler,\n} from \"../../query/unit-of-work\";\nimport { buildCondition, type Condition } from \"../../query/condition-builder\";\nimport type { DrizzleConfig } from \"./drizzle-adapter\";\nimport { type ColumnType, type TableType, parseDrizzle } from \"./shared\";\nimport { encodeValues, ReferenceSubquery } from \"../../query/result-transform\";\nimport { serialize } from \"../../schema/serialize\";\nimport { decodeCursor, serializeCursorValues } from \"../../query/cursor\";\nimport type { CompiledJoin } from \"../../query/orm/orm\";\nimport { getOrderedJoinColumns } from \"./join-column-utils\";\n\nexport type DrizzleCompiledQuery = {\n sql: string;\n params: unknown[];\n};\n\n/**\n * Create a Drizzle-specific Unit of Work compiler\n *\n * This compiler translates UOW operations into Drizzle query functions\n * that can be executed as a batch/transaction.\n *\n * @param schema - The database schema\n * @param config - Drizzle configuration\n * @param onQuery - Optional callback to receive compiled queries for logging/debugging\n * @returns A UOWCompiler instance for Drizzle\n */\nexport function createDrizzleUOWCompiler<TSchema extends AnySchema>(\n schema: TSchema,\n config: DrizzleConfig,\n onQuery?: (query: DrizzleCompiledQuery) => void,\n): UOWCompiler<TSchema, DrizzleCompiledQuery> {\n const [db, drizzleTables] = parseDrizzle(config.db);\n const { provider } = config;\n\n /**\n * Convert a Fragno table to a Drizzle table\n * @throws Error if table is not found in Drizzle schema\n */\n function toDrizzleTable(table: AnyTable): TableType {\n const tableName = table.ormName;\n const out = drizzleTables[tableName];\n if (out) {\n return out;\n }\n\n throw new Error(\n `[Drizzle] Unknown table name ${tableName}, is it included in your Drizzle schema?`,\n );\n }\n\n /**\n * Convert a Fragno column to a Drizzle column\n * @throws Error if column is not found in Drizzle table\n */\n function toDrizzleColumn(col: AnyColumn): ColumnType {\n const fragnoTable = schema.tables[col.tableName];\n if (!fragnoTable) {\n throw new Error(`[Drizzle] Unknown table ${col.tableName} for column ${col.ormName}.`);\n }\n\n const table = toDrizzleTable(fragnoTable);\n const out = table[col.ormName];\n if (out) {\n return out;\n }\n\n throw new Error(`[Drizzle] Unknown column name ${col.ormName} in ${fragnoTable.ormName}.`);\n }\n\n /**\n * Build a WHERE clause from a condition using Drizzle's query builder\n */\n function buildWhere(condition: Condition): Drizzle.SQL | undefined {\n if (condition.type === \"compare\") {\n const left = toDrizzleColumn(condition.a);\n const op = condition.operator;\n let right = condition.b;\n if (right instanceof Column) {\n right = toDrizzleColumn(right);\n } else {\n // Serialize non-Column values (e.g., FragnoId -> string, Date -> number for SQLite)\n right = serialize(right, condition.a, provider);\n }\n\n switch (op) {\n case \"=\":\n return Drizzle.eq(left, right);\n case \"!=\":\n return Drizzle.ne(left, right);\n case \">\":\n return Drizzle.gt(left, right);\n case \">=\":\n return Drizzle.gte(left, right);\n case \"<\":\n return Drizzle.lt(left, right);\n case \"<=\":\n return Drizzle.lte(left, right);\n case \"in\": {\n return Drizzle.inArray(left, right as never[]);\n }\n case \"not in\":\n return Drizzle.notInArray(left, right as never[]);\n case \"is\":\n return right === null ? Drizzle.isNull(left) : Drizzle.eq(left, right);\n case \"is not\":\n return right === null ? Drizzle.isNotNull(left) : Drizzle.ne(left, right);\n case \"contains\": {\n right =\n typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n return Drizzle.like(left, right as string);\n }\n case \"not contains\": {\n right =\n typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n return Drizzle.notLike(left, right as string);\n }\n case \"ends with\": {\n right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n return Drizzle.like(left, right as string);\n }\n case \"not ends with\": {\n right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n return Drizzle.notLike(left, right as string);\n }\n case \"starts with\": {\n right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n return Drizzle.like(left, right as string);\n }\n case \"not starts with\": {\n right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n return Drizzle.notLike(left, right as string);\n }\n\n default:\n throw new Error(`Unsupported operator: ${op}`);\n }\n }\n\n if (condition.type === \"and\") {\n return Drizzle.and(...condition.items.map((item) => buildWhere(item)));\n }\n\n if (condition.type === \"not\") {\n const result = buildWhere(condition.item);\n if (!result) return;\n\n return Drizzle.not(result);\n }\n\n return Drizzle.or(...condition.items.map((item) => buildWhere(item)));\n }\n\n /**\n * Process reference subqueries in encoded values, converting them to Drizzle SQL subqueries\n */\n function processReferenceSubqueries(values: Record<string, unknown>): Record<string, unknown> {\n const processed: Record<string, unknown> = {};\n\n for (const [key, value] of Object.entries(values)) {\n if (value instanceof ReferenceSubquery) {\n const refTable = value.referencedTable;\n const externalId = value.externalIdValue;\n const internalIdCol = refTable.getInternalIdColumn();\n const idCol = refTable.getIdColumn();\n const drizzleRefTable = toDrizzleTable(refTable);\n const drizzleIdCol = toDrizzleColumn(idCol);\n\n // Create a parameterized SQL subquery using Drizzle's query builder\n // Safe cast: we're building a SQL subquery that returns a single bigint value\n processed[key] = db\n .select({ value: drizzleRefTable[internalIdCol.ormName] })\n .from(drizzleRefTable)\n .where(Drizzle.eq(drizzleIdCol, externalId))\n .limit(1) as unknown;\n } else {\n processed[key] = value;\n }\n }\n\n return processed;\n }\n\n /**\n * Get table from schema by name\n * @throws Error if table is not found in schema\n */\n function getTable(name: unknown): AnyTable {\n const table = schema.tables[name as string];\n if (!table) {\n throw new Error(`Invalid table name ${name}.`);\n }\n return table;\n }\n\n /**\n * Get the version to check for a given ID and checkVersion flag.\n * @returns The version to check or undefined if no check is required.\n * @throws Error if the ID is a string and checkVersion is true.\n */\n function getVersionToCheck(id: FragnoId | string, checkVersion: boolean): number | undefined {\n if (!checkVersion) {\n return undefined;\n }\n\n if (typeof id === \"string\") {\n throw new Error(\n `Cannot use checkVersion with a string ID. Version checking requires a FragnoId with version information.`,\n );\n }\n\n return id.version;\n }\n\n /**\n * Process joins recursively to support nested joins with orderBy and limit\n */\n function processJoins(\n joins: CompiledJoin[],\n ): Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> {\n const result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> = {};\n\n for (const join of joins) {\n const { options, relation } = join;\n\n if (!options) {\n continue;\n }\n\n const targetTable = relation.table;\n const joinName = relation.name;\n\n // Build columns for this join using shared utility\n const selectOption = options.select === undefined ? true : options.select;\n const orderedColumns = getOrderedJoinColumns(targetTable, selectOption);\n const joinColumns: Record<string, boolean> = {};\n for (const colName of orderedColumns) {\n joinColumns[colName] = true;\n }\n\n // Build orderBy for this join\n let joinOrderBy: Drizzle.SQL[] | undefined;\n if (options.orderBy && options.orderBy.length > 0) {\n joinOrderBy = options.orderBy.map(([col, direction]) => {\n const drizzleCol = toDrizzleColumn(col);\n return direction === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n });\n }\n\n // Build WHERE clause for this join if provided\n let joinWhere: Drizzle.SQL | undefined;\n if (options.where) {\n joinWhere = buildWhere(options.where);\n }\n\n // Build the join config\n const joinConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n columns: joinColumns,\n orderBy: joinOrderBy,\n limit: options.limit,\n where: joinWhere,\n };\n\n // Recursively process nested joins\n if (options.join && options.join.length > 0) {\n joinConfig.with = processJoins(options.join);\n }\n\n result[joinName] = joinConfig;\n }\n\n return result;\n }\n\n return {\n compileRetrievalOperation(op: RetrievalOperation<TSchema>): DrizzleCompiledQuery | null {\n switch (op.type) {\n case \"count\": {\n // Build WHERE clause\n let whereClause: Drizzle.SQL | undefined;\n if (op.options.where) {\n const condition = buildCondition(op.table.columns, op.options.where);\n if (condition === false) {\n // Never matches - return null\n return null;\n }\n if (condition !== true) {\n whereClause = buildWhere(condition);\n }\n }\n\n const drizzleTable = toDrizzleTable(op.table);\n const query = db.select({ count: Drizzle.count() }).from(drizzleTable);\n\n const compiledQuery = whereClause ? query.where(whereClause).toSQL() : query.toSQL();\n onQuery?.(compiledQuery);\n return compiledQuery;\n }\n\n case \"find\": {\n const {\n useIndex: _useIndex,\n orderByIndex,\n joins,\n after,\n before,\n pageSize,\n ...findOptions\n } = op.options;\n\n // Get index columns for ordering and cursor pagination\n let indexColumns: AnyColumn[] = [];\n let orderDirection: \"asc\" | \"desc\" = \"asc\";\n\n if (orderByIndex) {\n const index = op.table.indexes[orderByIndex.indexName];\n orderDirection = orderByIndex.direction;\n\n if (!index) {\n // If _primary index doesn't exist, fall back to ID column\n if (orderByIndex.indexName === \"_primary\") {\n indexColumns = [op.table.getIdColumn()];\n } else {\n throw new Error(\n `Index \"${orderByIndex.indexName}\" not found on table \"${op.table.name}\"`,\n );\n }\n } else {\n indexColumns = index.columns;\n }\n }\n\n // Convert orderByIndex to orderBy format\n let orderBy: Drizzle.SQL[] | undefined;\n if (indexColumns.length > 0) {\n orderBy = indexColumns.map((col) => {\n const drizzleCol = toDrizzleColumn(col);\n return orderDirection === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n });\n }\n\n // Build query configuration\n const columns: Record<string, boolean> = {};\n const select = findOptions.select;\n\n if (select === true || select === undefined) {\n for (const col of Object.values(op.table.columns)) {\n columns[col.ormName] = true;\n }\n } else {\n for (const k of select) {\n columns[op.table.columns[k].ormName] = true;\n }\n // Always include hidden columns (for FragnoId construction with internal ID and version)\n for (const col of Object.values(op.table.columns)) {\n if (col.isHidden && !columns[col.ormName]) {\n columns[col.ormName] = true;\n }\n }\n }\n\n // Build WHERE clause with cursor conditions\n const whereClauses: Drizzle.SQL[] = [];\n\n // Add user-defined where clause\n if (findOptions.where) {\n const condition = buildCondition(op.table.columns, findOptions.where);\n if (condition === false) {\n // Never matches - return null to indicate this query should be skipped\n return null;\n }\n if (condition !== true) {\n const clause = buildWhere(condition);\n if (clause) {\n whereClauses.push(clause);\n }\n }\n }\n\n // Add cursor-based pagination conditions\n if ((after || before) && indexColumns.length > 0) {\n const cursor = after || before;\n const cursorData = decodeCursor(cursor!);\n const serializedValues = serializeCursorValues(cursorData, indexColumns, provider);\n\n // Build tuple comparison for cursor pagination\n // For \"after\" with \"asc\": (col1, col2, ...) > (val1, val2, ...)\n // For \"before\" with \"desc\": reverse the comparison\n const isAfter = !!after;\n const useGreaterThan =\n (isAfter && orderDirection === \"asc\") || (!isAfter && orderDirection === \"desc\");\n\n if (indexColumns.length === 1) {\n // Simple single-column case\n const col = toDrizzleColumn(indexColumns[0]!);\n const val = serializedValues[indexColumns[0]!.ormName];\n whereClauses.push(useGreaterThan ? Drizzle.gt(col, val) : Drizzle.lt(col, val));\n } else {\n // Multi-column tuple comparison using SQL\n const drizzleCols = indexColumns.map((c) => toDrizzleColumn(c));\n const vals = indexColumns.map((c) => serializedValues[c.ormName]);\n const operator = useGreaterThan ? \">\" : \"<\";\n // Safe cast: building a SQL comparison expression for cursor pagination\n // Build the tuple comparison: (col1, col2) > (val1, val2)\n const colsSQL = Drizzle.sql.join(drizzleCols, Drizzle.sql.raw(\", \"));\n const valsSQL = Drizzle.sql.join(\n vals.map((v) => Drizzle.sql`${v}`),\n Drizzle.sql.raw(\", \"),\n );\n whereClauses.push(\n Drizzle.sql`(${colsSQL}) ${Drizzle.sql.raw(operator)} (${valsSQL})`,\n );\n }\n }\n\n const whereClause = whereClauses.length > 0 ? Drizzle.and(...whereClauses) : undefined;\n\n const queryConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n columns,\n limit: pageSize,\n where: whereClause,\n orderBy,\n with: {},\n };\n\n // Process joins recursively to support nested joins\n if (joins) {\n queryConfig.with = processJoins(joins);\n }\n\n const compiledQuery = db.query[op.table.ormName].findMany(queryConfig).toSQL();\n onQuery?.(compiledQuery);\n return compiledQuery;\n }\n }\n },\n\n compileMutationOperation(\n op: MutationOperation<TSchema>,\n ): CompiledMutation<DrizzleCompiledQuery> | null {\n switch (op.type) {\n case \"create\": {\n const table = getTable(op.table);\n const drizzleTable = toDrizzleTable(table);\n // encodeValues now handles runtime defaults automatically\n const encodedValues = encodeValues(op.values, table, true, provider);\n const values = processReferenceSubqueries(encodedValues);\n\n const compiledQuery = db.insert(drizzleTable).values(values).toSQL();\n onQuery?.(compiledQuery);\n return {\n query: compiledQuery,\n expectedAffectedRows: null, // creates don't need affected row checks\n };\n }\n\n case \"update\": {\n const table = getTable(op.table);\n const idColumn = table.getIdColumn();\n const versionColumn = table.getVersionColumn();\n const drizzleTable = toDrizzleTable(table);\n\n const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n // Build WHERE clause that filters by ID and optionally by version\n const condition =\n versionToCheck !== undefined\n ? buildCondition(table.columns, (eb) =>\n eb.and(\n eb(idColumn.ormName, \"=\", externalId),\n eb(versionColumn.ormName, \"=\", versionToCheck),\n ),\n )\n : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n // Handle boolean cases\n if (condition === false) {\n // Never matches - skip this operation\n return null;\n }\n\n const whereClause = condition === true ? undefined : buildWhere(condition);\n const encodedSetValues = encodeValues(op.set, table, false, provider);\n const setValues = processReferenceSubqueries(encodedSetValues);\n\n // Automatically increment _version for optimistic concurrency control\n // Safe cast: we're building a SQL expression for incrementing the version\n setValues[versionColumn.ormName] = Drizzle.sql.raw(\n `COALESCE(${versionColumn.ormName}, 0) + 1`,\n ) as unknown;\n\n const compiledQuery = db.update(drizzleTable).set(setValues).where(whereClause).toSQL();\n onQuery?.(compiledQuery);\n return {\n query: compiledQuery,\n expectedAffectedRows: op.checkVersion ? 1 : null,\n };\n }\n\n case \"delete\": {\n const table = getTable(op.table);\n const idColumn = table.getIdColumn();\n const versionColumn = table.getVersionColumn();\n const drizzleTable = toDrizzleTable(table);\n\n const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n // Build WHERE clause that filters by ID and optionally by version\n const condition =\n versionToCheck !== undefined\n ? buildCondition(table.columns, (eb) =>\n eb.and(\n eb(idColumn.ormName, \"=\", externalId),\n eb(versionColumn.ormName, \"=\", versionToCheck),\n ),\n )\n : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n // Handle boolean cases\n if (condition === false) {\n // Never matches - skip this operation\n return null;\n }\n\n const whereClause = condition === true ? undefined : buildWhere(condition);\n\n const compiledQuery = db.delete(drizzleTable).where(whereClause).toSQL();\n onQuery?.(compiledQuery);\n return {\n query: compiledQuery,\n expectedAffectedRows: op.checkVersion ? 1 : null,\n };\n }\n }\n },\n };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAkCA,SAAgB,yBACd,QACA,QACA,SAC4C;CAC5C,MAAM,CAAC,IAAI,iBAAiB,aAAa,OAAO,GAAG;CACnD,MAAM,EAAE,aAAa;;;;;CAMrB,SAAS,eAAe,OAA4B;EAClD,MAAM,YAAY,MAAM;EACxB,MAAM,MAAM,cAAc;AAC1B,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MACR,gCAAgC,UAAU,0CAC3C;;;;;;CAOH,SAAS,gBAAgB,KAA4B;EACnD,MAAM,cAAc,OAAO,OAAO,IAAI;AACtC,MAAI,CAAC,YACH,OAAM,IAAI,MAAM,2BAA2B,IAAI,UAAU,cAAc,IAAI,QAAQ,GAAG;EAIxF,MAAM,MADQ,eAAe,YAAY,CACvB,IAAI;AACtB,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MAAM,iCAAiC,IAAI,QAAQ,MAAM,YAAY,QAAQ,GAAG;;;;;CAM5F,SAAS,WAAW,WAA+C;AACjE,MAAI,UAAU,SAAS,WAAW;GAChC,MAAM,OAAO,gBAAgB,UAAU,EAAE;GACzC,MAAM,KAAK,UAAU;GACrB,IAAI,QAAQ,UAAU;AACtB,OAAI,iBAAiB,OACnB,SAAQ,gBAAgB,MAAM;OAG9B,SAAQ,UAAU,OAAO,UAAU,GAAG,SAAS;AAGjD,WAAQ,IAAR;IACE,KAAK,IACH,QAAOA,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,KACH,QAAOC,QAAgB,MAAM,MAAiB;IAEhD,KAAK,SACH,QAAOC,WAAmB,MAAM,MAAiB;IACnD,KAAK,KACH,QAAO,UAAU,OAAOC,OAAe,KAAK,GAAGR,GAAW,MAAM,MAAM;IACxE,KAAK,SACH,QAAO,UAAU,OAAOS,UAAkB,KAAK,GAAGR,GAAW,MAAM,MAAM;IAC3E,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOS,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAG/C,QACE,OAAM,IAAI,MAAM,yBAAyB,KAAK;;;AAIpD,MAAI,UAAU,SAAS,MACrB,QAAOC,IAAY,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;AAGxE,MAAI,UAAU,SAAS,OAAO;GAC5B,MAAM,SAAS,WAAW,UAAU,KAAK;AACzC,OAAI,CAAC,OAAQ;AAEb,UAAOC,IAAY,OAAO;;AAG5B,SAAOC,GAAW,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;;;;;CAMvE,SAAS,2BAA2B,QAA0D;EAC5F,MAAMC,YAAqC,EAAE;AAE7C,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,OAAO,CAC/C,KAAI,iBAAiB,mBAAmB;GACtC,MAAM,WAAW,MAAM;GACvB,MAAM,aAAa,MAAM;GACzB,MAAM,gBAAgB,SAAS,qBAAqB;GACpD,MAAM,QAAQ,SAAS,aAAa;GACpC,MAAM,kBAAkB,eAAe,SAAS;GAChD,MAAM,eAAe,gBAAgB,MAAM;AAI3C,aAAU,OAAO,GACd,OAAO,EAAE,OAAO,gBAAgB,cAAc,UAAU,CAAC,CACzD,KAAK,gBAAgB,CACrB,MAAMf,GAAW,cAAc,WAAW,CAAC,CAC3C,MAAM,EAAE;QAEX,WAAU,OAAO;AAIrB,SAAO;;;;;;CAOT,SAAS,SAAS,MAAyB;EACzC,MAAM,QAAQ,OAAO,OAAO;AAC5B,MAAI,CAAC,MACH,OAAM,IAAI,MAAM,sBAAsB,KAAK,GAAG;AAEhD,SAAO;;;;;;;CAQT,SAAS,kBAAkB,IAAuB,cAA2C;AAC3F,MAAI,CAAC,aACH;AAGF,MAAI,OAAO,OAAO,SAChB,OAAM,IAAI,MACR,2GACD;AAGH,SAAO,GAAG;;;;;CAMZ,SAAS,aACP,OACwD;EACxD,MAAMgB,SAAiE,EAAE;AAEzE,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,EAAE,SAAS,aAAa;AAE9B,OAAI,CAAC,QACH;GAGF,MAAM,cAAc,SAAS;GAC7B,MAAM,WAAW,SAAS;GAI1B,MAAM,iBAAiB,sBAAsB,aADxB,QAAQ,WAAW,SAAY,OAAO,QAAQ,OACI;GACvE,MAAMC,cAAuC,EAAE;AAC/C,QAAK,MAAM,WAAW,eACpB,aAAY,WAAW;GAIzB,IAAIC;AACJ,OAAI,QAAQ,WAAW,QAAQ,QAAQ,SAAS,EAC9C,eAAc,QAAQ,QAAQ,KAAK,CAAC,KAAK,eAAe;IACtD,MAAM,aAAa,gBAAgB,IAAI;AACvC,WAAO,cAAc,QAAQC,IAAY,WAAW,GAAGC,KAAa,WAAW;KAC/E;GAIJ,IAAIC;AACJ,OAAI,QAAQ,MACV,aAAY,WAAW,QAAQ,MAAM;GAIvC,MAAMC,aAAqD;IACzD,SAAS;IACT,SAAS;IACT,OAAO,QAAQ;IACf,OAAO;IACR;AAGD,OAAI,QAAQ,QAAQ,QAAQ,KAAK,SAAS,EACxC,YAAW,OAAO,aAAa,QAAQ,KAAK;AAG9C,UAAO,YAAY;;AAGrB,SAAO;;AAGT,QAAO;EACL,0BAA0B,IAA8D;AACtF,WAAQ,GAAG,MAAX;IACE,KAAK,SAAS;KAEZ,IAAIC;AACJ,SAAI,GAAG,QAAQ,OAAO;MACpB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,GAAG,QAAQ,MAAM;AACpE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,KAChB,eAAc,WAAW,UAAU;;KAIvC,MAAM,eAAe,eAAe,GAAG,MAAM;KAC7C,MAAM,QAAQ,GAAG,OAAO,EAAE,OAAOC,OAAe,EAAE,CAAC,CAAC,KAAK,aAAa;KAEtE,MAAM,gBAAgB,cAAc,MAAM,MAAM,YAAY,CAAC,OAAO,GAAG,MAAM,OAAO;AACpF,eAAU,cAAc;AACxB,YAAO;;IAGT,KAAK,QAAQ;KACX,MAAM,EACJ,UAAU,WACV,cACA,OACA,OACA,QACA,SACA,GAAG,gBACD,GAAG;KAGP,IAAIC,eAA4B,EAAE;KAClC,IAAIC,iBAAiC;AAErC,SAAI,cAAc;MAChB,MAAM,QAAQ,GAAG,MAAM,QAAQ,aAAa;AAC5C,uBAAiB,aAAa;AAE9B,UAAI,CAAC,MAEH,KAAI,aAAa,cAAc,WAC7B,gBAAe,CAAC,GAAG,MAAM,aAAa,CAAC;UAEvC,OAAM,IAAI,MACR,UAAU,aAAa,UAAU,wBAAwB,GAAG,MAAM,KAAK,GACxE;UAGH,gBAAe,MAAM;;KAKzB,IAAIC;AACJ,SAAI,aAAa,SAAS,EACxB,WAAU,aAAa,KAAK,QAAQ;MAClC,MAAM,aAAa,gBAAgB,IAAI;AACvC,aAAO,mBAAmB,QAAQR,IAAY,WAAW,GAAGC,KAAa,WAAW;OACpF;KAIJ,MAAMQ,UAAmC,EAAE;KAC3C,MAAM,SAAS,YAAY;AAE3B,SAAI,WAAW,QAAQ,WAAW,OAChC,MAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,SAAQ,IAAI,WAAW;UAEpB;AACL,WAAK,MAAM,KAAK,OACd,SAAQ,GAAG,MAAM,QAAQ,GAAG,WAAW;AAGzC,WAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,KAAI,IAAI,YAAY,CAAC,QAAQ,IAAI,SAC/B,SAAQ,IAAI,WAAW;;KAM7B,MAAMC,eAA8B,EAAE;AAGtC,SAAI,YAAY,OAAO;MACrB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,YAAY,MAAM;AACrE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,MAAM;OACtB,MAAM,SAAS,WAAW,UAAU;AACpC,WAAI,OACF,cAAa,KAAK,OAAO;;;AAM/B,UAAK,SAAS,WAAW,aAAa,SAAS,GAAG;MAGhD,MAAM,mBAAmB,sBADN,aADJ,SAAS,OACgB,EACmB,cAAc,SAAS;MAKlF,MAAM,UAAU,CAAC,CAAC;MAClB,MAAM,iBACH,WAAW,mBAAmB,SAAW,CAAC,WAAW,mBAAmB;AAE3E,UAAI,aAAa,WAAW,GAAG;OAE7B,MAAM,MAAM,gBAAgB,aAAa,GAAI;OAC7C,MAAM,MAAM,iBAAiB,aAAa,GAAI;AAC9C,oBAAa,KAAK,iBAAiB3B,GAAW,KAAK,IAAI,GAAGE,GAAW,KAAK,IAAI,CAAC;aAC1E;OAEL,MAAM,cAAc,aAAa,KAAK,MAAM,gBAAgB,EAAE,CAAC;OAC/D,MAAM,OAAO,aAAa,KAAK,MAAM,iBAAiB,EAAE,SAAS;OACjE,MAAM,WAAW,iBAAiB,MAAM;OAGxC,MAAM,cAAsB,KAAK,iBAAyB,IAAI,KAAK,CAAC;OACpE,MAAM,cAAsB,KAC1B,KAAK,KAAK,MAAM,GAAW,GAAG,IAAI,MACtB,IAAI,KAAK,CACtB;AACD,oBAAa,KACX,GAAW,IAAI,QAAQ,QAAgB,IAAI,SAAS,CAAC,IAAI,QAAQ,GAClE;;;KAML,MAAM0B,cAAsD;MAC1D;MACA,OAAO;MACP,OALkB,aAAa,SAAS,IAAIlB,IAAY,GAAG,aAAa,GAAG;MAM3E;MACA,MAAM,EAAE;MACT;AAGD,SAAI,MACF,aAAY,OAAO,aAAa,MAAM;KAGxC,MAAM,gBAAgB,GAAG,MAAM,GAAG,MAAM,SAAS,SAAS,YAAY,CAAC,OAAO;AAC9E,eAAU,cAAc;AACxB,YAAO;;;;EAKb,yBACE,IAC+C;AAC/C,WAAQ,GAAG,MAAX;IACE,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,eAAe,eAAe,MAAM;KAG1C,MAAM,SAAS,2BADO,aAAa,GAAG,QAAQ,OAAO,MAAM,SAAS,CACZ;KAExD,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,OAAO,OAAO,CAAC,OAAO;AACpE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB;MACvB;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,YAAY,2BADO,aAAa,GAAG,KAAK,OAAO,OAAO,SAAS,CACP;AAI9D,eAAU,cAAc,eAAuB,IAC7C,YAAY,cAAc,QAAQ,UACnC;KAED,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,IAAI,UAAU,CAAC,MAAM,YAAY,CAAC,OAAO;AACvF,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,MAAM,YAAY,CAAC,OAAO;AACxE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;;;EAIR"}
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
import { decodeResult } from "../../query/result-transform.js";
|
|
2
|
+
import { getOrderedJoinColumns } from "./join-column-utils.js";
|
|
3
|
+
|
|
4
|
+
//#region src/adapters/drizzle/drizzle-uow-decoder.ts
|
|
5
|
+
/**
|
|
6
|
+
* Recursively transform join arrays to objects, handling nested joins.
|
|
7
|
+
*
|
|
8
|
+
* Drizzle joins use `json_build_array` where nested join data is appended after the parent's columns.
|
|
9
|
+
* For example, if post has columns [id, title, content, _internalId, _version] and a nested author join,
|
|
10
|
+
* the array will be: [id, title, content, _internalId, _version, authorArray]
|
|
11
|
+
*
|
|
12
|
+
* @param value - The join array from Drizzle
|
|
13
|
+
* @param joinInfo - Join metadata including nested joins
|
|
14
|
+
* @param relationName - Name of the current relation (for prefixing column names)
|
|
15
|
+
* @returns Object with flattened keys (relationName:columnName) for all levels
|
|
16
|
+
*/
|
|
17
|
+
function transformJoinArray(value, joinInfo, relationName) {
|
|
18
|
+
const result = {};
|
|
19
|
+
if (joinInfo.options === false) return result;
|
|
20
|
+
const targetTable = joinInfo.relation.table;
|
|
21
|
+
const orderedSelectedColumns = getOrderedJoinColumns(targetTable, joinInfo.options.select);
|
|
22
|
+
for (let i = 0; i < orderedSelectedColumns.length && i < value.length; i++) {
|
|
23
|
+
const columnName = orderedSelectedColumns[i];
|
|
24
|
+
if (columnName) result[`${relationName}:${columnName}`] = value[i];
|
|
25
|
+
}
|
|
26
|
+
if (joinInfo.options.join && joinInfo.options.join.length > 0) {
|
|
27
|
+
let nestedArrayIndex = orderedSelectedColumns.length;
|
|
28
|
+
for (const nestedJoin of joinInfo.options.join) {
|
|
29
|
+
const nestedRelationName = `${relationName}:${nestedJoin.relation.name}`;
|
|
30
|
+
const nestedValue = value[nestedArrayIndex];
|
|
31
|
+
if (Array.isArray(nestedValue)) {
|
|
32
|
+
const nestedResult = transformJoinArray(nestedValue, nestedJoin, nestedRelationName);
|
|
33
|
+
Object.assign(result, nestedResult);
|
|
34
|
+
}
|
|
35
|
+
nestedArrayIndex++;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
return result;
|
|
39
|
+
}
|
|
40
|
+
/**
|
|
41
|
+
* Drizzle joins using `json_build_array` so the result is a tuple of values that we need to map to
|
|
42
|
+
* the correct columns. This function handles nested joins recursively.
|
|
43
|
+
*
|
|
44
|
+
* @param row - Raw database result row that may contain join arrays
|
|
45
|
+
* @param op - The retrieval operation containing join information
|
|
46
|
+
* @returns Transformed row with join arrays converted to objects
|
|
47
|
+
*/
|
|
48
|
+
function transformJoinArraysToObjects(row, op) {
|
|
49
|
+
if (op.type !== "find" || !op.options?.joins) return row;
|
|
50
|
+
const transformedRow = { ...row };
|
|
51
|
+
for (const join of op.options.joins) {
|
|
52
|
+
const relationName = join.relation.name;
|
|
53
|
+
const value = row[relationName];
|
|
54
|
+
if (!Array.isArray(value)) continue;
|
|
55
|
+
if (join.options === false) continue;
|
|
56
|
+
if (!op.table.relations[relationName]) continue;
|
|
57
|
+
const joinResult = transformJoinArray(value, join, relationName);
|
|
58
|
+
Object.assign(transformedRow, joinResult);
|
|
59
|
+
delete transformedRow[relationName];
|
|
60
|
+
}
|
|
61
|
+
return transformedRow;
|
|
62
|
+
}
|
|
63
|
+
function createDrizzleUOWDecoder(_schema, provider) {
|
|
64
|
+
return (rawResults, ops) => {
|
|
65
|
+
if (rawResults.length !== ops.length) throw new Error("rawResults and ops must have the same length");
|
|
66
|
+
return rawResults.map((result, index) => {
|
|
67
|
+
const op = ops[index];
|
|
68
|
+
if (!op) throw new Error("op must be defined");
|
|
69
|
+
if (op.type === "count") {
|
|
70
|
+
if (result.rows.length > 0 && result.rows[0]) return result.rows[0]["count"];
|
|
71
|
+
return 0;
|
|
72
|
+
}
|
|
73
|
+
return result.rows.map((row) => {
|
|
74
|
+
return decodeResult(transformJoinArraysToObjects(row, op), op.table, provider);
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
//#endregion
|
|
81
|
+
export { createDrizzleUOWDecoder };
|
|
82
|
+
//# sourceMappingURL=drizzle-uow-decoder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"drizzle-uow-decoder.js","names":["result: Record<string, unknown>"],"sources":["../../../src/adapters/drizzle/drizzle-uow-decoder.ts"],"sourcesContent":["import type { AnySchema, AnyTable } from \"../../schema/create\";\nimport type { SQLProvider } from \"../../shared/providers\";\nimport type { RetrievalOperation, UOWDecoder } from \"../../query/unit-of-work\";\nimport { decodeResult } from \"../../query/result-transform\";\nimport type { DrizzleResult } from \"./drizzle-query\";\nimport { getOrderedJoinColumns } from \"./join-column-utils\";\n\n/**\n * Join information with nested join support\n */\ninterface JoinInfo {\n relation: { name: string; table: AnyTable };\n options:\n | {\n select: true | string[];\n join?: JoinInfo[];\n }\n | false;\n}\n\n/**\n * Recursively transform join arrays to objects, handling nested joins.\n *\n * Drizzle joins use `json_build_array` where nested join data is appended after the parent's columns.\n * For example, if post has columns [id, title, content, _internalId, _version] and a nested author join,\n * the array will be: [id, title, content, _internalId, _version, authorArray]\n *\n * @param value - The join array from Drizzle\n * @param joinInfo - Join metadata including nested joins\n * @param relationName - Name of the current relation (for prefixing column names)\n * @returns Object with flattened keys (relationName:columnName) for all levels\n */\nfunction transformJoinArray(\n value: unknown[],\n joinInfo: JoinInfo,\n relationName: string,\n): Record<string, unknown> {\n const result: Record<string, unknown> = {};\n\n if (joinInfo.options === false) {\n return result;\n }\n\n const targetTable = joinInfo.relation.table;\n\n // Get ordered columns using shared utility (must match compiler's column order)\n const orderedSelectedColumns = getOrderedJoinColumns(targetTable, joinInfo.options.select);\n\n // Map column values to flattened format: relationName:columnName\n for (let i = 0; i < orderedSelectedColumns.length && i < value.length; i++) {\n const columnName = orderedSelectedColumns[i];\n if (columnName) {\n result[`${relationName}:${columnName}`] = value[i];\n }\n }\n\n // Handle nested joins - they appear after all columns in the array\n if (joinInfo.options.join && joinInfo.options.join.length > 0) {\n let nestedArrayIndex = orderedSelectedColumns.length;\n\n for (const nestedJoin of joinInfo.options.join) {\n const nestedRelationName = `${relationName}:${nestedJoin.relation.name}`;\n const nestedValue = value[nestedArrayIndex];\n\n if (Array.isArray(nestedValue)) {\n // Recursively transform nested join\n const nestedResult = transformJoinArray(nestedValue, nestedJoin, nestedRelationName);\n Object.assign(result, nestedResult);\n }\n\n nestedArrayIndex++;\n }\n }\n\n return result;\n}\n\n/**\n * Drizzle joins using `json_build_array` so the result is a tuple of values that we need to map to\n * the correct columns. This function handles nested joins recursively.\n *\n * @param row - Raw database result row that may contain join arrays\n * @param op - The retrieval operation containing join information\n * @returns Transformed row with join arrays converted to objects\n */\nfunction transformJoinArraysToObjects(\n row: Record<string, unknown>,\n op: {\n type: string;\n table: AnyTable;\n options?: {\n joins?: JoinInfo[];\n };\n },\n): Record<string, unknown> {\n // Only process find operations with joins\n if (op.type !== \"find\" || !op.options?.joins) {\n return row;\n }\n\n const transformedRow = { ...row };\n\n for (const join of op.options.joins) {\n const relationName = join.relation.name;\n const value = row[relationName];\n\n // Skip if not an array (join didn't return data)\n if (!Array.isArray(value)) {\n continue;\n }\n\n // Skip if join options are false (join was disabled)\n if (join.options === false) {\n continue;\n }\n\n // Get the target table for this relation\n const relation = op.table.relations[relationName];\n if (!relation) {\n continue;\n }\n\n // Recursively transform this join and its nested joins\n const joinResult = transformJoinArray(value, join, relationName);\n Object.assign(transformedRow, joinResult);\n\n // Remove the original array property\n delete transformedRow[relationName];\n }\n\n return transformedRow;\n}\n\nexport function createDrizzleUOWDecoder<TSchema extends AnySchema>(\n _schema: TSchema,\n provider: SQLProvider,\n): UOWDecoder<TSchema, DrizzleResult> {\n return (rawResults, ops) => {\n if (rawResults.length !== ops.length) {\n throw new Error(\"rawResults and ops must have the same length\");\n }\n\n return rawResults.map((result, index) => {\n const op = ops[index] as RetrievalOperation<TSchema>;\n if (!op) {\n throw new Error(\"op must be defined\");\n }\n\n // Handle count operations - return the count value directly\n if (op.type === \"count\") {\n if (result.rows.length > 0 && result.rows[0]) {\n const row = result.rows[0];\n return (row as Record<string, unknown>)[\"count\"] as number;\n }\n return 0;\n }\n\n // Handle find operations - decode each row\n return result.rows.map((row) => {\n const transformedRow = transformJoinArraysToObjects(row, op);\n return decodeResult(transformedRow, op.table, provider);\n });\n });\n };\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAgCA,SAAS,mBACP,OACA,UACA,cACyB;CACzB,MAAMA,SAAkC,EAAE;AAE1C,KAAI,SAAS,YAAY,MACvB,QAAO;CAGT,MAAM,cAAc,SAAS,SAAS;CAGtC,MAAM,yBAAyB,sBAAsB,aAAa,SAAS,QAAQ,OAAO;AAG1F,MAAK,IAAI,IAAI,GAAG,IAAI,uBAAuB,UAAU,IAAI,MAAM,QAAQ,KAAK;EAC1E,MAAM,aAAa,uBAAuB;AAC1C,MAAI,WACF,QAAO,GAAG,aAAa,GAAG,gBAAgB,MAAM;;AAKpD,KAAI,SAAS,QAAQ,QAAQ,SAAS,QAAQ,KAAK,SAAS,GAAG;EAC7D,IAAI,mBAAmB,uBAAuB;AAE9C,OAAK,MAAM,cAAc,SAAS,QAAQ,MAAM;GAC9C,MAAM,qBAAqB,GAAG,aAAa,GAAG,WAAW,SAAS;GAClE,MAAM,cAAc,MAAM;AAE1B,OAAI,MAAM,QAAQ,YAAY,EAAE;IAE9B,MAAM,eAAe,mBAAmB,aAAa,YAAY,mBAAmB;AACpF,WAAO,OAAO,QAAQ,aAAa;;AAGrC;;;AAIJ,QAAO;;;;;;;;;;AAWT,SAAS,6BACP,KACA,IAOyB;AAEzB,KAAI,GAAG,SAAS,UAAU,CAAC,GAAG,SAAS,MACrC,QAAO;CAGT,MAAM,iBAAiB,EAAE,GAAG,KAAK;AAEjC,MAAK,MAAM,QAAQ,GAAG,QAAQ,OAAO;EACnC,MAAM,eAAe,KAAK,SAAS;EACnC,MAAM,QAAQ,IAAI;AAGlB,MAAI,CAAC,MAAM,QAAQ,MAAM,CACvB;AAIF,MAAI,KAAK,YAAY,MACnB;AAKF,MAAI,CADa,GAAG,MAAM,UAAU,cAElC;EAIF,MAAM,aAAa,mBAAmB,OAAO,MAAM,aAAa;AAChE,SAAO,OAAO,gBAAgB,WAAW;AAGzC,SAAO,eAAe;;AAGxB,QAAO;;AAGT,SAAgB,wBACd,SACA,UACoC;AACpC,SAAQ,YAAY,QAAQ;AAC1B,MAAI,WAAW,WAAW,IAAI,OAC5B,OAAM,IAAI,MAAM,+CAA+C;AAGjE,SAAO,WAAW,KAAK,QAAQ,UAAU;GACvC,MAAM,KAAK,IAAI;AACf,OAAI,CAAC,GACH,OAAM,IAAI,MAAM,qBAAqB;AAIvC,OAAI,GAAG,SAAS,SAAS;AACvB,QAAI,OAAO,KAAK,SAAS,KAAK,OAAO,KAAK,GAExC,QADY,OAAO,KAAK,GACgB;AAE1C,WAAO;;AAIT,UAAO,OAAO,KAAK,KAAK,QAAQ;AAE9B,WAAO,aADgB,6BAA6B,KAAK,GAAG,EACxB,GAAG,OAAO,SAAS;KACvD;IACF"}
|
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
import { SQL, StringChunk, sql } from "../../node_modules/.bun/drizzle-orm@0.44.6_4fae081eecb963e2/node_modules/drizzle-orm/sql/sql.js";
|
|
2
|
+
|
|
3
|
+
//#region src/adapters/drizzle/drizzle-uow-executor.ts
|
|
4
|
+
/**
|
|
5
|
+
* Convert a DrizzleCompiledQuery (SQL string + params) to a Drizzle SQL object
|
|
6
|
+
*
|
|
7
|
+
* This reconstructs the SQL object with proper parameter chunks by parsing
|
|
8
|
+
* the SQL string and replacing placeholders ($1, $2, etc.) with Param objects.
|
|
9
|
+
* Uses Drizzle's exported classes (StringChunk, Param, SQL) to build the queryChunks.
|
|
10
|
+
*/
|
|
11
|
+
function toSQL(query) {
|
|
12
|
+
const { sql: sqlString, params } = query;
|
|
13
|
+
const placeholderRegex = /\$(\d+)/g;
|
|
14
|
+
const queryChunks = [];
|
|
15
|
+
let lastIndex = 0;
|
|
16
|
+
let match;
|
|
17
|
+
while ((match = placeholderRegex.exec(sqlString)) !== null) {
|
|
18
|
+
const textBefore = sqlString.substring(lastIndex, match.index);
|
|
19
|
+
if (textBefore) queryChunks.push(new StringChunk(textBefore));
|
|
20
|
+
const paramIndex = parseInt(match[1], 10) - 1;
|
|
21
|
+
queryChunks.push(sql.param(params[paramIndex]));
|
|
22
|
+
lastIndex = match.index + match[0].length;
|
|
23
|
+
}
|
|
24
|
+
const textAfter = sqlString.substring(lastIndex);
|
|
25
|
+
if (textAfter) queryChunks.push(new StringChunk(textAfter));
|
|
26
|
+
return new SQL(queryChunks);
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Get the number of affected rows from a Drizzle query result
|
|
30
|
+
*/
|
|
31
|
+
function getAffectedRows(result) {
|
|
32
|
+
if (Array.isArray(result)) return result.length;
|
|
33
|
+
if (result && typeof result === "object" && "affectedRows" in result && typeof result["affectedRows"] === "number") return result["affectedRows"];
|
|
34
|
+
if (result && typeof result === "object" && "rowCount" in result && (typeof result["rowCount"] === "number" || typeof result["rowCount"] === "bigint")) {
|
|
35
|
+
const rowCount = result["rowCount"];
|
|
36
|
+
if (rowCount > Number.MAX_SAFE_INTEGER) throw new Error(`rowCount BigInt value ${rowCount.toString()} exceeds JS safe integer range`);
|
|
37
|
+
return Number(rowCount);
|
|
38
|
+
}
|
|
39
|
+
if (result && typeof result === "object") {
|
|
40
|
+
if ("changes" in result && typeof result["changes"] === "number") return result["changes"];
|
|
41
|
+
}
|
|
42
|
+
throw new Error(`Unable to determine affected rows from result: ${JSON.stringify(result)}`);
|
|
43
|
+
}
|
|
44
|
+
/**
|
|
45
|
+
* Execute the retrieval phase of a Unit of Work using Drizzle
|
|
46
|
+
*
|
|
47
|
+
* All retrieval queries are executed inside a single transaction to ensure
|
|
48
|
+
* snapshot isolation - all reads see a consistent view of the database.
|
|
49
|
+
*
|
|
50
|
+
* @param db - The Drizzle database instance
|
|
51
|
+
* @param retrievalBatch - Array of Drizzle SQL queries
|
|
52
|
+
* @returns Array of query results matching the retrieval operations order
|
|
53
|
+
*
|
|
54
|
+
* @example
|
|
55
|
+
* ```ts
|
|
56
|
+
* const retrievalResults = await executeDrizzleRetrievalPhase(db, compiled.retrievalBatch);
|
|
57
|
+
* const [users, posts] = retrievalResults;
|
|
58
|
+
* ```
|
|
59
|
+
*/
|
|
60
|
+
async function executeDrizzleRetrievalPhase(db, retrievalBatch) {
|
|
61
|
+
if (retrievalBatch.length === 0) return [];
|
|
62
|
+
const retrievalResults = [];
|
|
63
|
+
await db.transaction(async (tx) => {
|
|
64
|
+
for (const query of retrievalBatch) {
|
|
65
|
+
const result = await tx.execute(toSQL(query));
|
|
66
|
+
retrievalResults.push(result);
|
|
67
|
+
}
|
|
68
|
+
});
|
|
69
|
+
return retrievalResults;
|
|
70
|
+
}
|
|
71
|
+
/**
|
|
72
|
+
* Execute the mutation phase of a Unit of Work using Drizzle
|
|
73
|
+
*
|
|
74
|
+
* All mutation queries are executed in a transaction with optimistic locking.
|
|
75
|
+
* If any version check fails, the entire transaction is rolled back and
|
|
76
|
+
* success=false is returned.
|
|
77
|
+
*
|
|
78
|
+
* @param db - The Drizzle database instance
|
|
79
|
+
* @param mutationBatch - Array of compiled mutation SQL queries with expected affected rows
|
|
80
|
+
* @returns Object with success flag and internal IDs from create operations
|
|
81
|
+
*
|
|
82
|
+
* @example
|
|
83
|
+
* ```ts
|
|
84
|
+
* const { success } = await executeDrizzleMutationPhase(db, compiled.mutationBatch);
|
|
85
|
+
* if (!success) {
|
|
86
|
+
* console.log("Version conflict detected, retrying...");
|
|
87
|
+
* }
|
|
88
|
+
* ```
|
|
89
|
+
*/
|
|
90
|
+
async function executeDrizzleMutationPhase(db, mutationBatch) {
|
|
91
|
+
if (mutationBatch.length === 0) return {
|
|
92
|
+
success: true,
|
|
93
|
+
createdInternalIds: []
|
|
94
|
+
};
|
|
95
|
+
const createdInternalIds = [];
|
|
96
|
+
try {
|
|
97
|
+
await db.transaction(async (tx) => {
|
|
98
|
+
for (const compiledMutation of mutationBatch) {
|
|
99
|
+
const result = await tx.execute(toSQL(compiledMutation.query));
|
|
100
|
+
if (compiledMutation.expectedAffectedRows === null) if (Array.isArray(result) && result.length > 0) {
|
|
101
|
+
const row = result[0];
|
|
102
|
+
if ("_internalId" in row || "_internal_id" in row) {
|
|
103
|
+
const internalId = row["_internalId"] ?? row["_internal_id"];
|
|
104
|
+
createdInternalIds.push(internalId);
|
|
105
|
+
} else createdInternalIds.push(null);
|
|
106
|
+
} else createdInternalIds.push(null);
|
|
107
|
+
else {
|
|
108
|
+
const affectedRows = getAffectedRows(result);
|
|
109
|
+
if (affectedRows !== compiledMutation.expectedAffectedRows) throw new Error(`Version conflict: expected ${compiledMutation.expectedAffectedRows} rows affected, but got ${affectedRows}`);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
});
|
|
113
|
+
return {
|
|
114
|
+
success: true,
|
|
115
|
+
createdInternalIds
|
|
116
|
+
};
|
|
117
|
+
} catch (error) {
|
|
118
|
+
if (error instanceof Error && error.message.includes("Version conflict")) return { success: false };
|
|
119
|
+
throw error;
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
//#endregion
|
|
124
|
+
export { executeDrizzleMutationPhase, executeDrizzleRetrievalPhase };
|
|
125
|
+
//# sourceMappingURL=drizzle-uow-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"drizzle-uow-executor.js","names":["queryChunks: SQLChunk[]","match: RegExpExecArray | null","retrievalResults: DrizzleResult[]","createdInternalIds: (bigint | null)[]"],"sources":["../../../src/adapters/drizzle/drizzle-uow-executor.ts"],"sourcesContent":["import { SQL, StringChunk, sql, type SQLChunk } from \"drizzle-orm\";\nimport type { CompiledMutation, MutationResult } from \"../../query/unit-of-work\";\nimport type { DBType } from \"./shared\";\nimport type { DrizzleCompiledQuery } from \"./drizzle-uow-compiler\";\nimport type { DrizzleResult } from \"./drizzle-query\";\n\n/**\n * Convert a DrizzleCompiledQuery (SQL string + params) to a Drizzle SQL object\n *\n * This reconstructs the SQL object with proper parameter chunks by parsing\n * the SQL string and replacing placeholders ($1, $2, etc.) with Param objects.\n * Uses Drizzle's exported classes (StringChunk, Param, SQL) to build the queryChunks.\n */\nfunction toSQL(query: DrizzleCompiledQuery): SQL {\n const { sql: sqlString, params } = query;\n\n // Match parameter placeholders like $1, $2, etc.\n const placeholderRegex = /\\$(\\d+)/g;\n const queryChunks: SQLChunk[] = [];\n let lastIndex = 0;\n let match: RegExpExecArray | null;\n\n while ((match = placeholderRegex.exec(sqlString)) !== null) {\n // Add the string chunk before the placeholder\n const textBefore = sqlString.substring(lastIndex, match.index);\n if (textBefore) {\n queryChunks.push(new StringChunk(textBefore));\n }\n\n // Add the parameter value as a Param\n const paramIndex = parseInt(match[1]!, 10) - 1; // $1 is index 0\n queryChunks.push(sql.param(params[paramIndex]));\n\n lastIndex = match.index + match[0].length;\n }\n\n // Add any remaining string after the last placeholder\n const textAfter = sqlString.substring(lastIndex);\n if (textAfter) {\n queryChunks.push(new StringChunk(textAfter));\n }\n\n // Construct SQL object directly with queryChunks (same pattern as sql`` tagged template)\n // Safe: We're reusing Drizzle's SQL constructor the same way the sql`` function does\n return new SQL(queryChunks);\n}\n\n/**\n * Get the number of affected rows from a Drizzle query result\n */\nfunction getAffectedRows(result: unknown): number {\n // Drizzle returns different formats depending on the database\n // For MySQL: array with affectedRows property\n // For PostgreSQL/SQLite: array with rowCount or similar\n if (Array.isArray(result)) {\n // This is likely a select/returning result\n return result.length;\n }\n\n // Check for MySQL-style result\n if (\n result &&\n typeof result === \"object\" &&\n \"affectedRows\" in result &&\n typeof result[\"affectedRows\"] === \"number\"\n ) {\n return result[\"affectedRows\"];\n }\n\n // Check for PostgreSQL-style result with rowCount\n if (\n result &&\n typeof result === \"object\" &&\n \"rowCount\" in result &&\n (typeof result[\"rowCount\"] === \"number\" || typeof result[\"rowCount\"] === \"bigint\")\n ) {\n const rowCount = result[\"rowCount\"];\n if (rowCount > Number.MAX_SAFE_INTEGER) {\n throw new Error(`rowCount BigInt value ${rowCount.toString()} exceeds JS safe integer range`);\n }\n return Number(rowCount);\n }\n\n // For update/delete operations, Drizzle might return an object with affected rows info\n // Try to extract it from common patterns\n if (result && typeof result === \"object\") {\n // Check for changes/changes count\n if (\"changes\" in result && typeof result[\"changes\"] === \"number\") {\n return result[\"changes\"];\n }\n }\n\n throw new Error(`Unable to determine affected rows from result: ${JSON.stringify(result)}`);\n}\n\n/**\n * Execute the retrieval phase of a Unit of Work using Drizzle\n *\n * All retrieval queries are executed inside a single transaction to ensure\n * snapshot isolation - all reads see a consistent view of the database.\n *\n * @param db - The Drizzle database instance\n * @param retrievalBatch - Array of Drizzle SQL queries\n * @returns Array of query results matching the retrieval operations order\n *\n * @example\n * ```ts\n * const retrievalResults = await executeDrizzleRetrievalPhase(db, compiled.retrievalBatch);\n * const [users, posts] = retrievalResults;\n * ```\n */\nexport async function executeDrizzleRetrievalPhase(\n db: DBType,\n retrievalBatch: DrizzleCompiledQuery[],\n): Promise<DrizzleResult[]> {\n // If no retrieval operations, return empty array immediately\n if (retrievalBatch.length === 0) {\n return [];\n }\n\n const retrievalResults: DrizzleResult[] = [];\n\n // Execute all retrieval queries inside a transaction for snapshot isolation\n await db.transaction(async (tx) => {\n for (const query of retrievalBatch) {\n const result = (await tx.execute(toSQL(query))) as DrizzleResult;\n retrievalResults.push(result);\n }\n });\n\n return retrievalResults;\n}\n\n/**\n * Execute the mutation phase of a Unit of Work using Drizzle\n *\n * All mutation queries are executed in a transaction with optimistic locking.\n * If any version check fails, the entire transaction is rolled back and\n * success=false is returned.\n *\n * @param db - The Drizzle database instance\n * @param mutationBatch - Array of compiled mutation SQL queries with expected affected rows\n * @returns Object with success flag and internal IDs from create operations\n *\n * @example\n * ```ts\n * const { success } = await executeDrizzleMutationPhase(db, compiled.mutationBatch);\n * if (!success) {\n * console.log(\"Version conflict detected, retrying...\");\n * }\n * ```\n */\nexport async function executeDrizzleMutationPhase(\n db: DBType,\n mutationBatch: CompiledMutation<DrizzleCompiledQuery>[],\n): Promise<MutationResult> {\n // If there are no mutations, return success immediately\n if (mutationBatch.length === 0) {\n return { success: true, createdInternalIds: [] };\n }\n\n const createdInternalIds: (bigint | null)[] = [];\n\n // Execute mutation batch in a transaction\n try {\n await db.transaction(async (tx) => {\n for (const compiledMutation of mutationBatch) {\n const result = await tx.execute(toSQL(compiledMutation.query));\n\n // For creates (expectedAffectedRows === null), try to extract internal ID\n if (compiledMutation.expectedAffectedRows === null) {\n // Check if result is an array with rows (RETURNING clause supported)\n if (Array.isArray(result) && result.length > 0) {\n const row = result[0] as Record<string, unknown>;\n // Look for _internalId column in the returned row\n if (\"_internalId\" in row || \"_internal_id\" in row) {\n const internalId = (row[\"_internalId\"] ?? row[\"_internal_id\"]) as bigint;\n createdInternalIds.push(internalId);\n } else {\n // RETURNING supported but _internalId not found\n createdInternalIds.push(null);\n }\n } else {\n // No RETURNING support (e.g., MySQL)\n createdInternalIds.push(null);\n }\n } else {\n // Check affected rows for updates/deletes\n const affectedRows = getAffectedRows(result);\n\n if (affectedRows !== compiledMutation.expectedAffectedRows) {\n // Version conflict detected - the UPDATE/DELETE didn't affect the expected number of rows\n // This means either the row doesn't exist or the version has changed\n throw new Error(\n `Version conflict: expected ${compiledMutation.expectedAffectedRows} rows affected, but got ${affectedRows}`,\n );\n }\n }\n }\n });\n\n return { success: true, createdInternalIds };\n } catch (error) {\n // Transaction failed - could be version conflict or other constraint violation\n // Return success=false to indicate the UOW should be retried\n if (error instanceof Error && error.message.includes(\"Version conflict\")) {\n return { success: false };\n }\n\n // Other database errors should be thrown\n throw error;\n }\n}\n"],"mappings":";;;;;;;;;;AAaA,SAAS,MAAM,OAAkC;CAC/C,MAAM,EAAE,KAAK,WAAW,WAAW;CAGnC,MAAM,mBAAmB;CACzB,MAAMA,cAA0B,EAAE;CAClC,IAAI,YAAY;CAChB,IAAIC;AAEJ,SAAQ,QAAQ,iBAAiB,KAAK,UAAU,MAAM,MAAM;EAE1D,MAAM,aAAa,UAAU,UAAU,WAAW,MAAM,MAAM;AAC9D,MAAI,WACF,aAAY,KAAK,IAAI,YAAY,WAAW,CAAC;EAI/C,MAAM,aAAa,SAAS,MAAM,IAAK,GAAG,GAAG;AAC7C,cAAY,KAAK,IAAI,MAAM,OAAO,YAAY,CAAC;AAE/C,cAAY,MAAM,QAAQ,MAAM,GAAG;;CAIrC,MAAM,YAAY,UAAU,UAAU,UAAU;AAChD,KAAI,UACF,aAAY,KAAK,IAAI,YAAY,UAAU,CAAC;AAK9C,QAAO,IAAI,IAAI,YAAY;;;;;AAM7B,SAAS,gBAAgB,QAAyB;AAIhD,KAAI,MAAM,QAAQ,OAAO,CAEvB,QAAO,OAAO;AAIhB,KACE,UACA,OAAO,WAAW,YAClB,kBAAkB,UAClB,OAAO,OAAO,oBAAoB,SAElC,QAAO,OAAO;AAIhB,KACE,UACA,OAAO,WAAW,YAClB,cAAc,WACb,OAAO,OAAO,gBAAgB,YAAY,OAAO,OAAO,gBAAgB,WACzE;EACA,MAAM,WAAW,OAAO;AACxB,MAAI,WAAW,OAAO,iBACpB,OAAM,IAAI,MAAM,yBAAyB,SAAS,UAAU,CAAC,gCAAgC;AAE/F,SAAO,OAAO,SAAS;;AAKzB,KAAI,UAAU,OAAO,WAAW,UAE9B;MAAI,aAAa,UAAU,OAAO,OAAO,eAAe,SACtD,QAAO,OAAO;;AAIlB,OAAM,IAAI,MAAM,kDAAkD,KAAK,UAAU,OAAO,GAAG;;;;;;;;;;;;;;;;;;AAmB7F,eAAsB,6BACpB,IACA,gBAC0B;AAE1B,KAAI,eAAe,WAAW,EAC5B,QAAO,EAAE;CAGX,MAAMC,mBAAoC,EAAE;AAG5C,OAAM,GAAG,YAAY,OAAO,OAAO;AACjC,OAAK,MAAM,SAAS,gBAAgB;GAClC,MAAM,SAAU,MAAM,GAAG,QAAQ,MAAM,MAAM,CAAC;AAC9C,oBAAiB,KAAK,OAAO;;GAE/B;AAEF,QAAO;;;;;;;;;;;;;;;;;;;;;AAsBT,eAAsB,4BACpB,IACA,eACyB;AAEzB,KAAI,cAAc,WAAW,EAC3B,QAAO;EAAE,SAAS;EAAM,oBAAoB,EAAE;EAAE;CAGlD,MAAMC,qBAAwC,EAAE;AAGhD,KAAI;AACF,QAAM,GAAG,YAAY,OAAO,OAAO;AACjC,QAAK,MAAM,oBAAoB,eAAe;IAC5C,MAAM,SAAS,MAAM,GAAG,QAAQ,MAAM,iBAAiB,MAAM,CAAC;AAG9D,QAAI,iBAAiB,yBAAyB,KAE5C,KAAI,MAAM,QAAQ,OAAO,IAAI,OAAO,SAAS,GAAG;KAC9C,MAAM,MAAM,OAAO;AAEnB,SAAI,iBAAiB,OAAO,kBAAkB,KAAK;MACjD,MAAM,aAAc,IAAI,kBAAkB,IAAI;AAC9C,yBAAmB,KAAK,WAAW;WAGnC,oBAAmB,KAAK,KAAK;UAI/B,oBAAmB,KAAK,KAAK;SAE1B;KAEL,MAAM,eAAe,gBAAgB,OAAO;AAE5C,SAAI,iBAAiB,iBAAiB,qBAGpC,OAAM,IAAI,MACR,8BAA8B,iBAAiB,qBAAqB,0BAA0B,eAC/F;;;IAIP;AAEF,SAAO;GAAE,SAAS;GAAM;GAAoB;UACrC,OAAO;AAGd,MAAI,iBAAiB,SAAS,MAAM,QAAQ,SAAS,mBAAmB,CACtE,QAAO,EAAE,SAAS,OAAO;AAI3B,QAAM"}
|