rowpipe 2.0.0 → 2.2.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/README.md +420 -59
- package/dist/analytics/profiler.d.ts +56 -0
- package/dist/analytics/profiler.d.ts.map +1 -0
- package/dist/analytics/profiler.js +277 -0
- package/dist/analytics/profiler.js.map +1 -0
- package/dist/analytics/schema-inference.d.ts.map +1 -1
- package/dist/analytics/schema-inference.js +14 -0
- package/dist/analytics/schema-inference.js.map +1 -1
- package/dist/cli/commands/clean.d.ts +15 -0
- package/dist/cli/commands/clean.d.ts.map +1 -0
- package/dist/cli/commands/clean.js +108 -0
- package/dist/cli/commands/clean.js.map +1 -0
- package/dist/cli/commands/convert.d.ts +6 -0
- package/dist/cli/commands/convert.d.ts.map +1 -1
- package/dist/cli/commands/convert.js +6 -6
- package/dist/cli/commands/convert.js.map +1 -1
- package/dist/cli/commands/count.d.ts +12 -0
- package/dist/cli/commands/count.d.ts.map +1 -0
- package/dist/cli/commands/count.js +49 -0
- package/dist/cli/commands/count.js.map +1 -0
- package/dist/cli/commands/db.d.ts +35 -0
- package/dist/cli/commands/db.d.ts.map +1 -0
- package/dist/cli/commands/db.js +173 -0
- package/dist/cli/commands/db.js.map +1 -0
- package/dist/cli/commands/explain.d.ts +10 -0
- package/dist/cli/commands/explain.d.ts.map +1 -0
- package/dist/cli/commands/explain.js +98 -0
- package/dist/cli/commands/explain.js.map +1 -0
- package/dist/cli/commands/files.d.ts +2 -1
- package/dist/cli/commands/files.d.ts.map +1 -1
- package/dist/cli/commands/files.js +5 -2
- package/dist/cli/commands/files.js.map +1 -1
- package/dist/cli/commands/group.d.ts +12 -0
- package/dist/cli/commands/group.d.ts.map +1 -0
- package/dist/cli/commands/group.js +51 -0
- package/dist/cli/commands/group.js.map +1 -0
- package/dist/cli/commands/head.d.ts +7 -0
- package/dist/cli/commands/head.d.ts.map +1 -0
- package/dist/cli/commands/head.js +6 -0
- package/dist/cli/commands/head.js.map +1 -0
- package/dist/cli/commands/join.d.ts +23 -0
- package/dist/cli/commands/join.d.ts.map +1 -0
- package/dist/cli/commands/join.js +97 -0
- package/dist/cli/commands/join.js.map +1 -0
- package/dist/cli/commands/limit.d.ts +11 -0
- package/dist/cli/commands/limit.d.ts.map +1 -0
- package/dist/cli/commands/limit.js +40 -0
- package/dist/cli/commands/limit.js.map +1 -0
- package/dist/cli/commands/offset.d.ts +11 -0
- package/dist/cli/commands/offset.d.ts.map +1 -0
- package/dist/cli/commands/offset.js +39 -0
- package/dist/cli/commands/offset.js.map +1 -0
- package/dist/cli/commands/pipeline.d.ts +63 -0
- package/dist/cli/commands/pipeline.d.ts.map +1 -0
- package/dist/cli/commands/pipeline.js +239 -0
- package/dist/cli/commands/pipeline.js.map +1 -0
- package/dist/cli/commands/profile.d.ts +22 -0
- package/dist/cli/commands/profile.d.ts.map +1 -0
- package/dist/cli/commands/profile.js +53 -0
- package/dist/cli/commands/profile.js.map +1 -0
- package/dist/cli/commands/sort.d.ts +17 -0
- package/dist/cli/commands/sort.d.ts.map +1 -0
- package/dist/cli/commands/sort.js +55 -0
- package/dist/cli/commands/sort.js.map +1 -0
- package/dist/cli/commands/tail.d.ts +13 -0
- package/dist/cli/commands/tail.d.ts.map +1 -0
- package/dist/cli/commands/tail.js +39 -0
- package/dist/cli/commands/tail.js.map +1 -0
- package/dist/cli/commands/top.d.ts +19 -0
- package/dist/cli/commands/top.d.ts.map +1 -0
- package/dist/cli/commands/top.js +51 -0
- package/dist/cli/commands/top.js.map +1 -0
- package/dist/cli/commands/unique.d.ts +15 -0
- package/dist/cli/commands/unique.d.ts.map +1 -0
- package/dist/cli/commands/unique.js +44 -0
- package/dist/cli/commands/unique.js.map +1 -0
- package/dist/cli/commands/view.d.ts +12 -0
- package/dist/cli/commands/view.d.ts.map +1 -0
- package/dist/cli/commands/view.js +52 -0
- package/dist/cli/commands/view.js.map +1 -0
- package/dist/cli/commands/window.d.ts +10 -0
- package/dist/cli/commands/window.d.ts.map +1 -0
- package/dist/cli/commands/window.js +87 -0
- package/dist/cli/commands/window.js.map +1 -0
- package/dist/cli/index.js +448 -3
- package/dist/cli/index.js.map +1 -1
- package/dist/core/batch.d.ts +1 -1
- package/dist/core/batch.d.ts.map +1 -1
- package/dist/core/batch.js.map +1 -1
- package/dist/core/pipeline.d.ts +42 -2
- package/dist/core/pipeline.d.ts.map +1 -1
- package/dist/core/pipeline.js +70 -2
- package/dist/core/pipeline.js.map +1 -1
- package/dist/core/types.d.ts +21 -1
- package/dist/core/types.d.ts.map +1 -1
- package/dist/db/adapters/index.d.ts +6 -0
- package/dist/db/adapters/index.d.ts.map +1 -0
- package/dist/db/adapters/index.js +21 -0
- package/dist/db/adapters/index.js.map +1 -0
- package/dist/db/adapters/mysql.d.ts +16 -0
- package/dist/db/adapters/mysql.d.ts.map +1 -0
- package/dist/db/adapters/mysql.js +230 -0
- package/dist/db/adapters/mysql.js.map +1 -0
- package/dist/db/adapters/postgres.d.ts +17 -0
- package/dist/db/adapters/postgres.d.ts.map +1 -0
- package/dist/db/adapters/postgres.js +239 -0
- package/dist/db/adapters/postgres.js.map +1 -0
- package/dist/db/adapters/sqlite.d.ts +18 -0
- package/dist/db/adapters/sqlite.d.ts.map +1 -0
- package/dist/db/adapters/sqlite.js +210 -0
- package/dist/db/adapters/sqlite.js.map +1 -0
- package/dist/db/index.d.ts +11 -0
- package/dist/db/index.d.ts.map +1 -0
- package/dist/db/index.js +11 -0
- package/dist/db/index.js.map +1 -0
- package/dist/db/mapping.d.ts +45 -0
- package/dist/db/mapping.d.ts.map +1 -0
- package/dist/db/mapping.js +222 -0
- package/dist/db/mapping.js.map +1 -0
- package/dist/db/pushdown.d.ts +25 -0
- package/dist/db/pushdown.d.ts.map +1 -0
- package/dist/db/pushdown.js +165 -0
- package/dist/db/pushdown.js.map +1 -0
- package/dist/db/sink.d.ts +15 -0
- package/dist/db/sink.d.ts.map +1 -0
- package/dist/db/sink.js +47 -0
- package/dist/db/sink.js.map +1 -0
- package/dist/db/source.d.ts +14 -0
- package/dist/db/source.d.ts.map +1 -0
- package/dist/db/source.js +69 -0
- package/dist/db/source.js.map +1 -0
- package/dist/db/types.d.ts +80 -0
- package/dist/db/types.d.ts.map +1 -0
- package/dist/db/types.js +2 -0
- package/dist/db/types.js.map +1 -0
- package/dist/db/url.d.ts +15 -0
- package/dist/db/url.d.ts.map +1 -0
- package/dist/db/url.js +145 -0
- package/dist/db/url.js.map +1 -0
- package/dist/diff/hash.d.ts.map +1 -1
- package/dist/diff/hash.js +4 -3
- package/dist/diff/hash.js.map +1 -1
- package/dist/diff/key.d.ts.map +1 -1
- package/dist/diff/key.js +8 -1
- package/dist/diff/key.js.map +1 -1
- package/dist/diff/reporter.d.ts.map +1 -1
- package/dist/diff/reporter.js +5 -3
- package/dist/diff/reporter.js.map +1 -1
- package/dist/diff/storage/disk-index.d.ts.map +1 -1
- package/dist/diff/storage/disk-index.js +2 -1
- package/dist/diff/storage/disk-index.js.map +1 -1
- package/dist/diff/storage/memory-index.d.ts.map +1 -1
- package/dist/diff/storage/memory-index.js +3 -2
- package/dist/diff/storage/memory-index.js.map +1 -1
- package/dist/index.d.ts +20 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +22 -1
- package/dist/index.js.map +1 -1
- package/dist/planner/executor.d.ts +8 -0
- package/dist/planner/executor.d.ts.map +1 -0
- package/dist/planner/executor.js +93 -0
- package/dist/planner/executor.js.map +1 -0
- package/dist/planner/explain.d.ts +6 -0
- package/dist/planner/explain.d.ts.map +1 -0
- package/dist/planner/explain.js +91 -0
- package/dist/planner/explain.js.map +1 -0
- package/dist/planner/index.d.ts +5 -0
- package/dist/planner/index.d.ts.map +1 -0
- package/dist/planner/index.js +5 -0
- package/dist/planner/index.js.map +1 -0
- package/dist/planner/optimizer.d.ts +9 -0
- package/dist/planner/optimizer.d.ts.map +1 -0
- package/dist/planner/optimizer.js +78 -0
- package/dist/planner/optimizer.js.map +1 -0
- package/dist/planner/types.d.ts +67 -0
- package/dist/planner/types.d.ts.map +1 -0
- package/dist/planner/types.js +2 -0
- package/dist/planner/types.js.map +1 -0
- package/dist/readers/csv.d.ts.map +1 -1
- package/dist/readers/csv.js +1 -2
- package/dist/readers/csv.js.map +1 -1
- package/dist/readers/index.d.ts +5 -1
- package/dist/readers/index.d.ts.map +1 -1
- package/dist/readers/index.js +43 -0
- package/dist/readers/index.js.map +1 -1
- package/dist/readers/json.d.ts.map +1 -1
- package/dist/readers/json.js +1 -2
- package/dist/readers/json.js.map +1 -1
- package/dist/readers/jsonl.d.ts.map +1 -1
- package/dist/readers/jsonl.js +1 -2
- package/dist/readers/jsonl.js.map +1 -1
- package/dist/readers/multi-file.d.ts +29 -0
- package/dist/readers/multi-file.d.ts.map +1 -0
- package/dist/readers/multi-file.js +169 -0
- package/dist/readers/multi-file.js.map +1 -0
- package/dist/transforms/clean.d.ts +18 -0
- package/dist/transforms/clean.d.ts.map +1 -0
- package/dist/transforms/clean.js +118 -0
- package/dist/transforms/clean.js.map +1 -0
- package/dist/transforms/count.d.ts +21 -0
- package/dist/transforms/count.d.ts.map +1 -0
- package/dist/transforms/count.js +65 -0
- package/dist/transforms/count.js.map +1 -0
- package/dist/transforms/expression.d.ts.map +1 -1
- package/dist/transforms/expression.js +12 -0
- package/dist/transforms/expression.js.map +1 -1
- package/dist/transforms/group.d.ts +25 -0
- package/dist/transforms/group.d.ts.map +1 -0
- package/dist/transforms/group.js +71 -0
- package/dist/transforms/group.js.map +1 -0
- package/dist/transforms/jit-compiler.d.ts +50 -0
- package/dist/transforms/jit-compiler.d.ts.map +1 -0
- package/dist/transforms/jit-compiler.js +275 -0
- package/dist/transforms/jit-compiler.js.map +1 -0
- package/dist/transforms/join/index-storage.d.ts +33 -0
- package/dist/transforms/join/index-storage.d.ts.map +1 -0
- package/dist/transforms/join/index-storage.js +139 -0
- package/dist/transforms/join/index-storage.js.map +1 -0
- package/dist/transforms/join/index.d.ts +4 -0
- package/dist/transforms/join/index.d.ts.map +1 -0
- package/dist/transforms/join/index.js +4 -0
- package/dist/transforms/join/index.js.map +1 -0
- package/dist/transforms/join/join.d.ts +15 -0
- package/dist/transforms/join/join.d.ts.map +1 -0
- package/dist/transforms/join/join.js +216 -0
- package/dist/transforms/join/join.js.map +1 -0
- package/dist/transforms/join/types.d.ts +29 -0
- package/dist/transforms/join/types.d.ts.map +1 -0
- package/dist/transforms/join/types.js +2 -0
- package/dist/transforms/join/types.js.map +1 -0
- package/dist/transforms/limit.d.ts +7 -0
- package/dist/transforms/limit.d.ts.map +1 -0
- package/dist/transforms/limit.js +40 -0
- package/dist/transforms/limit.js.map +1 -0
- package/dist/transforms/offset.d.ts +7 -0
- package/dist/transforms/offset.d.ts.map +1 -0
- package/dist/transforms/offset.js +38 -0
- package/dist/transforms/offset.js.map +1 -0
- package/dist/transforms/sort/comparator.d.ts +43 -0
- package/dist/transforms/sort/comparator.d.ts.map +1 -0
- package/dist/transforms/sort/comparator.js +131 -0
- package/dist/transforms/sort/comparator.js.map +1 -0
- package/dist/transforms/sort/external-sort.d.ts +8 -0
- package/dist/transforms/sort/external-sort.d.ts.map +1 -0
- package/dist/transforms/sort/external-sort.js +159 -0
- package/dist/transforms/sort/external-sort.js.map +1 -0
- package/dist/transforms/sort/index.d.ts +10 -0
- package/dist/transforms/sort/index.d.ts.map +1 -0
- package/dist/transforms/sort/index.js +11 -0
- package/dist/transforms/sort/index.js.map +1 -0
- package/dist/transforms/tail.d.ts +6 -0
- package/dist/transforms/tail.d.ts.map +1 -0
- package/dist/transforms/tail.js +27 -0
- package/dist/transforms/tail.js.map +1 -0
- package/dist/transforms/top.d.ts +17 -0
- package/dist/transforms/top.d.ts.map +1 -0
- package/dist/transforms/top.js +52 -0
- package/dist/transforms/top.js.map +1 -0
- package/dist/transforms/unique.d.ts +13 -0
- package/dist/transforms/unique.d.ts.map +1 -0
- package/dist/transforms/unique.js +91 -0
- package/dist/transforms/unique.js.map +1 -0
- package/dist/transforms/window.d.ts +24 -0
- package/dist/transforms/window.d.ts.map +1 -0
- package/dist/transforms/window.js +317 -0
- package/dist/transforms/window.js.map +1 -0
- package/dist/ui/colors.d.ts +57 -0
- package/dist/ui/colors.d.ts.map +1 -0
- package/dist/ui/colors.js +94 -0
- package/dist/ui/colors.js.map +1 -0
- package/dist/ui/viewer.d.ts +67 -0
- package/dist/ui/viewer.d.ts.map +1 -0
- package/dist/ui/viewer.js +673 -0
- package/dist/ui/viewer.js.map +1 -0
- package/dist/utils/compression.d.ts +68 -19
- package/dist/utils/compression.d.ts.map +1 -1
- package/dist/utils/compression.js +222 -34
- package/dist/utils/compression.js.map +1 -1
- package/dist/utils/formatting.d.ts +5 -0
- package/dist/utils/formatting.d.ts.map +1 -1
- package/dist/utils/formatting.js +15 -0
- package/dist/utils/formatting.js.map +1 -1
- package/dist/utils/heap.d.ts +25 -0
- package/dist/utils/heap.d.ts.map +1 -0
- package/dist/utils/heap.js +90 -0
- package/dist/utils/heap.js.map +1 -0
- package/dist/utils/keystore.d.ts +80 -0
- package/dist/utils/keystore.d.ts.map +1 -0
- package/dist/utils/keystore.js +217 -0
- package/dist/utils/keystore.js.map +1 -0
- package/dist/utils/ring-buffer.d.ts +15 -0
- package/dist/utils/ring-buffer.d.ts.map +1 -0
- package/dist/utils/ring-buffer.js +40 -0
- package/dist/utils/ring-buffer.js.map +1 -0
- package/dist/writers/csv.d.ts.map +1 -1
- package/dist/writers/csv.js +13 -14
- package/dist/writers/csv.js.map +1 -1
- package/dist/writers/index.d.ts +1 -0
- package/dist/writers/index.d.ts.map +1 -1
- package/dist/writers/index.js +35 -0
- package/dist/writers/index.js.map +1 -1
- package/dist/writers/json.d.ts.map +1 -1
- package/dist/writers/json.js +6 -5
- package/dist/writers/json.js.map +1 -1
- package/dist/writers/jsonl.d.ts.map +1 -1
- package/dist/writers/jsonl.js +6 -5
- package/dist/writers/jsonl.js.map +1 -1
- package/dist/writers/markdown.d.ts.map +1 -1
- package/dist/writers/markdown.js +5 -4
- package/dist/writers/markdown.js.map +1 -1
- package/dist/writers/parquet.d.ts +1 -1
- package/dist/writers/parquet.d.ts.map +1 -1
- package/package.json +11 -2
- package/skills/rowpipe/SKILL.md +16 -3
|
@@ -0,0 +1,239 @@
|
|
|
1
|
+
import pg from "pg";
|
|
2
|
+
import QueryStream from "pg-query-stream";
|
|
3
|
+
import { escapeIdentifier, escapeQualifiedTable, generateCreateTableSql, postgresTypeToRowpipe } from "../mapping.js";
|
|
4
|
+
export class PostgresDatabaseAdapter {
|
|
5
|
+
dialect = "postgres";
|
|
6
|
+
sanitizedUrl;
|
|
7
|
+
client;
|
|
8
|
+
isConnected = false;
|
|
9
|
+
constructor(config) {
|
|
10
|
+
this.sanitizedUrl = config.sanitizedUrl;
|
|
11
|
+
this.client = new pg.Client({
|
|
12
|
+
connectionString: config.url,
|
|
13
|
+
ssl: config.ssl,
|
|
14
|
+
});
|
|
15
|
+
}
|
|
16
|
+
async ensureConnected() {
|
|
17
|
+
if (!this.isConnected) {
|
|
18
|
+
await this.client.connect();
|
|
19
|
+
this.isConnected = true;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
async *stream(options) {
|
|
23
|
+
await this.ensureConnected();
|
|
24
|
+
const batchSize = Math.max(1, options.batchSize || 1000);
|
|
25
|
+
const sql = options.query || (options.table ? `SELECT * FROM ${escapeQualifiedTable(options.table, "postgres")}` : "");
|
|
26
|
+
if (!sql) {
|
|
27
|
+
throw new Error("Either table or query must be specified for PostgreSQL stream");
|
|
28
|
+
}
|
|
29
|
+
const params = options.params || [];
|
|
30
|
+
const queryStream = new QueryStream(sql, params, { batchSize });
|
|
31
|
+
const stream = this.client.query(queryStream);
|
|
32
|
+
let currentBatch = [];
|
|
33
|
+
let offset = 0;
|
|
34
|
+
const onAbort = () => {
|
|
35
|
+
stream.destroy();
|
|
36
|
+
};
|
|
37
|
+
if (options.signal) {
|
|
38
|
+
options.signal.addEventListener("abort", onAbort, { once: true });
|
|
39
|
+
}
|
|
40
|
+
try {
|
|
41
|
+
for await (const row of stream) {
|
|
42
|
+
if (options.signal?.aborted) {
|
|
43
|
+
break;
|
|
44
|
+
}
|
|
45
|
+
currentBatch.push(row);
|
|
46
|
+
if (currentBatch.length >= batchSize) {
|
|
47
|
+
yield {
|
|
48
|
+
rows: currentBatch,
|
|
49
|
+
offset,
|
|
50
|
+
};
|
|
51
|
+
offset += currentBatch.length;
|
|
52
|
+
currentBatch = [];
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
if (currentBatch.length > 0 && !options.signal?.aborted) {
|
|
56
|
+
yield {
|
|
57
|
+
rows: currentBatch,
|
|
58
|
+
offset,
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
finally {
|
|
63
|
+
if (options.signal) {
|
|
64
|
+
options.signal.removeEventListener("abort", onAbort);
|
|
65
|
+
}
|
|
66
|
+
stream.destroy();
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
async write(input, options) {
|
|
70
|
+
await this.ensureConnected();
|
|
71
|
+
const startTime = Date.now();
|
|
72
|
+
let rowsWritten = 0;
|
|
73
|
+
let batchesWritten = 0;
|
|
74
|
+
let tableCreated = false;
|
|
75
|
+
const table = options.table;
|
|
76
|
+
if (!table) {
|
|
77
|
+
throw new Error("Target table name is required for database write");
|
|
78
|
+
}
|
|
79
|
+
const escapedTable = escapeQualifiedTable(table, "postgres");
|
|
80
|
+
if (options.truncate) {
|
|
81
|
+
await this.client.query(`TRUNCATE TABLE ${escapedTable};`);
|
|
82
|
+
}
|
|
83
|
+
const useTx = options.transaction !== false;
|
|
84
|
+
let insertColumns = [];
|
|
85
|
+
for await (const batch of input) {
|
|
86
|
+
if (batch.rows.length === 0)
|
|
87
|
+
continue;
|
|
88
|
+
// Auto create table from first batch if requested
|
|
89
|
+
if (options.createTable && !tableCreated) {
|
|
90
|
+
const firstRow = batch.rows[0];
|
|
91
|
+
const inferredSchema = {};
|
|
92
|
+
for (const [key, val] of Object.entries(firstRow)) {
|
|
93
|
+
if (typeof val === "number") {
|
|
94
|
+
inferredSchema[key] = Number.isInteger(val) ? "integer" : "number";
|
|
95
|
+
}
|
|
96
|
+
else if (typeof val === "bigint") {
|
|
97
|
+
inferredSchema[key] = "bigint";
|
|
98
|
+
}
|
|
99
|
+
else if (typeof val === "boolean") {
|
|
100
|
+
inferredSchema[key] = "boolean";
|
|
101
|
+
}
|
|
102
|
+
else {
|
|
103
|
+
inferredSchema[key] = "string";
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
const ddl = generateCreateTableSql(table, inferredSchema, "postgres", options.conflictColumns);
|
|
107
|
+
if (options.dryRun) {
|
|
108
|
+
process.stdout.write(`\n-- Dry-run DDL:\n${ddl}\n`);
|
|
109
|
+
return { rowsWritten: 0, batchesWritten: 0, durationMs: Date.now() - startTime, tableCreated: true };
|
|
110
|
+
}
|
|
111
|
+
await this.client.query(ddl);
|
|
112
|
+
tableCreated = true;
|
|
113
|
+
}
|
|
114
|
+
if (insertColumns.length === 0) {
|
|
115
|
+
insertColumns = Object.keys(batch.rows[0]);
|
|
116
|
+
}
|
|
117
|
+
if (useTx) {
|
|
118
|
+
await this.client.query("BEGIN;");
|
|
119
|
+
}
|
|
120
|
+
try {
|
|
121
|
+
const escapedCols = insertColumns.map((c) => escapeIdentifier(c, "postgres")).join(", ");
|
|
122
|
+
const MAX_PARAMS = 50000;
|
|
123
|
+
const chunkSize = Math.max(1, Math.floor(MAX_PARAMS / insertColumns.length));
|
|
124
|
+
for (let i = 0; i < batch.rows.length; i += chunkSize) {
|
|
125
|
+
const rowChunk = batch.rows.slice(i, i + chunkSize);
|
|
126
|
+
const valuesPlaceholderRows = [];
|
|
127
|
+
const flatParams = [];
|
|
128
|
+
let paramIdx = 1;
|
|
129
|
+
for (const row of rowChunk) {
|
|
130
|
+
const placeholders = [];
|
|
131
|
+
for (const col of insertColumns) {
|
|
132
|
+
const v = row[col];
|
|
133
|
+
placeholders.push(`$${paramIdx++}`);
|
|
134
|
+
if (typeof v === "object" && v !== null && !(v instanceof Uint8Array) && !(v instanceof Date)) {
|
|
135
|
+
flatParams.push(JSON.stringify(v));
|
|
136
|
+
}
|
|
137
|
+
else {
|
|
138
|
+
flatParams.push(v === undefined ? null : v);
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
valuesPlaceholderRows.push(`(${placeholders.join(", ")})`);
|
|
142
|
+
rowsWritten++;
|
|
143
|
+
}
|
|
144
|
+
let sql = `INSERT INTO ${escapedTable} (${escapedCols}) VALUES ${valuesPlaceholderRows.join(", ")}`;
|
|
145
|
+
if (options.upsert && options.conflictColumns && options.conflictColumns.length > 0) {
|
|
146
|
+
const conflictCols = options.conflictColumns.map((c) => escapeIdentifier(c, "postgres")).join(", ");
|
|
147
|
+
const updateCols = insertColumns
|
|
148
|
+
.filter((c) => !options.conflictColumns.includes(c))
|
|
149
|
+
.map((c) => `${escapeIdentifier(c, "postgres")} = EXCLUDED.${escapeIdentifier(c, "postgres")}`)
|
|
150
|
+
.join(", ");
|
|
151
|
+
if (updateCols.length > 0) {
|
|
152
|
+
sql += ` ON CONFLICT (${conflictCols}) DO UPDATE SET ${updateCols}`;
|
|
153
|
+
}
|
|
154
|
+
else {
|
|
155
|
+
sql += ` ON CONFLICT (${conflictCols}) DO NOTHING`;
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
await this.client.query(sql, flatParams);
|
|
159
|
+
}
|
|
160
|
+
if (useTx) {
|
|
161
|
+
await this.client.query("COMMIT;");
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
catch (err) {
|
|
165
|
+
if (useTx) {
|
|
166
|
+
try {
|
|
167
|
+
await this.client.query("ROLLBACK;");
|
|
168
|
+
}
|
|
169
|
+
catch {
|
|
170
|
+
// Ignore rollback errors
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
throw err;
|
|
174
|
+
}
|
|
175
|
+
batchesWritten++;
|
|
176
|
+
}
|
|
177
|
+
return {
|
|
178
|
+
rowsWritten,
|
|
179
|
+
batchesWritten,
|
|
180
|
+
durationMs: Date.now() - startTime,
|
|
181
|
+
tableCreated,
|
|
182
|
+
};
|
|
183
|
+
}
|
|
184
|
+
async getTables() {
|
|
185
|
+
await this.ensureConnected();
|
|
186
|
+
const res = await this.client.query(`SELECT table_schema || '.' || table_name AS full_name
|
|
187
|
+
FROM information_schema.tables
|
|
188
|
+
WHERE table_schema NOT IN ('pg_catalog', 'information_schema')
|
|
189
|
+
AND table_type = 'BASE TABLE'
|
|
190
|
+
ORDER BY table_schema, table_name`);
|
|
191
|
+
return res.rows.map((r) => r.full_name);
|
|
192
|
+
}
|
|
193
|
+
async getTableSchema(table) {
|
|
194
|
+
await this.ensureConnected();
|
|
195
|
+
let schemaName = "public";
|
|
196
|
+
let tableName = table;
|
|
197
|
+
if (table.includes(".")) {
|
|
198
|
+
const parts = table.split(".");
|
|
199
|
+
schemaName = parts[0];
|
|
200
|
+
tableName = parts[1];
|
|
201
|
+
}
|
|
202
|
+
const res = await this.client.query(`SELECT column_name, udt_name, data_type, is_nullable, column_default
|
|
203
|
+
FROM information_schema.columns
|
|
204
|
+
WHERE table_schema = $1 AND table_name = $2
|
|
205
|
+
ORDER BY ordinal_position`, [schemaName, tableName]);
|
|
206
|
+
const pkRes = await this.client.query(`SELECT kcu.column_name
|
|
207
|
+
FROM information_schema.table_constraints tc
|
|
208
|
+
JOIN information_schema.key_column_usage kcu
|
|
209
|
+
ON tc.constraint_name = kcu.constraint_name
|
|
210
|
+
AND tc.table_schema = kcu.table_schema
|
|
211
|
+
WHERE tc.constraint_type = 'PRIMARY KEY'
|
|
212
|
+
AND tc.table_schema = $1 AND tc.table_name = $2`, [schemaName, tableName]);
|
|
213
|
+
const primaryKeys = pkRes.rows.map((r) => r.column_name);
|
|
214
|
+
const columns = res.rows.map((r) => ({
|
|
215
|
+
name: r.column_name,
|
|
216
|
+
dbType: r.udt_name || r.data_type,
|
|
217
|
+
rowpipeType: postgresTypeToRowpipe(r.udt_name || r.data_type),
|
|
218
|
+
nullable: r.is_nullable === "YES",
|
|
219
|
+
isPrimaryKey: primaryKeys.includes(r.column_name),
|
|
220
|
+
defaultValue: r.column_default,
|
|
221
|
+
}));
|
|
222
|
+
return {
|
|
223
|
+
table,
|
|
224
|
+
columns,
|
|
225
|
+
primaryKeys,
|
|
226
|
+
};
|
|
227
|
+
}
|
|
228
|
+
async executeRaw(sql, params) {
|
|
229
|
+
await this.ensureConnected();
|
|
230
|
+
return this.client.query(sql, params);
|
|
231
|
+
}
|
|
232
|
+
async close() {
|
|
233
|
+
if (this.isConnected) {
|
|
234
|
+
await this.client.end();
|
|
235
|
+
this.isConnected = false;
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
//# sourceMappingURL=postgres.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.js","sourceRoot":"","sources":["../../../src/db/adapters/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,WAAW,MAAM,iBAAiB,CAAC;AAE1C,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAYtH,MAAM,OAAO,uBAAuB;IACzB,OAAO,GAAoB,UAAU,CAAC;IACtC,YAAY,CAAS;IACtB,MAAM,CAAY;IAClB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,MAAgC;QAC1C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC;YAC1B,gBAAgB,EAAE,MAAM,CAAC,GAAG;YAC5B,GAAG,EAAE,MAAM,CAAC,GAAG;SAChB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAA4B;QACxC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,iBAAiB,oBAAoB,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACvH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QACnF,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAE9C,IAAI,YAAY,GAAU,EAAE,CAAC;QAC7B,IAAI,MAAM,GAAG,CAAC,CAAC;QAEf,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC;QAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBAC/B,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC5B,MAAM;gBACR,CAAC;gBAED,YAAY,CAAC,IAAI,CAAC,GAAU,CAAC,CAAC;gBAE9B,IAAI,YAAY,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;oBACrC,MAAM;wBACJ,IAAI,EAAE,YAAY;wBAClB,MAAM;qBACP,CAAC;oBACF,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC;oBAC9B,YAAY,GAAG,EAAE,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;gBACxD,MAAM;oBACJ,IAAI,EAAE,YAAY;oBAClB,MAAM;iBACP,CAAC;YACJ,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACvD,CAAC;YACD,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CACT,KAA+B,EAC/B,OAA6B;QAE7B,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,IAAI,YAAY,GAAG,KAAK,CAAC;QAEzB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAE7D,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,YAAY,GAAG,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,KAAK,KAAK,CAAC;QAC5C,IAAI,aAAa,GAAa,EAAE,CAAC;QAEjC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAChC,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAEtC,kDAAkD;YAClD,IAAI,OAAO,CAAC,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;gBACzC,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC;gBAChC,MAAM,cAAc,GAAwB,EAAE,CAAC;gBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAClD,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;wBAC5B,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;oBACrE,CAAC;yBAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;wBACnC,cAAc,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;oBACjC,CAAC;yBAAM,IAAI,OAAO,GAAG,KAAK,SAAS,EAAE,CAAC;wBACpC,cAAc,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;oBAClC,CAAC;yBAAM,CAAC;wBACN,cAAc,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;oBACjC,CAAC;gBACH,CAAC;gBACD,MAAM,GAAG,GAAG,sBAAsB,CAAC,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;gBAC/F,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,CAAC;oBACpD,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;gBACvG,CAAC;gBACD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC7B,YAAY,GAAG,IAAI,CAAC;YACtB,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/B,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC;YAC9C,CAAC;YAED,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpC,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACzF,MAAM,UAAU,GAAG,KAAK,CAAC;gBACzB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;gBAE7E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;oBACtD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;oBACpD,MAAM,qBAAqB,GAAa,EAAE,CAAC;oBAC3C,MAAM,UAAU,GAAc,EAAE,CAAC;oBACjC,IAAI,QAAQ,GAAG,CAAC,CAAC;oBAEjB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;wBAC3B,MAAM,YAAY,GAAa,EAAE,CAAC;wBAClC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;4BAChC,MAAM,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;4BACnB,YAAY,CAAC,IAAI,CAAC,IAAI,QAAQ,EAAE,EAAE,CAAC,CAAC;4BACpC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,YAAY,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,EAAE,CAAC;gCAC9F,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;4BACrC,CAAC;iCAAM,CAAC;gCACN,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;4BAC9C,CAAC;wBACH,CAAC;wBACD,qBAAqB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wBAC3D,WAAW,EAAE,CAAC;oBAChB,CAAC;oBAED,IAAI,GAAG,GAAG,eAAe,YAAY,KAAK,WAAW,YAAY,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;oBAEpG,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACpF,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACpG,MAAM,UAAU,GAAG,aAAa;6BAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,eAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;6BACpD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,UAAU,CAAC,eAAe,gBAAgB,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,CAAC;6BAC9F,IAAI,CAAC,IAAI,CAAC,CAAC;wBAEd,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC1B,GAAG,IAAI,iBAAiB,YAAY,mBAAmB,UAAU,EAAE,CAAC;wBACtE,CAAC;6BAAM,CAAC;4BACN,GAAG,IAAI,iBAAiB,YAAY,cAAc,CAAC;wBACrD,CAAC;oBACH,CAAC;oBAED,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;gBAC3C,CAAC;gBAED,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC;wBACH,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;oBACvC,CAAC;oBAAC,MAAM,CAAC;wBACP,yBAAyB;oBAC3B,CAAC;gBACH,CAAC;gBACD,MAAM,GAAG,CAAC;YACZ,CAAC;YAED,cAAc,EAAE,CAAC;QACnB,CAAC;QAED,OAAO;YACL,WAAW;YACX,cAAc;YACd,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,YAAY;SACb,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CACjC;;;;yCAImC,CACpC,CAAC;QACF,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,IAAI,UAAU,GAAG,QAAQ,CAAC;QAC1B,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC/B,UAAU,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;YACvB,SAAS,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;QACxB,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CACjC;;;iCAG2B,EAC3B,CAAC,UAAU,EAAE,SAAS,CAAC,CACxB,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CACnC;;;;;;yDAMmD,EACnD,CAAC,UAAU,EAAE,SAAS,CAAC,CACxB,CAAC;QAEF,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;QAE9D,MAAM,OAAO,GAA0B,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YAC/D,IAAI,EAAE,CAAC,CAAC,WAAW;YACnB,MAAM,EAAE,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS;YACjC,WAAW,EAAE,qBAAqB,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,CAAC;YAC7D,QAAQ,EAAE,CAAC,CAAC,WAAW,KAAK,KAAK;YACjC,YAAY,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC;YACjD,YAAY,EAAE,CAAC,CAAC,cAAc;SAC/B,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,KAAK;YACL,OAAO;YACP,WAAW;SACZ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,MAAkB;QAC9C,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YACxB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { DataBatch, Row } from "../../core/types.js";
|
|
2
|
+
import type { DatabaseAdapter, DatabaseConnectionConfig, DatabaseDialect, DatabaseReadOptions, DatabaseWriteOptions, DatabaseWriteResult, TableSchemaMetadata } from "../types.js";
|
|
3
|
+
export declare class SqliteDatabaseAdapter implements DatabaseAdapter {
|
|
4
|
+
readonly dialect: DatabaseDialect;
|
|
5
|
+
readonly config: DatabaseConnectionConfig;
|
|
6
|
+
readonly sanitizedUrl: string;
|
|
7
|
+
private db;
|
|
8
|
+
private isClosed;
|
|
9
|
+
constructor(config: DatabaseConnectionConfig);
|
|
10
|
+
stream(options: DatabaseReadOptions): AsyncIterable<DataBatch>;
|
|
11
|
+
write(input: AsyncIterable<DataBatch>, options: DatabaseWriteOptions): Promise<DatabaseWriteResult>;
|
|
12
|
+
insertBatch(table: string, rows: Row[], options?: DatabaseWriteOptions): Promise<DatabaseWriteResult>;
|
|
13
|
+
getTables(): Promise<string[]>;
|
|
14
|
+
getTableSchema(table: string): Promise<TableSchemaMetadata>;
|
|
15
|
+
executeRaw(sql: string, params?: unknown[]): Promise<unknown>;
|
|
16
|
+
close(): Promise<void>;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=sqlite.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.d.ts","sourceRoot":"","sources":["../../../src/db/adapters/sqlite.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAE1D,OAAO,KAAK,EACV,eAAe,EACf,wBAAwB,EACxB,eAAe,EACf,mBAAmB,EACnB,oBAAoB,EACpB,mBAAmB,EAEnB,mBAAmB,EACpB,MAAM,aAAa,CAAC;AAYrB,qBAAa,qBAAsB,YAAW,eAAe;IAC3D,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAY;IAC7C,QAAQ,CAAC,MAAM,EAAE,wBAAwB,CAAC;IAC1C,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,EAAE,CAAe;IACzB,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,wBAAwB;IAOrC,MAAM,CAAC,OAAO,EAAE,mBAAmB,GAAG,aAAa,CAAC,SAAS,CAAC;IA4C/D,KAAK,CACT,KAAK,EAAE,aAAa,CAAC,SAAS,CAAC,EAC/B,OAAO,EAAE,oBAAoB,GAC5B,OAAO,CAAC,mBAAmB,CAAC;IAyGzB,WAAW,CACf,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,GAAG,EAAE,EACX,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,mBAAmB,CAAC;IAOzB,SAAS,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAQ9B,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC;IA6B3D,UAAU,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ7D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAS7B"}
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
import { DatabaseSync } from "node:sqlite";
|
|
2
|
+
import { escapeIdentifier, escapeQualifiedTable, generateCreateTableSql, sqliteTypeToRowpipe } from "../mapping.js";
|
|
3
|
+
function toSqliteParam(val) {
|
|
4
|
+
if (val === null || val === undefined)
|
|
5
|
+
return null;
|
|
6
|
+
if (typeof val === "number" || typeof val === "bigint" || typeof val === "string")
|
|
7
|
+
return val;
|
|
8
|
+
if (typeof val === "boolean")
|
|
9
|
+
return val ? 1 : 0;
|
|
10
|
+
if (val instanceof Date)
|
|
11
|
+
return val.toISOString();
|
|
12
|
+
if (Buffer.isBuffer(val) || val instanceof Uint8Array)
|
|
13
|
+
return val;
|
|
14
|
+
if (typeof val === "object")
|
|
15
|
+
return JSON.stringify(val);
|
|
16
|
+
return String(val);
|
|
17
|
+
}
|
|
18
|
+
export class SqliteDatabaseAdapter {
|
|
19
|
+
dialect = "sqlite";
|
|
20
|
+
config;
|
|
21
|
+
sanitizedUrl;
|
|
22
|
+
db;
|
|
23
|
+
isClosed = false;
|
|
24
|
+
constructor(config) {
|
|
25
|
+
this.config = config;
|
|
26
|
+
this.sanitizedUrl = config.sanitizedUrl;
|
|
27
|
+
const dbPath = config.filePath || ":memory:";
|
|
28
|
+
this.db = new DatabaseSync(dbPath);
|
|
29
|
+
}
|
|
30
|
+
async *stream(options) {
|
|
31
|
+
const batchSize = Math.max(1, options.batchSize || 1000);
|
|
32
|
+
const sql = options.query || (options.table ? `SELECT * FROM ${escapeQualifiedTable(options.table, "sqlite")}` : "");
|
|
33
|
+
if (!sql) {
|
|
34
|
+
throw new Error("Either table or query must be specified for SQLite stream");
|
|
35
|
+
}
|
|
36
|
+
const params = (options.params || []).map(toSqliteParam);
|
|
37
|
+
const stmt = this.db.prepare(sql);
|
|
38
|
+
if (typeof stmt.setReadBigInts === "function") {
|
|
39
|
+
stmt.setReadBigInts(true);
|
|
40
|
+
}
|
|
41
|
+
let currentBatch = [];
|
|
42
|
+
let offset = 0;
|
|
43
|
+
try {
|
|
44
|
+
for (const rawRow of stmt.iterate(...params)) {
|
|
45
|
+
if (options.signal?.aborted) {
|
|
46
|
+
break;
|
|
47
|
+
}
|
|
48
|
+
currentBatch.push(rawRow);
|
|
49
|
+
if (currentBatch.length >= batchSize) {
|
|
50
|
+
yield {
|
|
51
|
+
rows: currentBatch,
|
|
52
|
+
offset,
|
|
53
|
+
};
|
|
54
|
+
offset += currentBatch.length;
|
|
55
|
+
currentBatch = [];
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
if (currentBatch.length > 0 && !options.signal?.aborted) {
|
|
59
|
+
yield {
|
|
60
|
+
rows: currentBatch,
|
|
61
|
+
offset,
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
finally {
|
|
66
|
+
// Statement iteration cleanup is automatic in node:sqlite
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
async write(input, options) {
|
|
70
|
+
const startTime = Date.now();
|
|
71
|
+
let rowsWritten = 0;
|
|
72
|
+
let batchesWritten = 0;
|
|
73
|
+
let tableCreated = false;
|
|
74
|
+
const table = options.table;
|
|
75
|
+
if (!table) {
|
|
76
|
+
throw new Error("Target table name is required for database write");
|
|
77
|
+
}
|
|
78
|
+
const escapedTable = escapeQualifiedTable(table, "sqlite");
|
|
79
|
+
if (options.truncate) {
|
|
80
|
+
this.db.exec(`DELETE FROM ${escapedTable};`);
|
|
81
|
+
}
|
|
82
|
+
const useTx = options.transaction !== false;
|
|
83
|
+
let insertStmt = null;
|
|
84
|
+
let insertColumns = [];
|
|
85
|
+
for await (const batch of input) {
|
|
86
|
+
if (batch.rows.length === 0)
|
|
87
|
+
continue;
|
|
88
|
+
// Auto create table from first batch if requested
|
|
89
|
+
if (options.createTable && !tableCreated) {
|
|
90
|
+
const firstRow = batch.rows[0];
|
|
91
|
+
const inferredSchema = {};
|
|
92
|
+
for (const [key, val] of Object.entries(firstRow)) {
|
|
93
|
+
if (typeof val === "number") {
|
|
94
|
+
inferredSchema[key] = Number.isInteger(val) ? "integer" : "number";
|
|
95
|
+
}
|
|
96
|
+
else if (typeof val === "bigint") {
|
|
97
|
+
inferredSchema[key] = "bigint";
|
|
98
|
+
}
|
|
99
|
+
else if (typeof val === "boolean") {
|
|
100
|
+
inferredSchema[key] = "boolean";
|
|
101
|
+
}
|
|
102
|
+
else {
|
|
103
|
+
inferredSchema[key] = "string";
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
const ddl = generateCreateTableSql(table, inferredSchema, "sqlite", options.conflictColumns);
|
|
107
|
+
if (options.dryRun) {
|
|
108
|
+
process.stdout.write(`\n-- Dry-run DDL:\n${ddl}\n`);
|
|
109
|
+
return { rowsWritten: 0, batchesWritten: 0, durationMs: Date.now() - startTime, tableCreated: true };
|
|
110
|
+
}
|
|
111
|
+
this.db.exec(ddl);
|
|
112
|
+
tableCreated = true;
|
|
113
|
+
}
|
|
114
|
+
if (!insertStmt) {
|
|
115
|
+
insertColumns = Object.keys(batch.rows[0]);
|
|
116
|
+
const escapedCols = insertColumns.map((c) => escapeIdentifier(c, "sqlite")).join(", ");
|
|
117
|
+
const placeholders = insertColumns.map(() => "?").join(", ");
|
|
118
|
+
let sql = `INSERT INTO ${escapedTable} (${escapedCols}) VALUES (${placeholders})`;
|
|
119
|
+
if (options.upsert && options.conflictColumns && options.conflictColumns.length > 0) {
|
|
120
|
+
const conflictCols = options.conflictColumns.map((c) => escapeIdentifier(c, "sqlite")).join(", ");
|
|
121
|
+
const updateCols = insertColumns
|
|
122
|
+
.filter((c) => !options.conflictColumns.includes(c))
|
|
123
|
+
.map((c) => `${escapeIdentifier(c, "sqlite")} = excluded.${escapeIdentifier(c, "sqlite")}`)
|
|
124
|
+
.join(", ");
|
|
125
|
+
if (updateCols.length > 0) {
|
|
126
|
+
sql += ` ON CONFLICT (${conflictCols}) DO UPDATE SET ${updateCols}`;
|
|
127
|
+
}
|
|
128
|
+
else {
|
|
129
|
+
sql += ` ON CONFLICT (${conflictCols}) DO NOTHING`;
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
insertStmt = this.db.prepare(sql);
|
|
133
|
+
}
|
|
134
|
+
if (useTx) {
|
|
135
|
+
this.db.exec("BEGIN TRANSACTION;");
|
|
136
|
+
}
|
|
137
|
+
try {
|
|
138
|
+
for (const row of batch.rows) {
|
|
139
|
+
const vals = insertColumns.map((c) => toSqliteParam(row[c]));
|
|
140
|
+
insertStmt.run(...vals);
|
|
141
|
+
rowsWritten++;
|
|
142
|
+
}
|
|
143
|
+
if (useTx) {
|
|
144
|
+
this.db.exec("COMMIT;");
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
catch (err) {
|
|
148
|
+
if (useTx) {
|
|
149
|
+
this.db.exec("ROLLBACK;");
|
|
150
|
+
}
|
|
151
|
+
throw err;
|
|
152
|
+
}
|
|
153
|
+
batchesWritten++;
|
|
154
|
+
}
|
|
155
|
+
return {
|
|
156
|
+
rowsWritten,
|
|
157
|
+
batchesWritten,
|
|
158
|
+
durationMs: Date.now() - startTime,
|
|
159
|
+
tableCreated,
|
|
160
|
+
};
|
|
161
|
+
}
|
|
162
|
+
async insertBatch(table, rows, options) {
|
|
163
|
+
async function* singleBatch() {
|
|
164
|
+
yield { rows, offset: 0 };
|
|
165
|
+
}
|
|
166
|
+
return this.write(singleBatch(), { table, ...options });
|
|
167
|
+
}
|
|
168
|
+
async getTables() {
|
|
169
|
+
const stmt = this.db.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' ORDER BY name ASC");
|
|
170
|
+
const rows = stmt.all();
|
|
171
|
+
return rows.map((r) => r.name);
|
|
172
|
+
}
|
|
173
|
+
async getTableSchema(table) {
|
|
174
|
+
const stmt = this.db.prepare(`PRAGMA table_info(${escapeIdentifier(table, "sqlite")})`);
|
|
175
|
+
const rows = stmt.all();
|
|
176
|
+
const columns = rows.map((r) => ({
|
|
177
|
+
name: r.name,
|
|
178
|
+
dbType: r.type || "TEXT",
|
|
179
|
+
rowpipeType: sqliteTypeToRowpipe(r.type),
|
|
180
|
+
nullable: r.notnull === 0,
|
|
181
|
+
isPrimaryKey: r.pk > 0,
|
|
182
|
+
defaultValue: r.dflt_value,
|
|
183
|
+
}));
|
|
184
|
+
const primaryKeys = rows.filter((r) => r.pk > 0).map((r) => r.name);
|
|
185
|
+
return {
|
|
186
|
+
table,
|
|
187
|
+
columns,
|
|
188
|
+
primaryKeys,
|
|
189
|
+
};
|
|
190
|
+
}
|
|
191
|
+
async executeRaw(sql, params) {
|
|
192
|
+
if (params && params.length > 0) {
|
|
193
|
+
const stmt = this.db.prepare(sql);
|
|
194
|
+
return stmt.all(...params.map(toSqliteParam));
|
|
195
|
+
}
|
|
196
|
+
return this.db.exec(sql);
|
|
197
|
+
}
|
|
198
|
+
async close() {
|
|
199
|
+
if (this.isClosed)
|
|
200
|
+
return;
|
|
201
|
+
this.isClosed = true;
|
|
202
|
+
try {
|
|
203
|
+
this.db.close();
|
|
204
|
+
}
|
|
205
|
+
catch {
|
|
206
|
+
// Ignore if already closed
|
|
207
|
+
}
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
//# sourceMappingURL=sqlite.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.js","sourceRoot":"","sources":["../../../src/db/adapters/sqlite.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAYpH,SAAS,aAAa,CAAC,GAAY;IACjC,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACnD,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC;IAC9F,IAAI,OAAO,GAAG,KAAK,SAAS;QAAE,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,IAAI,GAAG,YAAY,IAAI;QAAE,OAAO,GAAG,CAAC,WAAW,EAAE,CAAC;IAClD,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,YAAY,UAAU;QAAE,OAAO,GAAG,CAAC;IAClE,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACxD,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,OAAO,qBAAqB;IACvB,OAAO,GAAoB,QAAQ,CAAC;IACpC,MAAM,CAA2B;IACjC,YAAY,CAAS;IACtB,EAAE,CAAe;IACjB,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAAY,MAAgC;QAC1C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,IAAI,UAAU,CAAC;QAC7C,IAAI,CAAC,EAAE,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAA4B;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,iBAAiB,oBAAoB,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,OAAQ,IAAY,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;YACtD,IAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,YAAY,GAAU,EAAE,CAAC;QAC7B,IAAI,MAAM,GAAG,CAAC,CAAC;QAEf,IAAI,CAAC;YACH,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC,EAAE,CAAC;gBAC7C,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC5B,MAAM;gBACR,CAAC;gBAED,YAAY,CAAC,IAAI,CAAC,MAAa,CAAC,CAAC;gBAEjC,IAAI,YAAY,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;oBACrC,MAAM;wBACJ,IAAI,EAAE,YAAY;wBAClB,MAAM;qBACP,CAAC;oBACF,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC;oBAC9B,YAAY,GAAG,EAAE,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;gBACxD,MAAM;oBACJ,IAAI,EAAE,YAAY;oBAClB,MAAM;iBACP,CAAC;YACJ,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,0DAA0D;QAC5D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CACT,KAA+B,EAC/B,OAA6B;QAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,IAAI,YAAY,GAAG,KAAK,CAAC;QAEzB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAE3D,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,YAAY,GAAG,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,KAAK,KAAK,CAAC;QAE5C,IAAI,UAAU,GAAQ,IAAI,CAAC;QAC3B,IAAI,aAAa,GAAa,EAAE,CAAC;QAEjC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAChC,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAEtC,kDAAkD;YAClD,IAAI,OAAO,CAAC,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;gBACzC,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC;gBAChC,MAAM,cAAc,GAAwB,EAAE,CAAC;gBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAClD,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;wBAC5B,cAAc,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;oBACrE,CAAC;yBAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;wBACnC,cAAc,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;oBACjC,CAAC;yBAAM,IAAI,OAAO,GAAG,KAAK,SAAS,EAAE,CAAC;wBACpC,cAAc,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;oBAClC,CAAC;yBAAM,CAAC;wBACN,cAAc,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;oBACjC,CAAC;gBACH,CAAC;gBACD,MAAM,GAAG,GAAG,sBAAsB,CAAC,KAAK,EAAE,cAAc,EAAE,QAAQ,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;gBAC7F,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,CAAC;oBACpD,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;gBACvG,CAAC;gBACD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,YAAY,GAAG,IAAI,CAAC;YACtB,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC;gBAC5C,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvF,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE7D,IAAI,GAAG,GAAG,eAAe,YAAY,KAAK,WAAW,aAAa,YAAY,GAAG,CAAC;gBAElF,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpF,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAClG,MAAM,UAAU,GAAG,aAAa;yBAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,eAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;yBACpD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,QAAQ,CAAC,eAAe,gBAAgB,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,CAAC;yBAC1F,IAAI,CAAC,IAAI,CAAC,CAAC;oBAEd,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC1B,GAAG,IAAI,iBAAiB,YAAY,mBAAmB,UAAU,EAAE,CAAC;oBACtE,CAAC;yBAAM,CAAC;wBACN,GAAG,IAAI,iBAAiB,YAAY,cAAc,CAAC;oBACrD,CAAC;gBACH,CAAC;gBAED,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACpC,CAAC;YAED,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrC,CAAC;YAED,IAAI,CAAC;gBACH,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC7B,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7D,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;oBACxB,WAAW,EAAE,CAAC;gBAChB,CAAC;gBAED,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC5B,CAAC;gBACD,MAAM,GAAG,CAAC;YACZ,CAAC;YAED,cAAc,EAAE,CAAC;QACnB,CAAC;QAED,OAAO;YACL,WAAW;YACX,cAAc;YACd,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,YAAY;SACb,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,KAAa,EACb,IAAW,EACX,OAA8B;QAE9B,KAAK,SAAS,CAAC,CAAC,WAAW;YACzB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,kGAAkG,CACnG,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAA6B,CAAC;QACnD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,qBAAqB,gBAAgB,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC;QACxF,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAOnB,CAAC;QAEH,MAAM,OAAO,GAA0B,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,MAAM,EAAE,CAAC,CAAC,IAAI,IAAI,MAAM;YACxB,WAAW,EAAE,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC;YACxC,QAAQ,EAAE,CAAC,CAAC,OAAO,KAAK,CAAC;YACzB,YAAY,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC;YACtB,YAAY,EAAE,CAAC,CAAC,UAAU;SAC3B,CAAC,CAAC,CAAC;QAEJ,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEpE,OAAO;YACL,KAAK;YACL,OAAO;YACP,WAAW;SACZ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,MAAkB;QAC9C,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC;YACP,2BAA2B;QAC7B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
export * from "./types.js";
|
|
2
|
+
export * from "./url.js";
|
|
3
|
+
export * from "./mapping.js";
|
|
4
|
+
export * from "./pushdown.js";
|
|
5
|
+
export * from "./adapters/index.js";
|
|
6
|
+
export * from "./adapters/sqlite.js";
|
|
7
|
+
export * from "./adapters/postgres.js";
|
|
8
|
+
export * from "./adapters/mysql.js";
|
|
9
|
+
export * from "./source.js";
|
|
10
|
+
export * from "./sink.js";
|
|
11
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,UAAU,CAAC;AACzB,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC"}
|
package/dist/db/index.js
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
export * from "./types.js";
|
|
2
|
+
export * from "./url.js";
|
|
3
|
+
export * from "./mapping.js";
|
|
4
|
+
export * from "./pushdown.js";
|
|
5
|
+
export * from "./adapters/index.js";
|
|
6
|
+
export * from "./adapters/sqlite.js";
|
|
7
|
+
export * from "./adapters/postgres.js";
|
|
8
|
+
export * from "./adapters/mysql.js";
|
|
9
|
+
export * from "./source.js";
|
|
10
|
+
export * from "./sink.js";
|
|
11
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,UAAU,CAAC;AACzB,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC"}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
import type { ColumnType } from "../core/types.js";
|
|
2
|
+
import type { DatabaseDialect } from "./types.js";
|
|
3
|
+
/**
|
|
4
|
+
* Escapes a single SQL identifier (table/column name) safely according to dialect.
|
|
5
|
+
*/
|
|
6
|
+
export declare function escapeIdentifier(name: string, dialect: DatabaseDialect): string;
|
|
7
|
+
/**
|
|
8
|
+
* Escapes a qualified table name (e.g. "public.users" -> `"public"."users"`).
|
|
9
|
+
*/
|
|
10
|
+
export declare function escapeQualifiedTable(table: string, dialect: DatabaseDialect): string;
|
|
11
|
+
/**
|
|
12
|
+
* Maps PostgreSQL column types to Rowpipe logical ColumnType.
|
|
13
|
+
*/
|
|
14
|
+
export declare function postgresTypeToRowpipe(dbType: string): ColumnType;
|
|
15
|
+
/**
|
|
16
|
+
* Maps MySQL column types to Rowpipe logical ColumnType.
|
|
17
|
+
*/
|
|
18
|
+
export declare function mysqlTypeToRowpipe(dbType: string): ColumnType;
|
|
19
|
+
/**
|
|
20
|
+
* Maps SQLite column types to Rowpipe logical ColumnType.
|
|
21
|
+
*/
|
|
22
|
+
export declare function sqliteTypeToRowpipe(dbType: string): ColumnType;
|
|
23
|
+
/**
|
|
24
|
+
* Maps Rowpipe logical ColumnType to PostgreSQL DDL data type.
|
|
25
|
+
*/
|
|
26
|
+
export declare function rowpipeTypeToPostgres(type: ColumnType): string;
|
|
27
|
+
/**
|
|
28
|
+
* Maps Rowpipe logical ColumnType to MySQL DDL data type.
|
|
29
|
+
*/
|
|
30
|
+
export declare function rowpipeTypeToMysql(type: ColumnType): string;
|
|
31
|
+
/**
|
|
32
|
+
* Maps Rowpipe logical ColumnType to SQLite DDL data type.
|
|
33
|
+
*/
|
|
34
|
+
export declare function rowpipeTypeToSqlite(type: ColumnType): string;
|
|
35
|
+
/**
|
|
36
|
+
* Generates dialect-specific CREATE TABLE SQL DDL.
|
|
37
|
+
*/
|
|
38
|
+
export declare function generateCreateTableSql(table: string, schema: Record<string, ColumnType> | {
|
|
39
|
+
columns: Array<{
|
|
40
|
+
name: string;
|
|
41
|
+
type: ColumnType;
|
|
42
|
+
nullable?: boolean;
|
|
43
|
+
}>;
|
|
44
|
+
}, dialect: DatabaseDialect, primaryKeys?: string[]): string;
|
|
45
|
+
//# sourceMappingURL=mapping.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mapping.d.ts","sourceRoot":"","sources":["../../src/db/mapping.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAElD;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,GAAG,MAAM,CAO/E;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,GAAG,MAAM,CAGpF;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,CA8BhE;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,CA8B7D;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,CAyB9D;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,CAuB9D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,CAuB3D;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,CAe5D;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAA;CAAE,EAC/G,OAAO,EAAE,eAAe,EACxB,WAAW,CAAC,EAAE,MAAM,EAAE,GACrB,MAAM,CAgCR"}
|