@quillsql/node 0.7.6 → 0.7.8
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/dist/cjs/db/CachedConnection.d.ts +3 -8
- package/dist/cjs/db/CachedConnection.js.map +1 -1
- package/dist/cjs/db/ClickHouse.js +124 -124
- package/dist/cjs/db/ClickHouse.js.map +1 -1
- package/dist/cjs/db/DatabaseHelper.d.ts +16 -12
- package/dist/cjs/db/DatabaseHelper.js +20 -0
- package/dist/cjs/db/DatabaseHelper.js.map +1 -1
- package/dist/cjs/db/Databricks.d.ts +29 -0
- package/dist/cjs/db/Databricks.js +210 -0
- package/dist/cjs/db/Databricks.js.map +1 -0
- package/dist/cjs/db/Mssql.d.ts +2 -2
- package/dist/cjs/db/Mssql.js +79 -44
- package/dist/cjs/db/Mssql.js.map +1 -1
- package/dist/cjs/db/Postgres.js +5 -1
- package/dist/cjs/db/Postgres.js.map +1 -1
- package/dist/cjs/index.d.ts +5 -3
- package/dist/cjs/index.js +26 -11
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/models/Filters.js +57 -49
- package/dist/cjs/models/Filters.js.map +1 -1
- package/dist/cjs/utils/tenants.js +6 -2
- package/dist/cjs/utils/tenants.js.map +1 -1
- package/dist/esm/db/CachedConnection.d.ts +3 -8
- package/dist/esm/db/CachedConnection.js.map +1 -1
- package/dist/esm/db/ClickHouse.js +124 -124
- package/dist/esm/db/ClickHouse.js.map +1 -1
- package/dist/esm/db/DatabaseHelper.d.ts +16 -12
- package/dist/esm/db/DatabaseHelper.js +22 -2
- package/dist/esm/db/DatabaseHelper.js.map +1 -1
- package/dist/esm/db/Databricks.d.ts +29 -0
- package/dist/esm/db/Databricks.js +199 -0
- package/dist/esm/db/Databricks.js.map +1 -0
- package/dist/esm/db/Mssql.d.ts +2 -2
- package/dist/esm/db/Mssql.js +80 -45
- package/dist/esm/db/Mssql.js.map +1 -1
- package/dist/esm/db/Postgres.js +5 -1
- package/dist/esm/db/Postgres.js.map +1 -1
- package/dist/esm/index.d.ts +5 -3
- package/dist/esm/index.js +26 -11
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/models/Filters.js +57 -49
- package/dist/esm/models/Filters.js.map +1 -1
- package/dist/esm/utils/tenants.js +6 -2
- package/dist/esm/utils/tenants.js.map +1 -1
- package/package.json +3 -1
|
@@ -12,170 +12,170 @@ import { createClient } from "@clickhouse/client";
|
|
|
12
12
|
import { capitalize, depluralize } from "../utils/textProcessing";
|
|
13
13
|
const CLICKHOUSE_PG_TYPE_MAP = {
|
|
14
14
|
// Signed Integer Types and Aliases
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
15
|
+
Int8: 21,
|
|
16
|
+
TINYINT: 21,
|
|
17
|
+
INT1: 21,
|
|
18
|
+
BYTE: 21,
|
|
19
|
+
"TINYINT SIGNED": 21,
|
|
20
|
+
"INT1 SIGNED": 21,
|
|
21
|
+
Int16: 21,
|
|
22
|
+
SMALLINT: 21,
|
|
23
|
+
"SMALLINT SIGNED": 21,
|
|
24
|
+
Int32: 23,
|
|
25
|
+
INT: 23,
|
|
26
|
+
INTEGER: 23,
|
|
27
|
+
MEDIUMINT: 23,
|
|
28
|
+
"MEDIUMINT SIGNED": 23,
|
|
29
|
+
"INT SIGNED": 23,
|
|
30
|
+
"INTEGER SIGNED": 23,
|
|
31
|
+
Int64: 20,
|
|
32
|
+
BIGINT: 20,
|
|
33
|
+
SIGNED: 20,
|
|
34
|
+
"BIGINT SIGNED": 20,
|
|
35
|
+
TIME: 20,
|
|
36
36
|
// Unsigned Integer Types and Aliases
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
37
|
+
UInt8: 21,
|
|
38
|
+
"TINYINT UNSIGNED": 21,
|
|
39
|
+
"INT1 UNSIGNED": 21,
|
|
40
|
+
UInt16: 21,
|
|
41
|
+
"SMALLINT UNSIGNED": 21,
|
|
42
|
+
UInt32: 23,
|
|
43
|
+
"MEDIUMINT UNSIGNED": 23,
|
|
44
|
+
"INT UNSIGNED": 23,
|
|
45
|
+
"INTEGER UNSIGNED": 23,
|
|
46
|
+
UInt64: 20,
|
|
47
|
+
UNSIGNED: 20,
|
|
48
|
+
"BIGINT UNSIGNED": 20,
|
|
49
|
+
BIT: 20,
|
|
50
|
+
SET: 20,
|
|
51
51
|
// Floating Point Types and Aliases
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
52
|
+
Float32: 700,
|
|
53
|
+
FLOAT: 700,
|
|
54
|
+
REAL: 700,
|
|
55
|
+
SINGLE: 700,
|
|
56
|
+
Float64: 701,
|
|
57
|
+
DOUBLE: 701,
|
|
58
|
+
"DOUBLE PRECISION": 701,
|
|
59
|
+
BFloat16: 700, // Mapped to REAL as closest approximation
|
|
60
60
|
// Decimal Types
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
61
|
+
Decimal: 1700,
|
|
62
|
+
Decimal32: 1700,
|
|
63
|
+
Decimal64: 1700,
|
|
64
|
+
Decimal128: 1700,
|
|
65
|
+
Decimal256: 1700,
|
|
66
66
|
// Boolean Type
|
|
67
|
-
|
|
67
|
+
Bool: 16,
|
|
68
68
|
// String Types and Aliases
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
69
|
+
String: 25,
|
|
70
|
+
LONGTEXT: 25,
|
|
71
|
+
MEDIUMTEXT: 25,
|
|
72
|
+
TINYTEXT: 25,
|
|
73
|
+
TEXT: 25,
|
|
74
|
+
LONGBLOB: 17,
|
|
75
|
+
MEDIUMBLOB: 17,
|
|
76
|
+
TINYBLOB: 17,
|
|
77
|
+
BLOB: 17,
|
|
78
|
+
VARCHAR: 1043,
|
|
79
|
+
CHAR: 1042,
|
|
80
|
+
"CHAR LARGE OBJECT": 25,
|
|
81
|
+
"CHAR VARYING": 1043,
|
|
82
|
+
"CHARACTER LARGE OBJECT": 25,
|
|
83
|
+
"CHARACTER VARYING": 1043,
|
|
84
|
+
"NCHAR LARGE OBJECT": 25,
|
|
85
|
+
"NCHAR VARYING": 1043,
|
|
86
|
+
"NATIONAL CHARACTER LARGE OBJECT": 25,
|
|
87
|
+
"NATIONAL CHARACTER VARYING": 1043,
|
|
88
|
+
"NATIONAL CHAR VARYING": 1043,
|
|
89
|
+
"NATIONAL CHARACTER": 1042,
|
|
90
|
+
"NATIONAL CHAR": 1042,
|
|
91
|
+
"BINARY LARGE OBJECT": 17,
|
|
92
|
+
"BINARY VARYING": 17,
|
|
93
93
|
// Fixed String
|
|
94
|
-
|
|
94
|
+
FixedString: 1042,
|
|
95
95
|
// Identifier Types
|
|
96
|
-
|
|
96
|
+
UUID: 2950,
|
|
97
97
|
// Date and Time Types
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
98
|
+
Date: 1082,
|
|
99
|
+
Date32: 1082,
|
|
100
|
+
DateTime: 1184,
|
|
101
|
+
DateTime64: 1184,
|
|
102
102
|
// Array Types
|
|
103
|
-
|
|
103
|
+
Array: 2277,
|
|
104
104
|
// JSON-like Types
|
|
105
|
-
|
|
106
|
-
|
|
105
|
+
JSON: 3802,
|
|
106
|
+
Nested: 3802,
|
|
107
107
|
// Binary Types
|
|
108
|
-
|
|
109
|
-
|
|
108
|
+
IPv4: 17,
|
|
109
|
+
IPv6: 17,
|
|
110
110
|
// Enum Types
|
|
111
|
-
|
|
112
|
-
|
|
111
|
+
Enum8: 10045, // pg_enum
|
|
112
|
+
Enum16: 10045, // pg_enum
|
|
113
113
|
// Geospatial-like Types
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
114
|
+
Point: 17,
|
|
115
|
+
Ring: 17,
|
|
116
|
+
Polygon: 17,
|
|
117
117
|
// Specialized Types
|
|
118
|
-
|
|
119
|
-
|
|
118
|
+
Nothing: 17,
|
|
119
|
+
Interval: 1186,
|
|
120
120
|
};
|
|
121
121
|
function parseClickHouseType(type) {
|
|
122
122
|
// Remove whitespace and handle common variations
|
|
123
|
-
const normalizedType = type.trim().replace(/\s+/g,
|
|
123
|
+
const normalizedType = type.trim().replace(/\s+/g, " ");
|
|
124
124
|
// Handle Object types
|
|
125
|
-
if (normalizedType.startsWith(
|
|
126
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
125
|
+
if (normalizedType.startsWith("Map(")) {
|
|
126
|
+
return CLICKHOUSE_PG_TYPE_MAP["JSON"];
|
|
127
127
|
}
|
|
128
|
-
if (normalizedType.startsWith(
|
|
129
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
128
|
+
if (normalizedType.startsWith("AggregateFunction(")) {
|
|
129
|
+
return CLICKHOUSE_PG_TYPE_MAP["JSON"];
|
|
130
130
|
}
|
|
131
|
-
if (normalizedType.startsWith(
|
|
132
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
131
|
+
if (normalizedType.startsWith("SimpleAggregateFunction(")) {
|
|
132
|
+
return CLICKHOUSE_PG_TYPE_MAP["JSON"];
|
|
133
133
|
}
|
|
134
134
|
// Handle Nullable types
|
|
135
|
-
if (normalizedType.startsWith(
|
|
135
|
+
if (normalizedType.startsWith("Nullable(")) {
|
|
136
136
|
const innerType = normalizedType.slice(9, -1);
|
|
137
137
|
return parseClickHouseType(innerType);
|
|
138
138
|
}
|
|
139
139
|
// Handle Array types
|
|
140
|
-
if (normalizedType.startsWith(
|
|
141
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
140
|
+
if (normalizedType.startsWith("Array(")) {
|
|
141
|
+
return CLICKHOUSE_PG_TYPE_MAP["Array"];
|
|
142
142
|
}
|
|
143
|
-
if (normalizedType.startsWith(
|
|
144
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
143
|
+
if (normalizedType.startsWith("Tuple(")) {
|
|
144
|
+
return CLICKHOUSE_PG_TYPE_MAP["Array"]; // Adjust based on your requirements
|
|
145
145
|
}
|
|
146
146
|
// Handle Enum types
|
|
147
|
-
if (normalizedType.startsWith(
|
|
148
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
147
|
+
if (normalizedType.startsWith("Enum8(")) {
|
|
148
|
+
return CLICKHOUSE_PG_TYPE_MAP["Enum8"];
|
|
149
149
|
}
|
|
150
|
-
if (normalizedType.startsWith(
|
|
151
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
150
|
+
if (normalizedType.startsWith("Enum16(")) {
|
|
151
|
+
return CLICKHOUSE_PG_TYPE_MAP["Enum16"];
|
|
152
152
|
}
|
|
153
153
|
// Handle Decimal types
|
|
154
|
-
if (normalizedType.startsWith(
|
|
155
|
-
normalizedType.startsWith(
|
|
156
|
-
normalizedType.startsWith(
|
|
157
|
-
normalizedType.startsWith(
|
|
158
|
-
normalizedType.startsWith(
|
|
159
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
154
|
+
if (normalizedType.startsWith("Decimal(") ||
|
|
155
|
+
normalizedType.startsWith("Decimal32(") ||
|
|
156
|
+
normalizedType.startsWith("Decimal64(") ||
|
|
157
|
+
normalizedType.startsWith("Decimal128(") ||
|
|
158
|
+
normalizedType.startsWith("Decimal256(")) {
|
|
159
|
+
return CLICKHOUSE_PG_TYPE_MAP["Decimal"];
|
|
160
160
|
}
|
|
161
161
|
// Handle DateTime with timezone
|
|
162
|
-
if (normalizedType.startsWith(
|
|
163
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
162
|
+
if (normalizedType.startsWith("DateTime(")) {
|
|
163
|
+
return CLICKHOUSE_PG_TYPE_MAP["DateTime"];
|
|
164
164
|
}
|
|
165
|
-
if (normalizedType.startsWith(
|
|
166
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
165
|
+
if (normalizedType.startsWith("DateTime64(")) {
|
|
166
|
+
return CLICKHOUSE_PG_TYPE_MAP["DateTime64"];
|
|
167
167
|
}
|
|
168
168
|
// Handle FixedString
|
|
169
|
-
if (normalizedType.startsWith(
|
|
170
|
-
return CLICKHOUSE_PG_TYPE_MAP[
|
|
169
|
+
if (normalizedType.startsWith("FixedString(")) {
|
|
170
|
+
return CLICKHOUSE_PG_TYPE_MAP["FixedString"];
|
|
171
171
|
}
|
|
172
172
|
// Handle LowCardinality
|
|
173
|
-
if (normalizedType.startsWith(
|
|
173
|
+
if (normalizedType.startsWith("LowCardinality(")) {
|
|
174
174
|
const innerType = normalizedType.slice(15, -1);
|
|
175
175
|
return parseClickHouseType(innerType);
|
|
176
176
|
}
|
|
177
177
|
// Direct lookup (case-insensitive)
|
|
178
|
-
const lookupType = Object.keys(CLICKHOUSE_PG_TYPE_MAP).find(key => key.toLowerCase() === normalizedType.toLowerCase());
|
|
178
|
+
const lookupType = Object.keys(CLICKHOUSE_PG_TYPE_MAP).find((key) => key.toLowerCase() === normalizedType.toLowerCase());
|
|
179
179
|
if (lookupType) {
|
|
180
180
|
return CLICKHOUSE_PG_TYPE_MAP[lookupType];
|
|
181
181
|
}
|
|
@@ -226,7 +226,7 @@ export function getTablesBySchemaClickHouse(client, schemaNames) {
|
|
|
226
226
|
const results = yield runQueryClickHouse(sql, client);
|
|
227
227
|
return results.rows.map((row) => ({
|
|
228
228
|
tableName: row.table_name,
|
|
229
|
-
schemaName: row.table_schema
|
|
229
|
+
schemaName: row.table_schema,
|
|
230
230
|
}));
|
|
231
231
|
})));
|
|
232
232
|
return allTables.flat();
|
|
@@ -307,8 +307,8 @@ export function formatClickHouseConfig(connectionString) {
|
|
|
307
307
|
const parsed = new URL(connectionString);
|
|
308
308
|
return {
|
|
309
309
|
url: `${parsed.protocol}//${parsed.hostname}:${parsed.port}`, // Base URL without credentials
|
|
310
|
-
username: parsed.username ||
|
|
311
|
-
password: parsed.password ||
|
|
310
|
+
username: parsed.username || "default", // Default to 'default' if not provided
|
|
311
|
+
password: parsed.password || "", // Default to empty password if not provided
|
|
312
312
|
};
|
|
313
313
|
}
|
|
314
314
|
//# sourceMappingURL=ClickHouse.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ClickHouse.js","sourceRoot":"","sources":["../../../src/db/ClickHouse.ts"],"names":[],"mappings":";;;;;;;;;AAAA,oCAAoC;AACpC,OAAO,EAAoB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGpE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAElE,MAAM,sBAAsB,GAA+B;
|
|
1
|
+
{"version":3,"file":"ClickHouse.js","sourceRoot":"","sources":["../../../src/db/ClickHouse.ts"],"names":[],"mappings":";;;;;;;;;AAAA,oCAAoC;AACpC,OAAO,EAAoB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGpE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAElE,MAAM,sBAAsB,GAA+B;IACzD,mCAAmC;IACnC,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,EAAE;IACX,IAAI,EAAE,EAAE;IACR,IAAI,EAAE,EAAE;IACR,gBAAgB,EAAE,EAAE;IACpB,aAAa,EAAE,EAAE;IAEjB,KAAK,EAAE,EAAE;IACT,QAAQ,EAAE,EAAE;IACZ,iBAAiB,EAAE,EAAE;IAErB,KAAK,EAAE,EAAE;IACT,GAAG,EAAE,EAAE;IACP,OAAO,EAAE,EAAE;IACX,SAAS,EAAE,EAAE;IACb,kBAAkB,EAAE,EAAE;IACtB,YAAY,EAAE,EAAE;IAChB,gBAAgB,EAAE,EAAE;IAEpB,KAAK,EAAE,EAAE;IACT,MAAM,EAAE,EAAE;IACV,MAAM,EAAE,EAAE;IACV,eAAe,EAAE,EAAE;IACnB,IAAI,EAAE,EAAE;IAER,qCAAqC;IACrC,KAAK,EAAE,EAAE;IACT,kBAAkB,EAAE,EAAE;IACtB,eAAe,EAAE,EAAE;IAEnB,MAAM,EAAE,EAAE;IACV,mBAAmB,EAAE,EAAE;IAEvB,MAAM,EAAE,EAAE;IACV,oBAAoB,EAAE,EAAE;IACxB,cAAc,EAAE,EAAE;IAClB,kBAAkB,EAAE,EAAE;IAEtB,MAAM,EAAE,EAAE;IACV,QAAQ,EAAE,EAAE;IACZ,iBAAiB,EAAE,EAAE;IACrB,GAAG,EAAE,EAAE;IACP,GAAG,EAAE,EAAE;IAEP,mCAAmC;IACnC,OAAO,EAAE,GAAG;IACZ,KAAK,EAAE,GAAG;IACV,IAAI,EAAE,GAAG;IACT,MAAM,EAAE,GAAG;IAEX,OAAO,EAAE,GAAG;IACZ,MAAM,EAAE,GAAG;IACX,kBAAkB,EAAE,GAAG;IAEvB,QAAQ,EAAE,GAAG,EAAE,0CAA0C;IAEzD,gBAAgB;IAChB,OAAO,EAAE,IAAI;IACb,SAAS,EAAE,IAAI;IACf,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,IAAI;IAChB,UAAU,EAAE,IAAI;IAEhB,eAAe;IACf,IAAI,EAAE,EAAE;IAER,2BAA2B;IAC3B,MAAM,EAAE,EAAE;IACV,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE;IACd,QAAQ,EAAE,EAAE;IACZ,IAAI,EAAE,EAAE;IACR,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE;IACd,QAAQ,EAAE,EAAE;IACZ,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,IAAI;IACb,IAAI,EAAE,IAAI;IACV,mBAAmB,EAAE,EAAE;IACvB,cAAc,EAAE,IAAI;IACpB,wBAAwB,EAAE,EAAE;IAC5B,mBAAmB,EAAE,IAAI;IACzB,oBAAoB,EAAE,EAAE;IACxB,eAAe,EAAE,IAAI;IACrB,iCAAiC,EAAE,EAAE;IACrC,4BAA4B,EAAE,IAAI;IAClC,uBAAuB,EAAE,IAAI;IAC7B,oBAAoB,EAAE,IAAI;IAC1B,eAAe,EAAE,IAAI;IACrB,qBAAqB,EAAE,EAAE;IACzB,gBAAgB,EAAE,EAAE;IAEpB,eAAe;IACf,WAAW,EAAE,IAAI;IAEjB,mBAAmB;IACnB,IAAI,EAAE,IAAI;IAEV,sBAAsB;IACtB,IAAI,EAAE,IAAI;IACV,MAAM,EAAE,IAAI;IACZ,QAAQ,EAAE,IAAI;IACd,UAAU,EAAE,IAAI;IAEhB,cAAc;IACd,KAAK,EAAE,IAAI;IAEX,kBAAkB;IAClB,IAAI,EAAE,IAAI;IACV,MAAM,EAAE,IAAI;IAEZ,eAAe;IACf,IAAI,EAAE,EAAE;IACR,IAAI,EAAE,EAAE;IAER,aAAa;IACb,KAAK,EAAE,KAAK,EAAE,UAAU;IACxB,MAAM,EAAE,KAAK,EAAE,UAAU;IAEzB,wBAAwB;IACxB,KAAK,EAAE,EAAE;IACT,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,EAAE;IAEX,oBAAoB;IACpB,OAAO,EAAE,EAAE;IACX,QAAQ,EAAE,IAAI;CACf,CAAC;AAEF,SAAS,mBAAmB,CAAC,IAAY;IACvC,iDAAiD;IACjD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAExD,sBAAsB;IAEtB,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,cAAc,CAAC,UAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACpD,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,cAAc,CAAC,UAAU,CAAC,0BAA0B,CAAC,EAAE,CAAC;QAC1D,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC3C,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9C,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,qBAAqB;IACrB,IAAI,cAAc,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,OAAO,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,cAAc,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,OAAO,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,oCAAoC;IAC9E,CAAC;IAED,oBAAoB;IACpB,IAAI,cAAc,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,OAAO,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACzC,OAAO,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,uBAAuB;IACvB,IACE,cAAc,CAAC,UAAU,CAAC,UAAU,CAAC;QACrC,cAAc,CAAC,UAAU,CAAC,YAAY,CAAC;QACvC,cAAc,CAAC,UAAU,CAAC,YAAY,CAAC;QACvC,cAAc,CAAC,UAAU,CAAC,aAAa,CAAC;QACxC,cAAc,CAAC,UAAU,CAAC,aAAa,CAAC,EACxC,CAAC;QACD,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,gCAAgC;IAChC,IAAI,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC3C,OAAO,sBAAsB,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,cAAc,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAC7C,OAAO,sBAAsB,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC;IAED,qBAAqB;IACrB,IAAI,cAAc,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QAC9C,OAAO,sBAAsB,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,mCAAmC;IACnC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACzD,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,cAAc,CAAC,WAAW,EAAE,CAC5D,CAAC;IAEF,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,sBAAsB,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,4BAA4B,IAAI,0BAA0B,CAAC,CAAC;IACzE,OAAO,IAAI,CAAC,CAAC,qBAAqB;AACpC,CAAC;AAQD,MAAM,UAAU,mBAAmB,CACjC,MAAkC;IAElC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACpC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAwB;IAC/D,MAAM,CAAC,KAAK,EAAE,CAAC;AACjB,CAAC;AAED,MAAM,UAAgB,kBAAkB,CACtC,GAAW,EACX,MAAwB;;;QAExB,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YACjC,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAElC,0BAA0B;QAC1B,MAAM,MAAM,GACV,CAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU,EAAE,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC;SAC5C,CAAC,CAAC,KAAI,EAAE,CAAC;QAEZ,OAAO;YACL,MAAM,EAAE,kBAAkB;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAW,EAAE,oBAAoB;SAC7C,CAAC;IACJ,CAAC;CAAA;AAED,MAAM,UAAgB,oBAAoB,CACxC,MAAwB;;QAExB,MAAM,GAAG,GAAG;;+EAEiE,CAAC;QAC9E,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACpD,CAAC;CAAA;AAED,MAAM,UAAgB,2BAA2B,CAC/C,MAAwB,EACxB,WAAqB;;QAErB,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CACjC,WAAW,CAAC,GAAG,CAAC,CAAO,MAAM,EAAE,EAAE;YAC/B,MAAM,GAAG,GAAG;;6CAE2B,MAAM,GAAG,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACtD,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAChC,SAAS,EAAE,GAAG,CAAC,UAAU;gBACzB,UAAU,EAAE,GAAG,CAAC,YAAY;aAC7B,CAAC,CAAC,CAAC;QACN,CAAC,CAAA,CAAC,CACH,CAAC;QACF,OAAO,SAAS,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;CAAA;AAED,MAAM,UAAgB,2BAA2B,CAC/C,MAAwB,EACxB,UAAkB,EAClB,SAAiB;;QAEjB,MAAM,GAAG,GAAG;;qCAEuB,UAAU,kBAAkB,SAAS,GAAG,CAAC;QAC5E,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACpD,CAAC;CAAA;AAED,MAAM,UAAgB,wBAAwB,CAC5C,MAAwB,EACxB,UAAkB,EAClB,SAAiB,EACjB,UAAkB;;QAElB,MAAM,qBAAqB,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,GAAG,GAAG;0BACc,UAAU;uBACb,SAAS;wBACR,UAAU;wBACV,qBAAqB,IAAI,UAAU;wBACnC,qBAAqB,GAAG,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC;QACzE,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,IAAI,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,OAAO,GAAG,CAAC,WAAW,CAAC;QACzB,CAAC,CAAC,CAAC;QACH,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAC1C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,MAAM,CACxC,CAAC;QACF,iBAAiB,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACpD,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,GAAG,GAAG;8BACoB,UAAU;2BACb,SAAS;+BACL,SAAS;;;iCAGP,UAAU;8BACb,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACtD,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC3C,OAAO,GAAG,CAAC,WAAW,CAAC;YACzB,CAAC,CAAC,CAAC;YACH,iBAAiB,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,iBAAiB,CAAC;IAC3B,CAAC;CAAA;AAED,MAAM,UAAgB,6BAA6B,CACjD,MAAwB,EACxB,UAAkB,EAClB,UAAuD;;QAIvD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,UAAU,CAAC,GAAG,CAAC,CAAO,SAAS,EAAE,EAAE;YACjC,MAAM,KAAK,GAAG;;;;;oCAKgB,SAAS,CAAC,UAAU;+BACzB,SAAS,CAAC,SAAS;aACrC,CAAC;YACR,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACxD,OAAO;gBACL,SAAS,EAAE,GAAG,SAAS,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,EAAE;gBAC3D,WAAW,EAAE,GAAG,SAAS,CAAC,UAAU,IAAI,SAAS,CAAC,SAAS,EAAE;gBAC7D,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;oBAChC,MAAM,OAAO,GAAG,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,0BAA0B;oBAE9E,OAAO;wBACL,UAAU,EAAE,GAAG,CAAC,UAAU;wBAC1B,WAAW,EAAE,GAAG,CAAC,UAAU;wBAC3B,UAAU,EAAE,OAAO;wBACnB,SAAS,EAAE,GAAG,CAAC,SAAS;qBACzB,CAAC;gBACJ,CAAC,CAAC;aACH,CAAC;QACJ,CAAC,CAAA,CAAC,CACH,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;CAAA;AAED,MAAM,UAAU,sBAAsB,CACpC,gBAAwB;IAExB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAEzC,OAAO;QACL,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,+BAA+B;QAC7F,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,SAAS,EAAE,uCAAuC;QAC/E,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,4CAA4C;KAC9E,CAAC;AACJ,CAAC"}
|
|
@@ -10,7 +10,11 @@ import { MysqlConnectionConfig } from "./Mysql";
|
|
|
10
10
|
import { ClickHouseConnectionConfig } from "./ClickHouse";
|
|
11
11
|
import { MssqlConnectionConfig } from "./Mssql";
|
|
12
12
|
import { ConnectionPool } from "mssql";
|
|
13
|
-
|
|
13
|
+
import { DatabricksConnectionConfig } from "./Databricks";
|
|
14
|
+
import { DBSQLClient } from "@databricks/sql";
|
|
15
|
+
export type DatabaseConnection = Pool | snowflake.Connection | BigQuery | MysqlPool | ClickHouseClient | ConnectionPool | DBSQLClient;
|
|
16
|
+
export type DatabaseConnectionConfig = PostgresConnectionConfig | SnowflakeConnectionConfig | BigQueryConfig | MysqlConnectionConfig | ClickHouseConnectionConfig | MssqlConnectionConfig | DatabricksConnectionConfig;
|
|
17
|
+
export type DatabaseConnectionType = "postgresql" | "snowflake" | "bigquery" | "mysql" | "clickhouse" | "mssql" | "databricks";
|
|
14
18
|
export interface QuillQueryResults {
|
|
15
19
|
fields: {
|
|
16
20
|
name: string;
|
|
@@ -20,20 +24,20 @@ export interface QuillQueryResults {
|
|
|
20
24
|
[fieldName: string]: any;
|
|
21
25
|
}[];
|
|
22
26
|
}
|
|
23
|
-
export declare function getDatabaseCredentials(databaseType:
|
|
24
|
-
export declare function connectToDatabase(databaseType:
|
|
25
|
-
export declare function withConnection<T>(databaseType:
|
|
26
|
-
export declare function runQueryByDatabase(databaseType:
|
|
27
|
-
export declare function connectAndRunQuery(databaseType:
|
|
28
|
-
export declare function disconnectFromDatabase(databaseType:
|
|
29
|
-
export declare function getSchemasByDatabase(databaseType:
|
|
30
|
-
export declare function getTablesBySchemaByDatabase(databaseType:
|
|
27
|
+
export declare function getDatabaseCredentials(databaseType: DatabaseConnectionType, connectionString: string): DatabaseConnectionConfig;
|
|
28
|
+
export declare function connectToDatabase(databaseType: DatabaseConnectionType, config: DatabaseConnectionConfig): DatabaseConnection;
|
|
29
|
+
export declare function withConnection<T>(databaseType: DatabaseConnectionType, connectionString: string, callback: (connection: DatabaseConnection) => T): Promise<T>;
|
|
30
|
+
export declare function runQueryByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection, sql: string): Promise<QuillQueryResults> | undefined;
|
|
31
|
+
export declare function connectAndRunQuery(databaseType: DatabaseConnectionType, connectionString: string, sql: string): Promise<QuillQueryResults | undefined>;
|
|
32
|
+
export declare function disconnectFromDatabase(databaseType: DatabaseConnectionType, database: DatabaseConnection): void | Promise<void>;
|
|
33
|
+
export declare function getSchemasByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection): Promise<string[] | undefined>;
|
|
34
|
+
export declare function getTablesBySchemaByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection, schemaName: string | string[]): Promise<string[] | {
|
|
31
35
|
tableName: string;
|
|
32
36
|
schemaName: string;
|
|
33
37
|
}[] | undefined>;
|
|
34
|
-
export declare function getColumnsByTableByDatabase(databaseType:
|
|
35
|
-
export declare function getForeignKeysByDatabase(databaseType:
|
|
36
|
-
export declare function getColumnInfoBySchemaByDatabase(databaseType:
|
|
38
|
+
export declare function getColumnsByTableByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection, schemaName: string, tableName: string): Promise<string[] | undefined>;
|
|
39
|
+
export declare function getForeignKeysByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection, schemaName: string, tableName: string, primaryKey: string): Promise<string[] | undefined>;
|
|
40
|
+
export declare function getColumnInfoBySchemaByDatabase(databaseType: DatabaseConnectionType, connection: DatabaseConnection, schemaName: string, tables: string[] | {
|
|
37
41
|
tableName: string;
|
|
38
42
|
schemaName: string;
|
|
39
43
|
}[]): Promise<{
|
|
@@ -11,8 +11,9 @@ import { connectToPostgres, disconnectFromPostgres, formatPostgresConfig, getTab
|
|
|
11
11
|
import { connectToSnowflake, disconnectFromSnowflake, formatSnowflakeConfig, getTablesBySchemaSnowflake, getColumnsByTableSnowflake, runQuerySnowflake, getForeignKeysSnowflake, getSchemaColumnInfoSnowflake, getSchemasSnowflake, } from "./Snowflake";
|
|
12
12
|
import { formatBigQueryConfig, connectToBigQuery, runQueryBigQuery, getTablesBySchemaBigQuery, getColumnsByTableBigQuery, getForeignKeysBigQuery, getSchemaColumnInfoBigQuery, getSchemaBigQuery, } from "./BigQuery";
|
|
13
13
|
import { connectToMysql, disconnectFromMysql, formatMysqlConfig, runQueryMysql, getTablesBySchemaMysql, getColumnsByTableMysql, getForeignKeysMysql, getSchemaColumnInfoMysql, getSchemasMysql, } from "./Mysql";
|
|
14
|
-
import { connectToClickHouse, disconnectFromClickHouse, formatClickHouseConfig, getColumnsByTableClickHouse, getForeignKeysClickHouse, getSchemaColumnInfoClickHouse, getSchemasClickHouse, getTablesBySchemaClickHouse, runQueryClickHouse } from "./ClickHouse";
|
|
15
|
-
import { connectToMssql, disconnectFromMssql, formatMssqlConfig, getColumnsByTableMssql, getSchemaColumnInfoMssql, getSchemasMssql, getTablesBySchemaMssql, runQueryMssql } from "./Mssql";
|
|
14
|
+
import { connectToClickHouse, disconnectFromClickHouse, formatClickHouseConfig, getColumnsByTableClickHouse, getForeignKeysClickHouse, getSchemaColumnInfoClickHouse, getSchemasClickHouse, getTablesBySchemaClickHouse, runQueryClickHouse, } from "./ClickHouse";
|
|
15
|
+
import { connectToMssql, disconnectFromMssql, formatMssqlConfig, getColumnsByTableMssql, getSchemaColumnInfoMssql, getSchemasMssql, getTablesBySchemaMssql, runQueryMssql, } from "./Mssql";
|
|
16
|
+
import { connectToDatabricks, disconnectFromDatabricks, formatDatabricksConfig, getColumnsByTableDatabricks, getSchemaColumnInfoDatabricks, getSchemasDatabricks, getTablesBySchemaDatabricks, runQueryDatabricks, } from "./Databricks";
|
|
16
17
|
export function getDatabaseCredentials(databaseType, connectionString) {
|
|
17
18
|
switch (databaseType.toLowerCase()) {
|
|
18
19
|
case "postgres":
|
|
@@ -29,6 +30,8 @@ export function getDatabaseCredentials(databaseType, connectionString) {
|
|
|
29
30
|
return formatClickHouseConfig(connectionString);
|
|
30
31
|
case "mssql":
|
|
31
32
|
return formatMssqlConfig(connectionString);
|
|
33
|
+
case "databricks":
|
|
34
|
+
return formatDatabricksConfig(connectionString);
|
|
32
35
|
default:
|
|
33
36
|
throw new Error("Invalid database type");
|
|
34
37
|
}
|
|
@@ -49,6 +52,8 @@ export function connectToDatabase(databaseType, config) {
|
|
|
49
52
|
return connectToClickHouse(config);
|
|
50
53
|
case "mssql":
|
|
51
54
|
return connectToMssql(config);
|
|
55
|
+
case "databricks":
|
|
56
|
+
return connectToDatabricks(config);
|
|
52
57
|
default:
|
|
53
58
|
return connectToPostgres(config);
|
|
54
59
|
}
|
|
@@ -84,6 +89,8 @@ export function runQueryByDatabase(databaseType, connection, sql) {
|
|
|
84
89
|
return runQueryClickHouse(sql, connection);
|
|
85
90
|
case "mssql":
|
|
86
91
|
return runQueryMssql(sql, connection);
|
|
92
|
+
case "databricks":
|
|
93
|
+
return runQueryDatabricks(sql, connection);
|
|
87
94
|
default:
|
|
88
95
|
return undefined;
|
|
89
96
|
}
|
|
@@ -109,6 +116,8 @@ export function disconnectFromDatabase(databaseType, database) {
|
|
|
109
116
|
return disconnectFromClickHouse(database);
|
|
110
117
|
case "mssql":
|
|
111
118
|
return disconnectFromMssql(database);
|
|
119
|
+
case "databricks":
|
|
120
|
+
return disconnectFromDatabricks(database);
|
|
112
121
|
default:
|
|
113
122
|
return undefined;
|
|
114
123
|
}
|
|
@@ -130,6 +139,8 @@ export function getSchemasByDatabase(databaseType, connection) {
|
|
|
130
139
|
return getSchemasClickHouse(connection);
|
|
131
140
|
case "mssql":
|
|
132
141
|
return getSchemasMssql(connection);
|
|
142
|
+
case "databricks":
|
|
143
|
+
return getSchemasDatabricks(connection);
|
|
133
144
|
default:
|
|
134
145
|
return undefined;
|
|
135
146
|
}
|
|
@@ -153,6 +164,8 @@ export function getTablesBySchemaByDatabase(databaseType, connection, schemaName
|
|
|
153
164
|
return getTablesBySchemaClickHouse(connection, schemaName);
|
|
154
165
|
case "mssql":
|
|
155
166
|
return getTablesBySchemaMssql(connection, schemaName);
|
|
167
|
+
case "databricks":
|
|
168
|
+
return getTablesBySchemaDatabricks(connection, schemaName);
|
|
156
169
|
default:
|
|
157
170
|
return undefined;
|
|
158
171
|
}
|
|
@@ -176,6 +189,8 @@ export function getColumnsByTableByDatabase(databaseType, connection, schemaName
|
|
|
176
189
|
return getColumnsByTableClickHouse(connection, schemaName, tableName);
|
|
177
190
|
case "mssql":
|
|
178
191
|
return getColumnsByTableMssql(connection, schemaName, tableName);
|
|
192
|
+
case "databricks":
|
|
193
|
+
return getColumnsByTableDatabricks(connection, schemaName, tableName);
|
|
179
194
|
default:
|
|
180
195
|
return undefined;
|
|
181
196
|
}
|
|
@@ -199,6 +214,9 @@ export function getForeignKeysByDatabase(databaseType, connection, schemaName, t
|
|
|
199
214
|
case "mssql":
|
|
200
215
|
// getForeignKeys is a deprecated function that is never used
|
|
201
216
|
return [];
|
|
217
|
+
case "databricks":
|
|
218
|
+
// getForeignKeys is a deprecated function that is never used
|
|
219
|
+
return [];
|
|
202
220
|
default:
|
|
203
221
|
return undefined;
|
|
204
222
|
}
|
|
@@ -220,6 +238,8 @@ export function getColumnInfoBySchemaByDatabase(databaseType, connection, schema
|
|
|
220
238
|
return getSchemaColumnInfoClickHouse(connection, schemaName, tables);
|
|
221
239
|
case "mssql":
|
|
222
240
|
return getSchemaColumnInfoMssql(connection, schemaName, tables);
|
|
241
|
+
case "databricks":
|
|
242
|
+
return getSchemaColumnInfoDatabricks(connection, schemaName, tables);
|
|
223
243
|
default:
|
|
224
244
|
return undefined;
|
|
225
245
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"DatabaseHelper.js","sourceRoot":"","sources":["../../../src/db/DatabaseHelper.ts"],"names":[],"mappings":";;;;;;;;;AAKA,OAAO,EAEL,iBAAiB,EACjB,sBAAsB,EACtB,oBAAoB,EACpB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,2BAA2B,EAC3B,kBAAkB,GACnB,MAAM,YAAY,CAAC;AACpB,OAAO,EAEL,kBAAkB,EAClB,uBAAuB,EACvB,qBAAqB,EACrB,0BAA0B,EAC1B,0BAA0B,EAC1B,iBAAiB,EACjB,uBAAuB,EACvB,4BAA4B,EAC5B,mBAAmB,GACpB,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,oBAAoB,EAEpB,iBAAiB,EACjB,gBAAgB,EAChB,yBAAyB,EACzB,yBAAyB,EACzB,sBAAsB,EACtB,2BAA2B,EAC3B,iBAAiB,GAClB,MAAM,YAAY,CAAC;AACpB,OAAO,EAEL,cAAc,EACd,mBAAmB,EACnB,iBAAiB,EACjB,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,mBAAmB,EACnB,wBAAwB,EACxB,eAAe,GAChB,MAAM,SAAS,CAAC;AACjB,OAAO,
|
|
1
|
+
{"version":3,"file":"DatabaseHelper.js","sourceRoot":"","sources":["../../../src/db/DatabaseHelper.ts"],"names":[],"mappings":";;;;;;;;;AAKA,OAAO,EAEL,iBAAiB,EACjB,sBAAsB,EACtB,oBAAoB,EACpB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,2BAA2B,EAC3B,kBAAkB,GACnB,MAAM,YAAY,CAAC;AACpB,OAAO,EAEL,kBAAkB,EAClB,uBAAuB,EACvB,qBAAqB,EACrB,0BAA0B,EAC1B,0BAA0B,EAC1B,iBAAiB,EACjB,uBAAuB,EACvB,4BAA4B,EAC5B,mBAAmB,GACpB,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,oBAAoB,EAEpB,iBAAiB,EACjB,gBAAgB,EAChB,yBAAyB,EACzB,yBAAyB,EACzB,sBAAsB,EACtB,2BAA2B,EAC3B,iBAAiB,GAClB,MAAM,YAAY,CAAC;AACpB,OAAO,EAEL,cAAc,EACd,mBAAmB,EACnB,iBAAiB,EACjB,aAAa,EACb,sBAAsB,EACtB,sBAAsB,EACtB,mBAAmB,EACnB,wBAAwB,EACxB,eAAe,GAChB,MAAM,SAAS,CAAC;AACjB,OAAO,EAEL,mBAAmB,EACnB,wBAAwB,EACxB,sBAAsB,EACtB,2BAA2B,EAC3B,wBAAwB,EACxB,6BAA6B,EAC7B,oBAAoB,EACpB,2BAA2B,EAC3B,kBAAkB,GACnB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,cAAc,EACd,mBAAmB,EACnB,iBAAiB,EACjB,sBAAsB,EACtB,wBAAwB,EACxB,eAAe,EACf,sBAAsB,EAEtB,aAAa,GACd,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,mBAAmB,EAEnB,wBAAwB,EACxB,sBAAsB,EACtB,2BAA2B,EAC3B,6BAA6B,EAC7B,oBAAoB,EACpB,2BAA2B,EAC3B,kBAAkB,GACnB,MAAM,cAAc,CAAC;AAmCtB,MAAM,UAAU,sBAAsB,CACpC,YAAoC,EACpC,gBAAwB;IAExB,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;QACnC,KAAK,UAAU;YACb,OAAO,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAChD,KAAK,YAAY;YACf,OAAO,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAChD,KAAK,WAAW;YACd,OAAO,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QACjD,KAAK,UAAU;YACb,OAAO,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAChD,KAAK,OAAO;YACV,OAAO,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;QAC7C,KAAK,YAAY;YACf,OAAO,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;QAClD,KAAK,OAAO;YACV,OAAO,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;QAC7C,KAAK,YAAY;YACf,OAAO,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;QAClD;YACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,YAAoC,EACpC,MAAgC;IAEhC,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;QACnC,KAAK,UAAU;YACb,OAAO,iBAAiB,CAAC,MAAkC,CAAC,CAAC;QAC/D,KAAK,YAAY;YACf,OAAO,iBAAiB,CAAC,MAAkC,CAAC,CAAC;QAC/D,KAAK,WAAW;YACd,OAAO,kBAAkB,CAAC,MAAmC,CAAC,CAAC;QACjE,KAAK,UAAU;YACb,OAAO,iBAAiB,CAAC,MAAwB,CAAC,CAAC;QACrD,KAAK,OAAO;YACV,OAAO,cAAc,CAAC,MAA+B,CAAC,CAAC;QACzD,KAAK,YAAY;YACf,OAAO,mBAAmB,CAAC,MAAoC,CAAC,CAAC;QACnE,KAAK,OAAO;YACV,OAAO,cAAc,CAAC,MAA+B,CAAC,CAAC;QACzD,KAAK,YAAY;YACf,OAAO,mBAAmB,CAAC,MAAoC,CAAC,CAAC;QACnE;YACE,OAAO,iBAAiB,CAAC,MAAkC,CAAC,CAAC;IACjE,CAAC;AACH,CAAC;AAED,MAAM,UAAgB,cAAc,CAClC,YAAoC,EACpC,gBAAwB,EACxB,QAA+C;;QAE/C,MAAM,MAAM,GAAG,sBAAsB,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;QACtE,MAAM,UAAU,GAAG,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC3D,IAAI,CAAC;YACH,OAAO,MAAM,QAAQ,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,OAAO,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAC/B,MAAM,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC,CAC/D,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,sBAAsB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;CAAA;AAED,MAAM,UAAU,kBAAkB,CAChC,YAAoC,EACpC,UAA8B,EAC9B,GAAW;IAEX,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;QACnC,KAAK,UAAU;YACb,OAAO,gBAAgB,CAAC,GAAG,EAAE,UAAkB,CAAC,CAAC;QACnD,KAAK,YAAY;YACf,OAAO,gBAAgB,CAAC,GAAG,EAAE,UAAkB,CAAC,CAAC;QACnD,KAAK,WAAW;YACd,OAAO,iBAAiB,CAAC,GAAG,EAAE,UAAkC,CAAC,CAAC;QACpE,KAAK,UAAU;YACb,OAAO,gBAAgB,CAAC,GAAG,EAAE,UAAsB,CAAC,CAAC;QACvD,KAAK,OAAO;YACV,OAAO,aAAa,CAAC,GAAG,EAAE,UAAuB,CAAC,CAAC;QACrD,KAAK,YAAY;YACf,OAAO,kBAAkB,CAAC,GAAG,EAAE,UAA8B,CAAC,CAAC;QACjE,KAAK,OAAO;YACV,OAAO,aAAa,CAAC,GAAG,EAAE,UAA4B,CAAC,CAAC;QAC1D,KAAK,YAAY;YACf,OAAO,kBAAkB,CAAC,GAAG,EAAE,UAAyB,CAAC,CAAC;QAC5D;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,MAAM,UAAgB,kBAAkB,CACtC,YAAoC,EACpC,gBAAwB,EACxB,GAAW;;QAEX,OAAO,cAAc,CACnB,YAAY,EACZ,gBAAgB,EAChB,CAAO,UAAU,EAAE,EAAE,gDACnB,OAAA,MAAM,kBAAkB,CAAC,YAAY,EAAE,UAAU,EAAE,GAAG,CAAC,CAAA,GAAA,CAC1D,CAAC;IACJ,CAAC;CAAA;AAED,MAAM,UAAU,sBAAsB,CACpC,YAAoC,EACpC,QAA4B;IAE5B,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;QACnC,KAAK,UAAU;YACb,OAAO,sBAAsB,CAAC,QAAgB,CAAC,CAAC;QAClD,KAAK,YAAY;YACf,OAAO,sBAAsB,CAAC,QAAgB,CAAC,CAAC;QAClD,KAAK,WAAW;YACd,OAAO,uBAAuB,CAAC,QAAgC,CAAC,CAAC;QACnE,KAAK,UAAU;YACb,OAAO,CAAC,4CAA4C;QACtD,KAAK,OAAO;YACV,OAAO,mBAAmB,CAAC,QAAqB,CAAC,CAAC;QACpD,KAAK,YAAY;YACf,OAAO,wBAAwB,CAAC,QAA4B,CAAC,CAAC;QAChE,KAAK,OAAO;YACV,OAAO,mBAAmB,CAAC,QAA0B,CAAC,CAAC;QACzD,KAAK,YAAY;YACf,OAAO,wBAAwB,CAAC,QAAuB,CAAC,CAAC;QAC3D;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,MAAM,UAAgB,oBAAoB,CACxC,YAAoC,EACpC,UAA8B;;QAE9B,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;YACnC,KAAK,UAAU;gBACb,OAAO,kBAAkB,CAAC,UAAkB,CAAC,CAAC;YAChD,KAAK,YAAY;gBACf,OAAO,kBAAkB,CAAC,UAAkB,CAAC,CAAC;YAChD,KAAK,WAAW;gBACd,OAAO,mBAAmB,CAAC,UAAkC,CAAC,CAAC;YACjE,KAAK,UAAU;gBACb,OAAO,iBAAiB,CAAC,UAAsB,CAAC,CAAC;YACnD,KAAK,OAAO;gBACV,OAAO,eAAe,CAAC,UAAuB,CAAC,CAAC;YAClD,KAAK,YAAY;gBACf,OAAO,oBAAoB,CAAC,UAA8B,CAAC,CAAC;YAC9D,KAAK,OAAO;gBACV,OAAO,eAAe,CAAC,UAA4B,CAAC,CAAC;YACvD,KAAK,YAAY;gBACf,OAAO,oBAAoB,CAAC,UAAyB,CAAC,CAAC;YACzD;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;CAAA;AAED,6BAA6B;AAC7B,MAAM,UAAgB,2BAA2B,CAC/C,YAAoC,EACpC,UAA8B,EAC9B,UAA6B;;QAE7B,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;YACnC,KAAK,UAAU;gBACb,OAAO,yBAAyB,CAC9B,UAAkB,EAClB,UAAsB,CACvB,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,yBAAyB,CAC9B,UAAkB,EAClB,UAAsB,CACvB,CAAC;YACJ,KAAK,WAAW;gBACd,OAAO,0BAA0B,CAC/B,UAAkC,EAClC,UAAsB,CACvB,CAAC;YACJ,KAAK,UAAU;gBACb,OAAO,yBAAyB,CAC9B,UAAsB,EACtB,UAAsB,CACvB,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO,sBAAsB,CAC3B,UAAuB,EACvB,UAAsB,CACvB,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAChC,UAA8B,EAC9B,UAAsB,CACvB,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO,sBAAsB,CAC3B,UAA4B,EAC5B,UAAsB,CACvB,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAChC,UAAyB,EACzB,UAAsB,CACvB,CAAC;YACJ;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;CAAA;AAED,6BAA6B;AAC7B,MAAM,UAAgB,2BAA2B,CAC/C,YAAoC,EACpC,UAA8B,EAC9B,UAAkB,EAClB,SAAiB;;QAEjB,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;YACnC,KAAK,UAAU;gBACb,OAAO,yBAAyB,CAC9B,UAAkB,EAClB,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,yBAAyB,CAC9B,UAAkB,EAClB,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,WAAW;gBACd,OAAO,0BAA0B,CAC/B,UAAkC,EAClC,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,UAAU;gBACb,OAAO,yBAAyB,CAC9B,UAAsB,EACtB,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO,sBAAsB,CAC3B,UAAuB,EACvB,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAChC,UAA8B,EAC9B,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO,sBAAsB,CAC3B,UAA4B,EAC5B,UAAU,EACV,SAAS,CACV,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAChC,UAAyB,EACzB,UAAU,EACV,SAAS,CACV,CAAC;YACJ;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;CAAA;AAED,MAAM,UAAgB,wBAAwB,CAC5C,YAAoC,EACpC,UAA8B,EAC9B,UAAkB,EAClB,SAAiB,EACjB,UAAkB;;QAElB,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;YACnC,KAAK,UAAU;gBACb,OAAO,sBAAsB,CAC3B,UAAkB,EAClB,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,sBAAsB,CAC3B,UAAkB,EAClB,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,WAAW;gBACd,OAAO,uBAAuB,CAC5B,UAAkC,EAClC,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,UAAU;gBACb,OAAO,sBAAsB,CAC3B,UAAsB,EACtB,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,OAAO;gBACV,OAAO,mBAAmB,CACxB,UAAuB,EACvB,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,YAAY;gBACf,OAAO,wBAAwB,CAC7B,UAA8B,EAC9B,UAAU,EACV,SAAS,EACT,UAAU,CACX,CAAC;YACJ,KAAK,OAAO;gBACV,6DAA6D;gBAC7D,OAAO,EAAE,CAAC;YACZ,KAAK,YAAY;gBACf,6DAA6D;gBAC7D,OAAO,EAAE,CAAC;YACZ;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;CAAA;AAED,MAAM,UAAU,+BAA+B,CAC7C,YAAoC,EACpC,UAA8B,EAC9B,UAAkB,EAClB,MAA8D;IAE9D,QAAQ,YAAY,CAAC,WAAW,EAAE,EAAE,CAAC;QACnC,KAAK,UAAU;YACb,OAAO,2BAA2B,CAChC,UAAkB,EAClB,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,YAAY;YACf,OAAO,2BAA2B,CAChC,UAAkB,EAClB,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,WAAW;YACd,OAAO,4BAA4B,CACjC,UAAkC,EAClC,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,UAAU;YACb,OAAO,2BAA2B,CAChC,UAAsB,EACtB,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,OAAO;YACV,OAAO,wBAAwB,CAC7B,UAAuB,EACvB,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,YAAY;YACf,OAAO,6BAA6B,CAClC,UAA8B,EAC9B,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,OAAO;YACV,OAAO,wBAAwB,CAC7B,UAA4B,EAC5B,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ,KAAK,YAAY;YACf,OAAO,6BAA6B,CAClC,UAAyB,EACzB,UAAU,EACV,MAAqD,CACtD,CAAC;QACJ;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { DBSQLClient } from "@databricks/sql";
|
|
2
|
+
import { QuillQueryResults } from "./DatabaseHelper";
|
|
3
|
+
export type DatabricksConnectionConfig = {
|
|
4
|
+
host: string;
|
|
5
|
+
path: string;
|
|
6
|
+
token: string;
|
|
7
|
+
};
|
|
8
|
+
export declare function connectToDatabricks(config: DatabricksConnectionConfig): DBSQLClient;
|
|
9
|
+
export declare function disconnectFromDatabricks(client: DBSQLClient): void;
|
|
10
|
+
export declare function runQueryDatabricks(sql: string, client: DBSQLClient): Promise<QuillQueryResults>;
|
|
11
|
+
export declare function getSchemasDatabricks(client: DBSQLClient): Promise<string[]>;
|
|
12
|
+
export declare function getTablesBySchemaDatabricks(client: DBSQLClient, schemaNames: string[]): Promise<{
|
|
13
|
+
tableName: string;
|
|
14
|
+
schemaName: string;
|
|
15
|
+
catalogName: string;
|
|
16
|
+
}[]>;
|
|
17
|
+
export declare function getColumnsByTableDatabricks(client: DBSQLClient, schemaName: string, tableName: string): Promise<string[]>;
|
|
18
|
+
export declare function getForeignKeysDatabricks(): Promise<string[]>;
|
|
19
|
+
export declare function getSchemaColumnInfoDatabricks(client: DBSQLClient, schemaName: string, tableNames: {
|
|
20
|
+
tableName: string;
|
|
21
|
+
schemaName: string;
|
|
22
|
+
}[]): Promise<{
|
|
23
|
+
tableName: string;
|
|
24
|
+
columns: {
|
|
25
|
+
columnName: string;
|
|
26
|
+
dataTypeID: number;
|
|
27
|
+
}[];
|
|
28
|
+
}[]>;
|
|
29
|
+
export declare function formatDatabricksConfig(connectionString: string): DatabricksConnectionConfig;
|