@l-v-yonsama/multi-platform-database-drivers 0.1.52 → 0.1.55
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/built/src/drivers/AwsDriver.d.ts +2 -2
- package/built/src/drivers/BaseDriver.d.ts +6 -6
- package/built/src/drivers/BaseDriver.js +9 -52
- package/built/src/drivers/BaseDriver.js.map +1 -1
- package/built/src/drivers/MySQLDriver.d.ts +4 -4
- package/built/src/drivers/MySQLDriver.js +19 -20
- package/built/src/drivers/MySQLDriver.js.map +1 -1
- package/built/src/drivers/PostgresDriver.d.ts +5 -5
- package/built/src/drivers/PostgresDriver.js +20 -19
- package/built/src/drivers/PostgresDriver.js.map +1 -1
- package/built/src/drivers/RDSBaseDriver.d.ts +11 -1
- package/built/src/drivers/RDSBaseDriver.js +85 -1
- package/built/src/drivers/RDSBaseDriver.js.map +1 -1
- package/built/src/drivers/RedisDriver.d.ts +2 -2
- package/built/src/drivers/RedisDriver.js +1 -1
- package/built/src/drivers/RedisDriver.js.map +1 -1
- package/built/src/drivers/aws/AwsCloudwatchServiceClient.d.ts +2 -2
- package/built/src/drivers/aws/AwsCloudwatchServiceClient.js +1 -1
- package/built/src/drivers/aws/AwsCloudwatchServiceClient.js.map +1 -1
- package/built/src/drivers/aws/AwsS3ServiceClient.d.ts +2 -2
- package/built/src/drivers/aws/AwsS3ServiceClient.js +1 -1
- package/built/src/drivers/aws/AwsS3ServiceClient.js.map +1 -1
- package/built/src/drivers/aws/AwsSQSServiceClient.d.ts +2 -2
- package/built/src/drivers/aws/AwsSQSServiceClient.js +2 -2
- package/built/src/drivers/aws/AwsSQSServiceClient.js.map +1 -1
- package/built/src/helpers/ResourceHelper.d.ts +8 -0
- package/built/src/helpers/ResourceHelper.js +94 -0
- package/built/src/helpers/ResourceHelper.js.map +1 -0
- package/built/src/helpers/SQLHelper.d.ts +2 -2
- package/built/src/helpers/SQLHelper.js +11 -11
- package/built/src/helpers/SQLHelper.js.map +1 -1
- package/built/src/helpers/index.d.ts +1 -0
- package/built/src/helpers/index.js +1 -0
- package/built/src/helpers/index.js.map +1 -1
- package/built/src/resource/DbResource.d.ts +27 -26
- package/built/src/resource/DbResource.js +71 -86
- package/built/src/resource/DbResource.js.map +1 -1
- package/built/src/resource/MultipleResultSetDataHolder.d.ts +2 -1
- package/built/src/resource/MultipleResultSetDataHolder.js.map +1 -1
- package/built/src/resource/ResultSetDataHolder.d.ts +12 -41
- package/built/src/resource/ResultSetDataHolder.js +5 -18
- package/built/src/resource/ResultSetDataHolder.js.map +1 -1
- package/built/src/types/CompareKey.d.ts +9 -0
- package/built/src/types/CompareKey.js +3 -0
- package/built/src/types/CompareKey.js.map +1 -0
- package/built/src/types/ResourceType.d.ts +3 -1
- package/built/src/types/ResourceType.js +3 -1
- package/built/src/types/ResourceType.js.map +1 -1
- package/built/src/types/index.d.ts +2 -0
- package/built/src/types/index.js +2 -0
- package/built/src/types/index.js.map +1 -1
- package/built/src/types/resource/Annonations.d.ts +33 -0
- package/built/src/types/resource/Annonations.js +36 -0
- package/built/src/types/resource/Annonations.js.map +1 -0
- package/built/src/types/resource/index.d.ts +1 -0
- package/built/src/types/resource/index.js +5 -0
- package/built/src/types/resource/index.js.map +1 -0
- package/package.json +2 -2
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { BaseDriver } from './BaseDriver';
|
|
2
|
-
import { DbConnection,
|
|
2
|
+
import { DbConnection, AwsDatabase } from '../resource';
|
|
3
3
|
import { AwsCredentialIdentityProvider, AwsCredentialIdentity } from '@aws-sdk/types';
|
|
4
4
|
import { AwsSQSServiceClient } from './aws/AwsSQSServiceClient';
|
|
5
5
|
import { AwsCloudwatchServiceClient } from './aws/AwsCloudwatchServiceClient';
|
|
@@ -22,7 +22,7 @@ export declare class AwsDriver extends BaseDriver {
|
|
|
22
22
|
getClientByResourceType<T extends AwsServiceClient = AwsServiceClient>(resourceType: ResourceType): T | undefined;
|
|
23
23
|
private createAwsCredential;
|
|
24
24
|
connectSub(): Promise<string>;
|
|
25
|
-
getInfomationSchemas(): Promise<
|
|
25
|
+
getInfomationSchemas(): Promise<AwsDatabase[]>;
|
|
26
26
|
test(with_connect?: boolean): Promise<string>;
|
|
27
27
|
closeSub(): Promise<string>;
|
|
28
28
|
}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
|
-
import { ColumnResolver,
|
|
2
|
-
import { GeneralResult, ScanParams } from '../types';
|
|
1
|
+
import { ColumnResolver, DbConnection, DbDatabase, ResultSetDataHolder, SchemaAndTableHints } from '../resource';
|
|
2
|
+
import { CompareKey, GeneralResult, ScanParams } from '../types';
|
|
3
3
|
import { DBError } from './DBError';
|
|
4
|
+
import { Statement } from 'pgsql-ast-parser';
|
|
4
5
|
export interface Scannable {
|
|
5
6
|
scan(params: ScanParams): Promise<ResultSetDataHolder>;
|
|
6
7
|
}
|
|
@@ -16,12 +17,10 @@ export declare abstract class BaseDriver {
|
|
|
16
17
|
initBaseStatus(): void;
|
|
17
18
|
isNeedsSsh(): boolean;
|
|
18
19
|
isQuery(sql: string): boolean;
|
|
20
|
+
parseQuery(sql: string): Statement | undefined;
|
|
19
21
|
createColumnResolver(sql?: string): ColumnResolver;
|
|
20
22
|
parseSchemaAndTableHints(sql: string): SchemaAndTableHints;
|
|
21
23
|
flow<T = any>(f: () => Promise<T>): Promise<GeneralResult<T>>;
|
|
22
|
-
resetDefaultSchema(database: DbDatabase): void;
|
|
23
|
-
resolveColumn(column: string, resolver?: ColumnResolver): DbColumn | undefined;
|
|
24
|
-
resolveColumnComment(column: string, resolver?: ColumnResolver): string;
|
|
25
24
|
connectToSshServer(): Promise<string>;
|
|
26
25
|
connect(): Promise<string>;
|
|
27
26
|
disconnect(): Promise<string>;
|
|
@@ -33,6 +32,7 @@ export declare abstract class BaseDriver {
|
|
|
33
32
|
}
|
|
34
33
|
export interface RequestSqlOptions {
|
|
35
34
|
binds?: string[];
|
|
36
|
-
needsColumnResolve?: boolean;
|
|
37
35
|
maxRows?: number;
|
|
36
|
+
tableName?: string;
|
|
37
|
+
compareKeys?: CompareKey[];
|
|
38
38
|
}
|
|
@@ -5,8 +5,8 @@ const tslib_1 = require("tslib");
|
|
|
5
5
|
const tunnel = tslib_1.__importStar(require("tunnel-ssh"));
|
|
6
6
|
const get_port_please_1 = require("get-port-please");
|
|
7
7
|
const fs = tslib_1.__importStar(require("fs"));
|
|
8
|
-
const types_1 = require("../types");
|
|
9
8
|
const DBError_1 = require("./DBError");
|
|
9
|
+
const pgsql_ast_parser_1 = require("pgsql-ast-parser");
|
|
10
10
|
function isScannable(arg) {
|
|
11
11
|
if (!arg) {
|
|
12
12
|
return false;
|
|
@@ -40,6 +40,14 @@ class BaseDriver {
|
|
|
40
40
|
}
|
|
41
41
|
return false;
|
|
42
42
|
}
|
|
43
|
+
parseQuery(sql) {
|
|
44
|
+
try {
|
|
45
|
+
return (0, pgsql_ast_parser_1.parseFirst)(sql);
|
|
46
|
+
}
|
|
47
|
+
catch (_) {
|
|
48
|
+
}
|
|
49
|
+
return undefined;
|
|
50
|
+
}
|
|
43
51
|
createColumnResolver(sql) {
|
|
44
52
|
if (sql) {
|
|
45
53
|
const hints = this.parseSchemaAndTableHints(sql);
|
|
@@ -85,57 +93,6 @@ class BaseDriver {
|
|
|
85
93
|
};
|
|
86
94
|
});
|
|
87
95
|
}
|
|
88
|
-
resetDefaultSchema(database) {
|
|
89
|
-
const searchNames = [];
|
|
90
|
-
if (this.conRes.database) {
|
|
91
|
-
searchNames.push(this.conRes.database);
|
|
92
|
-
}
|
|
93
|
-
if (this.conRes.user) {
|
|
94
|
-
searchNames.push(this.conRes.user);
|
|
95
|
-
}
|
|
96
|
-
searchNames.push('public');
|
|
97
|
-
for (const searchName of searchNames) {
|
|
98
|
-
const child = database.getChildByName(searchName, {
|
|
99
|
-
resouceType: types_1.ResourceType.Schema,
|
|
100
|
-
});
|
|
101
|
-
if (child) {
|
|
102
|
-
child.isDefault = true;
|
|
103
|
-
return;
|
|
104
|
-
}
|
|
105
|
-
}
|
|
106
|
-
const child = database
|
|
107
|
-
.getChildren()
|
|
108
|
-
.find((it) => it.getResouceType() === types_1.ResourceType.Schema);
|
|
109
|
-
if (child) {
|
|
110
|
-
child.isDefault = true;
|
|
111
|
-
}
|
|
112
|
-
}
|
|
113
|
-
resolveColumn(column, resolver) {
|
|
114
|
-
if (resolver === undefined) {
|
|
115
|
-
return undefined;
|
|
116
|
-
}
|
|
117
|
-
if (this.conRes) {
|
|
118
|
-
for (let i = 0; i < resolver.hints.list.length; i++) {
|
|
119
|
-
const hints = resolver.hints.list[i];
|
|
120
|
-
const t = this.conRes.findResource(types_1.ResourceType.Table, hints.table);
|
|
121
|
-
if (t) {
|
|
122
|
-
return t.getChildByName(column, { unwrapQuote: true });
|
|
123
|
-
}
|
|
124
|
-
else {
|
|
125
|
-
}
|
|
126
|
-
}
|
|
127
|
-
}
|
|
128
|
-
else {
|
|
129
|
-
}
|
|
130
|
-
return undefined;
|
|
131
|
-
}
|
|
132
|
-
resolveColumnComment(column, resolver) {
|
|
133
|
-
const c = this.resolveColumn(column, resolver);
|
|
134
|
-
if (c && c.comment) {
|
|
135
|
-
return c.comment;
|
|
136
|
-
}
|
|
137
|
-
return '';
|
|
138
|
-
}
|
|
139
96
|
connectToSshServer() {
|
|
140
97
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
141
98
|
this.sshLocalPort = yield (0, get_port_please_1.getPort)({ portRange: [13000, 15100] });
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"BaseDriver.js","sourceRoot":"","sources":["../../../src/drivers/BaseDriver.ts"],"names":[],"mappings":";;;;AAAA,2DAAqC;AACrC,qDAA0C;AAC1C,+CAAyB;
|
|
1
|
+
{"version":3,"file":"BaseDriver.js","sourceRoot":"","sources":["../../../src/drivers/BaseDriver.ts"],"names":[],"mappings":";;;;AAAA,2DAAqC;AACrC,qDAA0C;AAC1C,+CAAyB;AAWzB,uCAAoC;AACpC,uDAAgE;AAMhE,SAAgB,WAAW,CAAC,GAAQ;IAClC,IAAI,CAAC,GAAG,EAAE;QACR,OAAO,KAAK,CAAC;KACd;IACD,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC;AACnE,CAAC;AALD,kCAKC;AAED,MAAsB,UAAU;IAM9B,YAAY,MAAoB;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAE3B,CAAC;IACD,OAAO;QACL,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;IAC/B,CAAC;IACD,gBAAgB;QACd,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;IACrC,CAAC;IACD,OAAO,CAAC,GAAW;QACjB,GAAG,GAAG,GAAG,CAAC,iBAAiB,EAAE,CAAC;QAC9B,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC/B,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAErD,IAAI,GAAG,CAAC,KAAK,CAAC,yBAAyB,CAAC,EAAE;YACxC,OAAO,IAAI,CAAC;SACb;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU,CAAC,GAAW;QACpB,IAAI;YACF,OAAO,IAAA,6BAAU,EAAC,GAAG,CAAC,CAAC;SACxB;QAAC,OAAO,CAAC,EAAE;SAEX;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,oBAAoB,CAAC,GAAY;QAC/B,IAAI,GAAG,EAAE;YACP,MAAM,KAAK,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;YACjD,OAAO,EAAE,KAAK,EAAE,CAAC;SAClB;QACD,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;IACjC,CAAC;IAED,wBAAwB,CAAC,GAAW;QAClC,MAAM,GAAG,GAAwB,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,gDAAgD,CAAC;QAClE,IAAI,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/B,OAAO,KAAK,IAAI,IAAI,EAAE;YAMpB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACrD,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;SAC5B;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAEK,IAAI,CAAU,CAAmB;;YACrC,IAAI,EAAE,GAAG,IAAI,CAAC;YACd,IAAI,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,MAAS,CAAC;YACd,IAAI,OAAO,EAAE;gBACX,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,OAAO;iBACR,CAAC;aACH;YACD,IAAI;gBACF,MAAM,GAAG,MAAM,CAAC,EAAE,CAAC;aACpB;YAAC,OAAO,CAAC,EAAE;gBACV,EAAE,GAAG,KAAK,CAAC;gBACX,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;aACrB;oBAAS;gBACR,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;aACzB;YACD,OAAO;gBACL,EAAE;gBACF,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;KAAA;IAkCK,kBAAkB;;YACtB,IAAI,CAAC,YAAY,GAAG,MAAM,IAAA,yBAAO,EAAC,EAAE,SAAS,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;YAEjE,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;oBACjD,SAAS,EAAE,WAAW;oBACtB,SAAS,EAAE,IAAI,CAAC,YAAY;iBAC7B,CAAC,CAAC;gBACH,IAAI,OAAO,CAAC,UAAU,KAAK,YAAY,EAAE;oBACvC,OAAO,CAAC,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;iBACtE;gBAED,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,OAAO,EAAE,UAAU,GAAU;oBACnD,IAAI,GAAG,EAAE;wBACP,MAAM,CAAC,GAAG,CAAC,CAAC;qBACb;yBAAM;wBACL,OAAO,CAAC,EAAE,CAAC,CAAC;qBACb;gBACH,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,UAAU,GAAU;oBAC7C,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;gBAChD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,OAAO;;YACX,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI;gBACF,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,IAAI,IAAI,CAAC,MAAM,EAAE;oBACf,IAAI,IAAI,CAAC,UAAU,EAAE,EAAE;wBACrB,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;qBACjC;oBACD,WAAW,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;iBACvC;qBAAM;oBACL,WAAW,GAAG,gCAAgC,CAAC;iBAChD;aACF;YAAC,OAAO,CAAC,EAAE;gBACV,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC;aACzB;YACD,IAAI,CAAC,WAAW,GAAG,WAAW,KAAK,EAAE,CAAC;YACtC,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAEK,UAAU;;YACd,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI;gBACF,IAAI,IAAI,CAAC,MAAM,EAAE;oBACf,IAAI,IAAI,CAAC,WAAW,EAAE;wBACpB,WAAW,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;qBACrC;yBAAM;qBAEN;iBACF;qBAAM;oBACL,WAAW,GAAG,gCAAgC,CAAC;iBAChD;aACF;YAAC,OAAO,CAAC,EAAE;gBACV,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC;aACzB;oBAAS;gBACR,IAAI,IAAI,CAAC,SAAS,EAAE;oBAClB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;oBACvB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;iBAC5B;gBACD,IAAI,CAAC,cAAc,EAAE,CAAC;aACvB;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAOD,aAAa,CAAC,OAAe,EAAE,WAAgB;QAC7C,OAAO,IAAI,iBAAO,CAChB,OAAO,EACP,WAAW,CAAC,IAAI,EAChB,WAAW,CAAC,KAAK,EACjB,WAAW,CAAC,UAAU,EACtB,WAAW,CAAC,QAAQ,CACrB,CAAC;IACJ,CAAC;CACF;AAjND,gCAiNC"}
|
|
@@ -1,16 +1,16 @@
|
|
|
1
1
|
import { RequestSqlOptions } from './BaseDriver';
|
|
2
|
-
import {
|
|
2
|
+
import { DbColumn, DbConnection, DbSchema, DbTable, RdhKey, RdsDatabase, ResultSetDataHolder, SchemaAndTableHints, TableRows } from '../resource';
|
|
3
3
|
import { RDSBaseDriver } from './RDSBaseDriver';
|
|
4
4
|
export declare class MySQLDriver extends RDSBaseDriver {
|
|
5
5
|
private client;
|
|
6
6
|
constructor(conRes: DbConnection);
|
|
7
|
-
fieldInfo2Key(fieldInfo: any,
|
|
7
|
+
fieldInfo2Key(fieldInfo: any, table?: DbTable): RdhKey;
|
|
8
8
|
connectSub(): Promise<string>;
|
|
9
9
|
protected getTestSqlStatement(): string;
|
|
10
10
|
requestSql(sql: string, options?: RequestSqlOptions): Promise<ResultSetDataHolder>;
|
|
11
11
|
countTables(tables: SchemaAndTableHints, options: any): Promise<TableRows[]>;
|
|
12
|
-
|
|
13
|
-
getSchemas(dbDatabase:
|
|
12
|
+
getInfomationSchemasSub(): Promise<Array<RdsDatabase>>;
|
|
13
|
+
getSchemas(dbDatabase: RdsDatabase): Promise<Array<DbSchema>>;
|
|
14
14
|
getTables(dbSchema: DbSchema): Promise<Array<DbTable>>;
|
|
15
15
|
getColumns(schemaName: string, dbTable: DbTable): Promise<Array<DbColumn>>;
|
|
16
16
|
closeSub(): Promise<string>;
|
|
@@ -12,12 +12,15 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
12
12
|
constructor(conRes) {
|
|
13
13
|
super(conRes);
|
|
14
14
|
}
|
|
15
|
-
fieldInfo2Key(fieldInfo,
|
|
16
|
-
|
|
17
|
-
resolver.hints.list.push({ table: fieldInfo.orgTable });
|
|
18
|
-
}
|
|
15
|
+
fieldInfo2Key(fieldInfo, table) {
|
|
16
|
+
var _a, _b;
|
|
19
17
|
const name = enum_values_1.EnumValues.getNameFromValue(MySQLColumnType_1.MySQLColumnType, MySQLColumnType_1.MySQLColumnType.parseByFieldInfo(fieldInfo));
|
|
20
|
-
|
|
18
|
+
let comment = '';
|
|
19
|
+
if (table) {
|
|
20
|
+
comment =
|
|
21
|
+
(_b = (_a = table.children.find((it) => it.name === fieldInfo.name)) === null || _a === void 0 ? void 0 : _a.comment) !== null && _b !== void 0 ? _b : '';
|
|
22
|
+
}
|
|
23
|
+
const key = new resource_1.RdhKey(fieldInfo.name, types_1.GeneralColumnType.parse(name), comment);
|
|
21
24
|
if (key.type === types_1.GeneralColumnType.UNKNOWN) {
|
|
22
25
|
}
|
|
23
26
|
return key;
|
|
@@ -50,15 +53,14 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
50
53
|
requestSql(sql, options) {
|
|
51
54
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
52
55
|
let rdh = new resource_1.ResultSetDataHolder([]);
|
|
56
|
+
const ast = this.parseQuery(sql);
|
|
57
|
+
const astTableName = this.getTableName(ast);
|
|
58
|
+
const dbTable = this.getDbTable(astTableName);
|
|
53
59
|
if (this.client) {
|
|
54
60
|
let binds = [];
|
|
55
61
|
if (options && options.binds) {
|
|
56
62
|
binds = options.binds;
|
|
57
63
|
}
|
|
58
|
-
let resolver;
|
|
59
|
-
if (options && options.needsColumnResolve === true) {
|
|
60
|
-
resolver = this.createColumnResolver(sql);
|
|
61
|
-
}
|
|
62
64
|
const [rows, fields] = yield this.client.execute(sql, binds);
|
|
63
65
|
if (fields === undefined) {
|
|
64
66
|
const results = rows;
|
|
@@ -82,7 +84,7 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
82
84
|
else {
|
|
83
85
|
rdh = new resource_1.ResultSetDataHolder(fields === undefined
|
|
84
86
|
? []
|
|
85
|
-
: fields.map((f) => this.fieldInfo2Key(f,
|
|
87
|
+
: fields.map((f) => this.fieldInfo2Key(f, dbTable)));
|
|
86
88
|
rows.forEach((result) => {
|
|
87
89
|
rdh.addRow(result);
|
|
88
90
|
});
|
|
@@ -91,7 +93,7 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
91
93
|
else {
|
|
92
94
|
new Error('No connection');
|
|
93
95
|
}
|
|
94
|
-
|
|
96
|
+
this.setRdhMetaAndStatement(rdh, sql, astTableName, dbTable, options);
|
|
95
97
|
return rdh;
|
|
96
98
|
});
|
|
97
99
|
}
|
|
@@ -126,13 +128,10 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
126
128
|
return list;
|
|
127
129
|
});
|
|
128
130
|
}
|
|
129
|
-
|
|
131
|
+
getInfomationSchemasSub() {
|
|
130
132
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
131
|
-
if (!this.conRes) {
|
|
132
|
-
return [];
|
|
133
|
-
}
|
|
134
133
|
const dbResources = new Array();
|
|
135
|
-
const dbDatabase = new resource_1.
|
|
134
|
+
const dbDatabase = new resource_1.RdsDatabase(this.conRes.database);
|
|
136
135
|
dbResources.push(dbDatabase);
|
|
137
136
|
const dbSchemas = yield this.getSchemas(dbDatabase);
|
|
138
137
|
dbSchemas.forEach((res) => {
|
|
@@ -144,8 +143,8 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
144
143
|
dbTables.forEach((res) => dbSchema.addChild(res));
|
|
145
144
|
}
|
|
146
145
|
for (const dbSchema of dbSchemas) {
|
|
147
|
-
for (const dbTable of dbSchema.
|
|
148
|
-
const dbColumns = yield this.getColumns(dbSchema.
|
|
146
|
+
for (const dbTable of dbSchema.children) {
|
|
147
|
+
const dbColumns = yield this.getColumns(dbSchema.name, dbTable);
|
|
149
148
|
dbColumns.forEach((res) => dbTable.addChild(res));
|
|
150
149
|
}
|
|
151
150
|
}
|
|
@@ -173,7 +172,7 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
173
172
|
ELSE 'TABLE' END AS table_type,
|
|
174
173
|
TABLE_COMMENT as comment
|
|
175
174
|
FROM INFORMATION_SCHEMA.TABLES
|
|
176
|
-
WHERE TABLE_SCHEMA = '${dbSchema.
|
|
175
|
+
WHERE TABLE_SCHEMA = '${dbSchema.name}' `);
|
|
177
176
|
return rdh.rows.map((r) => {
|
|
178
177
|
const res = new resource_1.DbTable(r.values.name, r.values.table_type, r.values.comment);
|
|
179
178
|
return res;
|
|
@@ -182,7 +181,7 @@ class MySQLDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
182
181
|
}
|
|
183
182
|
getColumns(schemaName, dbTable) {
|
|
184
183
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
185
|
-
const binds = [schemaName, dbTable.
|
|
184
|
+
const binds = [schemaName, dbTable.name];
|
|
186
185
|
const rdh = yield this.requestSql(`SELECT COLUMN_NAME as name,
|
|
187
186
|
DATA_TYPE as col_type,
|
|
188
187
|
CASE WHEN IS_NULLABLE = 'YES' THEN 1 ELSE 0 END as nullable,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"MySQLDriver.js","sourceRoot":"","sources":["../../../src/drivers/MySQLDriver.ts"],"names":[],"mappings":";;;;AAAA,8DAAwC;AACxC,6CAAyC;AAEzC,8DAA2D;AAE3D,
|
|
1
|
+
{"version":3,"file":"MySQLDriver.js","sourceRoot":"","sources":["../../../src/drivers/MySQLDriver.ts"],"names":[],"mappings":";;;;AAAA,8DAAwC;AACxC,6CAAyC;AAEzC,8DAA2D;AAE3D,0CAUqB;AACrB,oCAA6C;AAC7C,mDAAgD;AAEhD,MAAa,WAAY,SAAQ,6BAAa;IAG5C,YAAY,MAAoB;QAC9B,KAAK,CAAC,MAAM,CAAC,CAAC;IAChB,CAAC;IAED,aAAa,CAAC,SAAS,EAAE,KAAe;;QACtC,MAAM,IAAI,GAAG,wBAAU,CAAC,gBAAgB,CACtC,iCAAe,EACf,iCAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAC5C,CAAC;QACF,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,EAAE;YACT,OAAO;gBACL,MAAA,MAAA,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,0CAAE,OAAO,mCAAI,EAAE,CAAC;SAC1E;QACD,MAAM,GAAG,GAAG,IAAI,iBAAM,CACpB,SAAS,CAAC,IAAI,EACd,yBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC7B,OAAO,CACR,CAAC;QACF,IAAI,GAAG,CAAC,IAAI,KAAK,yBAAiB,CAAC,OAAO,EAAE;SAE3C;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAEK,UAAU;;YACd,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG;gBACd,eAAe,EAAE,CAAC;gBAClB,kBAAkB,EAAE,IAAI;gBACxB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;aAC/B,CAAC;YACF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YACxC,IAAI;gBACF,YAAY,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;aAClC;YAAC,OAAO,CAAC,EAAE;gBACV,YAAY,GAAG,CAAC,CAAC,OAAO,CAAC;aAC1B;YACD,OAAO,YAAY,CAAC;QACtB,CAAC;KAAA;IAES,mBAAmB;QAC3B,OAAO,UAAU,CAAC;IACpB,CAAC;IAGK,UAAU,CACd,GAAW,EACX,OAA2B;;YAE3B,IAAI,GAAG,GAAG,IAAI,8BAAmB,CAAC,EAAE,CAAC,CAAC;YAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YAE9C,IAAI,IAAI,CAAC,MAAM,EAAE;gBACf,IAAI,KAAK,GAAa,EAAE,CAAC;gBACzB,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK,EAAE;oBAC5B,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;iBACvB;gBAED,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAE7D,IAAI,MAAM,KAAK,SAAS,EAAE;oBAWxB,MAAM,OAAO,GAAG,IAAuB,CAAC;oBAExC,GAAG,GAAG,IAAI,8BAAmB,CAAC;wBAC5B,YAAY;wBACZ,cAAc;wBACd,UAAU;wBACV,cAAc;wBACd,eAAe;wBACf,aAAa;qBACd,CAAC,CAAC;oBACH,GAAG,CAAC,MAAM,CAAC;wBACT,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,YAAY,EAAE,OAAO,CAAC,YAAY;wBAClC,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,YAAY,EAAE,OAAO,CAAC,YAAY;wBAClC,aAAa,EAAE,OAAO,CAAC,aAAa;wBACpC,WAAW,EAAE,OAAO,CAAC,WAAW;qBACjC,CAAC,CAAC;iBACJ;qBAAM;oBACL,GAAG,GAAG,IAAI,8BAAmB,CAC3B,MAAM,KAAK,SAAS;wBAClB,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CACtD,CAAC;oBACD,IAAY,CAAC,OAAO,CAAC,CAAC,MAAW,EAAE,EAAE;wBACpC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACrB,CAAC,CAAC,CAAC;iBACJ;aACF;iBAAM;gBACL,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;aAC5B;YACD,IAAI,CAAC,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAEtE,OAAO,GAAG,CAAC;QACb,CAAC;KAAA;IAEK,WAAW,CACf,MAA2B,EAC3B,OAAY;;YAEZ,MAAM,IAAI,GAAG,IAAI,KAAK,EAAa,CAAC;YACpC,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE;gBAC5B,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,IAAI,EAAE,CAAC,MAAM,EAAE;oBACb,MAAM,GAAG,EAAE,CAAC,MAAM,GAAG,GAAG,CAAC;iBAC1B;gBACD,IAAI,OAAO,IAAI,OAAO,CAAC,iBAAiB,EAAE;oBACxC,IAAI,OAAO,GAAG,CAAC,KAAK,CAAC,EAAE;wBACrB,OAAO,CAAC,iBAAiB,CAAC,SAAS,MAAM,GAAG,EAAE,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;qBAC7D;iBACF;gBACD,MAAM,GAAG,GAAG,iCAAiC,MAAM,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;gBACjE,IAAI;oBACF,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACnD,IAAI,OAAO,IAAK,OAAe,CAAC,MAAM,GAAG,CAAC,EAAE;wBAC1C,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;wBACvB,MAAM,GAAG,GAAc,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC/D,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;qBAChB;iBACF;gBAAC,OAAO,CAAC,EAAE;oBACV,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;iBAClB;gBACD,OAAO,EAAE,CAAC;aACX;YACD,OAAO,IAAI,CAAC;QACd,CAAC;KAAA;IAEK,uBAAuB;;YAC3B,MAAM,WAAW,GAAG,IAAI,KAAK,EAAe,CAAC;YAC7C,MAAM,UAAU,GAAG,IAAI,sBAAW,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACzD,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAE7B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YACpD,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;gBAChC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBAChD,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;aACnD;YACD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;gBAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE;oBACvC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAChE,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;iBACnD;aACF;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAGK,UAAU,CAAC,UAAuB;;YACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;;;oBAGlB,CAAC,CAAC;YAElB,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,GAAG,GAAG,IAAI,mBAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACxC,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,SAAS,CAAC,QAAkB;;YAChC,MAAM,GAAG,GAAG,MAAM,IAAI;iBACnB,UAAU,CAAC;;;;;;kCAMgB,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC;YAEjD,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,GAAG,GAAG,IAAI,kBAAO,CACrB,CAAC,CAAC,MAAM,CAAC,IAAI,EACb,CAAC,CAAC,MAAM,CAAC,UAAU,EACnB,CAAC,CAAC,MAAM,CAAC,OAAO,CACjB,CAAC;gBACF,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,UAAU,CACd,UAAkB,EAClB,OAAgB;;YAEhB,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAC/B;;;;;;;;uDAQiD,EACjD,EAAE,KAAK,EAAE,CACV,CAAC;YAEF,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,SAAS,GAAG,wBAAU,CAAC,gBAAgB,CAC3C,iCAAe,EACf,iCAAe,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CACzC,CAAC;gBACF,MAAM,GAAG,GAAG,IAAI,mBAAQ,CACtB,CAAC,CAAC,MAAM,CAAC,IAAI,EACb,yBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC,EAClC;oBACE,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC;oBACjC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO;oBACrB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,WAAW;oBAC7B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS;iBAC1B,EACD,CAAC,CAAC,MAAM,CAAC,OAAO,CACjB,CAAC;gBACF,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,QAAQ;;YACZ,IAAI;gBACF,IAAI,IAAI,CAAC,MAAM,EAAE;oBACf,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;iBACzB;gBACD,OAAO,EAAE,CAAC;aACX;YAAC,OAAO,CAAC,EAAE;gBACV,OAAO,CAAC,CAAC,OAAO,CAAC;aAClB;QACH,CAAC;KAAA;CACF;AAhQD,kCAgQC"}
|
|
@@ -1,18 +1,18 @@
|
|
|
1
1
|
import { default as pg } from 'pg';
|
|
2
2
|
import { RequestSqlOptions } from './BaseDriver';
|
|
3
|
-
import {
|
|
3
|
+
import { DbColumn, DbConnection, DbSchema, DbTable, RdhKey, RdsDatabase, ResultSetDataHolder, SchemaAndTableHints, TableRows } from '../resource';
|
|
4
4
|
import { RDSBaseDriver } from './RDSBaseDriver';
|
|
5
5
|
export declare class PostgresDriver extends RDSBaseDriver {
|
|
6
6
|
private pool;
|
|
7
7
|
constructor(conRes: DbConnection);
|
|
8
|
-
fieldInfo2Key(fieldInfo: pg.FieldDef,
|
|
8
|
+
fieldInfo2Key(fieldInfo: pg.FieldDef, table?: DbTable): RdhKey;
|
|
9
9
|
connectSub(): Promise<string>;
|
|
10
10
|
protected getTestSqlStatement(): string;
|
|
11
11
|
requestSql(sql: string, options?: RequestSqlOptions): Promise<ResultSetDataHolder>;
|
|
12
12
|
countTables(tables: SchemaAndTableHints, options: any): Promise<TableRows[]>;
|
|
13
|
-
|
|
14
|
-
asyncGetDatabases(connection_database: string): Promise<Array<
|
|
15
|
-
getSchemas(dbDatabase:
|
|
13
|
+
getInfomationSchemasSub(): Promise<Array<RdsDatabase>>;
|
|
14
|
+
asyncGetDatabases(connection_database: string): Promise<Array<RdsDatabase>>;
|
|
15
|
+
getSchemas(dbDatabase: RdsDatabase): Promise<Array<DbSchema>>;
|
|
16
16
|
getTables(dbSchema: DbSchema): Promise<Array<DbTable>>;
|
|
17
17
|
getColumns(schemaName: string, dbTable: DbTable): Promise<Array<DbColumn>>;
|
|
18
18
|
closeSub(): Promise<string>;
|
|
@@ -12,12 +12,18 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
12
12
|
constructor(conRes) {
|
|
13
13
|
super(conRes);
|
|
14
14
|
}
|
|
15
|
-
fieldInfo2Key(fieldInfo,
|
|
15
|
+
fieldInfo2Key(fieldInfo, table) {
|
|
16
|
+
var _a, _b;
|
|
16
17
|
if (fieldInfo.name.startsWith('c_')) {
|
|
17
18
|
console.log(`★ ${fieldInfo.name.substring(2).toUpperCase()} = ${fieldInfo.dataTypeID}`);
|
|
18
19
|
}
|
|
19
20
|
const name = enum_values_1.EnumValues.getNameFromValue(PostgresColumnType_1.PostgresColumnType, PostgresColumnType_1.PostgresColumnType.parse(fieldInfo.dataTypeID));
|
|
20
|
-
|
|
21
|
+
let comment = '';
|
|
22
|
+
if (table) {
|
|
23
|
+
comment =
|
|
24
|
+
(_b = (_a = table.children.find((it) => it.name === fieldInfo.name)) === null || _a === void 0 ? void 0 : _a.comment) !== null && _b !== void 0 ? _b : '';
|
|
25
|
+
}
|
|
26
|
+
const key = new resource_1.RdhKey(fieldInfo.name, types_1.GeneralColumnType.parse(name), comment);
|
|
21
27
|
return key;
|
|
22
28
|
}
|
|
23
29
|
connectSub() {
|
|
@@ -48,27 +54,26 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
48
54
|
requestSql(sql, options) {
|
|
49
55
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
50
56
|
let rdh = new resource_1.ResultSetDataHolder([]);
|
|
57
|
+
const ast = this.parseQuery(sql);
|
|
58
|
+
const astTableName = this.getTableName(ast);
|
|
59
|
+
const dbTable = this.getDbTable(astTableName);
|
|
51
60
|
let binds = [];
|
|
52
61
|
if (options && options.binds) {
|
|
53
62
|
binds = options.binds;
|
|
54
63
|
}
|
|
55
64
|
const results = yield this.pool.query(sql, binds);
|
|
56
65
|
if (results) {
|
|
57
|
-
let resolver;
|
|
58
|
-
if (options && options.needsColumnResolve === true) {
|
|
59
|
-
resolver = this.createColumnResolver(sql);
|
|
60
|
-
}
|
|
61
66
|
const fields = results.fields;
|
|
62
67
|
rdh = new resource_1.ResultSetDataHolder(fields === undefined
|
|
63
68
|
? []
|
|
64
|
-
: fields.map((f) => this.fieldInfo2Key(f,
|
|
69
|
+
: fields.map((f) => this.fieldInfo2Key(f, dbTable)));
|
|
65
70
|
if (results.rows) {
|
|
66
71
|
results.rows.forEach((result) => {
|
|
67
72
|
rdh.addRow(result);
|
|
68
73
|
});
|
|
69
74
|
}
|
|
70
75
|
}
|
|
71
|
-
|
|
76
|
+
this.setRdhMetaAndStatement(rdh, sql, astTableName, dbTable, options);
|
|
72
77
|
return rdh;
|
|
73
78
|
});
|
|
74
79
|
}
|
|
@@ -101,11 +106,8 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
101
106
|
return list;
|
|
102
107
|
});
|
|
103
108
|
}
|
|
104
|
-
|
|
109
|
+
getInfomationSchemasSub() {
|
|
105
110
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
106
|
-
if (!this.conRes) {
|
|
107
|
-
return [];
|
|
108
|
-
}
|
|
109
111
|
const dbResources = new Array();
|
|
110
112
|
const db_list = yield this.asyncGetDatabases(this.conRes.database);
|
|
111
113
|
db_list.forEach((db) => dbResources.push(db));
|
|
@@ -120,8 +122,8 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
120
122
|
dbTables.forEach((res) => dbSchema.addChild(res));
|
|
121
123
|
}
|
|
122
124
|
for (const dbSchema of dbSchemas) {
|
|
123
|
-
for (const dbTable of dbSchema.
|
|
124
|
-
const dbColumns = yield this.getColumns(dbSchema.
|
|
125
|
+
for (const dbTable of dbSchema.children) {
|
|
126
|
+
const dbColumns = yield this.getColumns(dbSchema.name, dbTable);
|
|
125
127
|
dbColumns.forEach((res) => dbTable.addChild(res));
|
|
126
128
|
}
|
|
127
129
|
}
|
|
@@ -134,9 +136,8 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
134
136
|
.requestSql(`SELECT datname AS name, pg_encoding_to_char(encoding) AS comment
|
|
135
137
|
FROM pg_database ORDER BY datname`);
|
|
136
138
|
return rdh.rows.map((r) => {
|
|
137
|
-
const res = new resource_1.
|
|
139
|
+
const res = new resource_1.RdsDatabase(r.values.name);
|
|
138
140
|
res.comment = r.values.comment;
|
|
139
|
-
res.disabled = res.name !== connection_database;
|
|
140
141
|
return res;
|
|
141
142
|
});
|
|
142
143
|
});
|
|
@@ -159,7 +160,7 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
159
160
|
.requestSql(`select quote_ident(m.relname) as qname, COALESCE(d.description, '') as comment
|
|
160
161
|
from pg_stat_all_tables as m
|
|
161
162
|
LEFT JOIN pg_description as d ON (m.relid = d.objoid AND d.objsubid=0)
|
|
162
|
-
WHERE m.schemaname='${dbSchema.
|
|
163
|
+
WHERE m.schemaname='${dbSchema.name}'
|
|
163
164
|
ORDER BY m.relname`);
|
|
164
165
|
const list = rdh.rows.map((r) => {
|
|
165
166
|
const res = new resource_1.DbTable(r.values.qname, 'TABLE', r.values.comment);
|
|
@@ -167,7 +168,7 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
167
168
|
});
|
|
168
169
|
rdh = yield this
|
|
169
170
|
.requestSql(`select quote_ident(viewname) as qname, definition from pg_catalog.pg_views
|
|
170
|
-
where schemaname = '${dbSchema.
|
|
171
|
+
where schemaname = '${dbSchema.name}'
|
|
171
172
|
order by viewname`);
|
|
172
173
|
return list.concat(rdh.rows.map((r) => {
|
|
173
174
|
const res = new resource_1.DbTable(r.values.qname, 'VIEW', '');
|
|
@@ -177,7 +178,7 @@ class PostgresDriver extends RDSBaseDriver_1.RDSBaseDriver {
|
|
|
177
178
|
}
|
|
178
179
|
getColumns(schemaName, dbTable) {
|
|
179
180
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
180
|
-
const binds = [schemaName, dbTable.
|
|
181
|
+
const binds = [schemaName, dbTable.name];
|
|
181
182
|
const rdh = yield this.requestSql(`select
|
|
182
183
|
col.COLUMN_NAME as name,
|
|
183
184
|
quote_ident(col.COLUMN_NAME) as qname,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"PostgresDriver.js","sourceRoot":"","sources":["../../../src/drivers/PostgresDriver.ts"],"names":[],"mappings":";;;;AAEA,oDAAmC;AACnC,6CAAyC;AACzC,oEAAiE;AAEjE,
|
|
1
|
+
{"version":3,"file":"PostgresDriver.js","sourceRoot":"","sources":["../../../src/drivers/PostgresDriver.ts"],"names":[],"mappings":";;;;AAEA,oDAAmC;AACnC,6CAAyC;AACzC,oEAAiE;AAEjE,0CAUqB;AACrB,oCAA6C;AAC7C,mDAAgD;AAEhD,MAAa,cAAe,SAAQ,6BAAa;IAG/C,YAAY,MAAoB;QAC9B,KAAK,CAAC,MAAM,CAAC,CAAC;IAChB,CAAC;IASD,aAAa,CAAC,SAAsB,EAAE,KAAe;;QACnD,IAAI,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE;YACnC,OAAO,CAAC,GAAG,CACT,KAAK,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,MAC5C,SAAS,CAAC,UACZ,EAAE,CACH,CAAC;SACH;QACD,MAAM,IAAI,GAAG,wBAAU,CAAC,gBAAgB,CACtC,uCAAkB,EAClB,uCAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,CAC/C,CAAC;QACF,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,EAAE;YACT,OAAO;gBACL,MAAA,MAAA,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,0CAAE,OAAO,mCAAI,EAAE,CAAC;SAC1E;QACD,MAAM,GAAG,GAAG,IAAI,iBAAM,CACpB,SAAS,CAAC,IAAI,EACd,yBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,EAC7B,OAAO,CACR,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEK,UAAU;;YACd,IAAI,WAAW,GAAG,EAAE,CAAC;YAErB,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAC3B;gBACE,IAAI,EAAE,IAAI;gBACV,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,UAAU;aACrB,EACD;gBACE,GAAG,EAAE,CAAC;gBACN,iBAAiB,EAAE,IAAI;gBACvB,uBAAuB,EAAE,IAAI;gBAC7B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;aAC/B,CACF,CAAC;YAEF,IAAI,CAAC,IAAI,GAAG,IAAI,YAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAYjC,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAEhC,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAES,mBAAmB;QAC3B,OAAO,cAAc,CAAC;IACxB,CAAC;IAGK,UAAU,CACd,GAAW,EACX,OAA2B;;YAG3B,IAAI,GAAG,GAAG,IAAI,8BAAmB,CAAC,EAAE,CAAC,CAAC;YAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YAE9C,IAAI,KAAK,GAAa,EAAE,CAAC;YACzB,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK,EAAE;gBAC5B,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;aACvB;YACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAYlD,IAAI,OAAO,EAAE;gBACX,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC9B,GAAG,GAAG,IAAI,8BAAmB,CAC3B,MAAM,KAAK,SAAS;oBAClB,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CACtD,CAAC;gBACF,IAAI,OAAO,CAAC,IAAI,EAAE;oBAChB,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAW,EAAE,EAAE;wBACnC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACrB,CAAC,CAAC,CAAC;iBACJ;aACF;YAED,IAAI,CAAC,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAEtE,OAAO,GAAG,CAAC;QACb,CAAC;KAAA;IAEK,WAAW,CACf,MAA2B,EAC3B,OAAY;;YAEZ,MAAM,IAAI,GAAG,IAAI,KAAK,EAAa,CAAC;YACpC,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE;gBAC5B,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,IAAI,EAAE,CAAC,MAAM,EAAE;oBACb,MAAM,GAAG,EAAE,CAAC,MAAM,GAAG,GAAG,CAAC;iBAC1B;gBACD,IAAI,OAAO,IAAI,OAAO,CAAC,iBAAiB,EAAE;oBACxC,IAAI,OAAO,GAAG,CAAC,KAAK,CAAC,EAAE;wBACrB,OAAO,CAAC,iBAAiB,CAAC,SAAS,MAAM,GAAG,EAAE,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;qBAC7D;iBACF;gBACD,MAAM,GAAG,GAAG,iCAAiC,MAAM,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;gBACjE,IAAI;oBACF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC/C,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;wBACtD,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;wBAC5B,MAAM,GAAG,GAAc,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC/D,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;qBAChB;iBAEF;gBAAC,OAAO,CAAC,EAAE,GAAE;gBACd,OAAO,EAAE,CAAC;aACX;YACD,OAAO,IAAI,CAAC;QACd,CAAC;KAAA;IAEK,uBAAuB;;YAC3B,MAAM,WAAW,GAAG,IAAI,KAAK,EAAe,CAAC;YAC7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACnE,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAExE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YACpD,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;gBAChC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBAChD,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;aACnD;YACD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;gBAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE;oBACvC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAChE,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;iBACnD;aACF;YAED,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAEK,iBAAiB,CACrB,mBAA2B;;YAE3B,MAAM,GAAG,GAAG,MAAM,IAAI;iBACnB,UAAU,CAAC;wCACsB,CAAC,CAAC;YACtC,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,GAAG,GAAG,IAAI,sBAAW,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3C,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC/B,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,UAAU,CAAC,UAAuB;;YACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;;;oBAGlB,CAAC,CAAC;YAElB,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,GAAG,GAAG,IAAI,mBAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACxC,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,SAAS,CAAC,QAAkB;;YAChC,IAAI,GAAG,GAAG,MAAM,IAAI;iBACjB,UAAU,CAAC;;;4BAGU,QAAQ,CAAC,IAAI;yBAChB,CAAC,CAAC;YAEvB,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC9B,MAAM,GAAG,GAAG,IAAI,kBAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACnE,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YAEH,GAAG,GAAG,MAAM,IAAI;iBACb,UAAU,CAAC;4BACU,QAAQ,CAAC,IAAI;wBACjB,CAAC,CAAC;YAEtB,OAAO,IAAI,CAAC,MAAM,CAChB,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACjB,MAAM,GAAG,GAAG,IAAI,kBAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBACpD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;KAAA;IAEK,UAAU,CACd,UAAkB,EAClB,OAAgB;;YAEhB,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAC/B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAoDqB,EACrB,EAAE,KAAK,EAAE,CACV,CAAC;YAEF,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACxB,MAAM,SAAS,GAAG,wBAAU,CAAC,gBAAgB,CAC3C,uCAAkB,EAClB,uCAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAC5C,CAAC;gBACF,MAAM,GAAG,GAAG,IAAI,mBAAQ,CACtB,CAAC,CAAC,MAAM,CAAC,KAAK,EACd,yBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC,EAClC;oBACE,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC;oBACjC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO;oBACrB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,WAAW;oBAC7B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS;iBAC1B,EACD,CAAC,CAAC,MAAM,CAAC,OAAO,CACjB,CAAC;gBACF,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,QAAQ;;YACZ,IAAI;gBACF,IAAI,IAAI,CAAC,IAAI,EAAE;oBACb,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;iBACvB;gBACD,OAAO,EAAE,CAAC;aACX;YAAC,OAAO,CAAC,EAAE;gBACV,OAAO,CAAC,CAAC,OAAO,CAAC;aAClB;QACH,CAAC;KAAA;CACF;AAxUD,wCAwUC"}
|
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
import { BaseDriver, RequestSqlOptions } from './BaseDriver';
|
|
2
|
-
import { DbConnection, ResultSetDataHolder, SchemaAndTableHints, TableRows } from '../resource';
|
|
2
|
+
import { DbConnection, DbTable, RdsDatabase, ResultSetDataHolder, SchemaAndTableHints, TableRows } from '../resource';
|
|
3
|
+
import { Statement } from 'pgsql-ast-parser';
|
|
4
|
+
import { CompareKey } from '../types';
|
|
3
5
|
export declare abstract class RDSBaseDriver extends BaseDriver {
|
|
4
6
|
constructor(conRes: DbConnection);
|
|
5
7
|
protected abstract getTestSqlStatement(): string;
|
|
@@ -7,7 +9,15 @@ export declare abstract class RDSBaseDriver extends BaseDriver {
|
|
|
7
9
|
viewData(tableName: string, options?: {
|
|
8
10
|
schemaName?: string;
|
|
9
11
|
columnNames?: string[];
|
|
12
|
+
maxRows?: number;
|
|
13
|
+
compareKeys?: CompareKey[];
|
|
10
14
|
}): Promise<ResultSetDataHolder>;
|
|
15
|
+
getInfomationSchemas(): Promise<Array<RdsDatabase>>;
|
|
16
|
+
abstract getInfomationSchemasSub(): Promise<Array<RdsDatabase>>;
|
|
17
|
+
getTableName(ast: Statement): string | undefined;
|
|
18
|
+
getDbTable(name: string): DbTable | undefined;
|
|
19
|
+
setRdhMetaAndStatement(rdh: ResultSetDataHolder, sql: string, astTableName?: string, dbTable?: DbTable, options?: RequestSqlOptions): void;
|
|
20
|
+
resetDefaultSchema(database: RdsDatabase): void;
|
|
11
21
|
abstract requestSql(sql: string, options?: RequestSqlOptions): Promise<ResultSetDataHolder>;
|
|
12
22
|
abstract countTables(tables: SchemaAndTableHints, options: any): Promise<TableRows[]>;
|
|
13
23
|
}
|
|
@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
3
3
|
exports.RDSBaseDriver = void 0;
|
|
4
4
|
const tslib_1 = require("tslib");
|
|
5
5
|
const BaseDriver_1 = require("./BaseDriver");
|
|
6
|
+
const types_1 = require("../types");
|
|
6
7
|
class RDSBaseDriver extends BaseDriver_1.BaseDriver {
|
|
7
8
|
constructor(conRes) {
|
|
8
9
|
super(conRes);
|
|
@@ -42,10 +43,93 @@ class RDSBaseDriver extends BaseDriver_1.BaseDriver {
|
|
|
42
43
|
cols = '*';
|
|
43
44
|
}
|
|
44
45
|
return yield this.requestSql(`SELECT ${cols} FROM ${sp}${tableName} `, {
|
|
45
|
-
|
|
46
|
+
tableName,
|
|
47
|
+
maxRows: options === null || options === void 0 ? void 0 : options.maxRows,
|
|
48
|
+
compareKeys: options === null || options === void 0 ? void 0 : options.compareKeys,
|
|
46
49
|
});
|
|
47
50
|
});
|
|
48
51
|
}
|
|
52
|
+
getInfomationSchemas() {
|
|
53
|
+
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
54
|
+
this.conRes.clearChildren();
|
|
55
|
+
if (!this.conRes) {
|
|
56
|
+
return [];
|
|
57
|
+
}
|
|
58
|
+
const dbResources = yield this.getInfomationSchemasSub();
|
|
59
|
+
dbResources.forEach((it) => this.conRes.addChild(it));
|
|
60
|
+
return dbResources;
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
getTableName(ast) {
|
|
64
|
+
if (ast) {
|
|
65
|
+
switch (ast.type) {
|
|
66
|
+
case 'select':
|
|
67
|
+
if (ast.from && ast.from[0].type === 'table') {
|
|
68
|
+
return ast.from[0].name.name;
|
|
69
|
+
}
|
|
70
|
+
break;
|
|
71
|
+
case 'insert':
|
|
72
|
+
return ast.into.name;
|
|
73
|
+
case 'update':
|
|
74
|
+
return ast.table.name;
|
|
75
|
+
case 'delete':
|
|
76
|
+
return ast.from.name;
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
return undefined;
|
|
80
|
+
}
|
|
81
|
+
getDbTable(name) {
|
|
82
|
+
if (!name) {
|
|
83
|
+
return undefined;
|
|
84
|
+
}
|
|
85
|
+
const list = this.conRes.findChildren({
|
|
86
|
+
keyword: name,
|
|
87
|
+
resouceType: types_1.ResourceType.Table,
|
|
88
|
+
recursively: true,
|
|
89
|
+
});
|
|
90
|
+
if (list) {
|
|
91
|
+
return list[0];
|
|
92
|
+
}
|
|
93
|
+
return undefined;
|
|
94
|
+
}
|
|
95
|
+
setRdhMetaAndStatement(rdh, sql, astTableName, dbTable, options) {
|
|
96
|
+
rdh.setSqlStatement(sql);
|
|
97
|
+
if (!rdh.meta) {
|
|
98
|
+
rdh.meta = {};
|
|
99
|
+
}
|
|
100
|
+
rdh.meta.tableName = options === null || options === void 0 ? void 0 : options.tableName;
|
|
101
|
+
rdh.meta.compareKeys = options === null || options === void 0 ? void 0 : options.compareKeys;
|
|
102
|
+
if (!rdh.meta.tableName) {
|
|
103
|
+
rdh.meta.tableName = astTableName;
|
|
104
|
+
}
|
|
105
|
+
if (!rdh.meta.compareKeys && rdh.meta.tableName) {
|
|
106
|
+
if (dbTable) {
|
|
107
|
+
rdh.meta.compareKeys = dbTable.getCompareKeys();
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
resetDefaultSchema(database) {
|
|
112
|
+
const searchNames = [];
|
|
113
|
+
if (this.conRes.database) {
|
|
114
|
+
searchNames.push(this.conRes.database);
|
|
115
|
+
}
|
|
116
|
+
if (this.conRes.user) {
|
|
117
|
+
searchNames.push(this.conRes.user);
|
|
118
|
+
}
|
|
119
|
+
searchNames.push('public');
|
|
120
|
+
for (const searchName of searchNames) {
|
|
121
|
+
const idx = database.children.findIndex((it) => it.name == searchName);
|
|
122
|
+
if (idx >= 0) {
|
|
123
|
+
database.children[idx].isDefault = true;
|
|
124
|
+
const [defaultSchema] = database.children.splice(idx, 1);
|
|
125
|
+
database.children.unshift(defaultSchema);
|
|
126
|
+
return;
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
if (database.children.length) {
|
|
130
|
+
database.children[0].isDefault = true;
|
|
131
|
+
}
|
|
132
|
+
}
|
|
49
133
|
}
|
|
50
134
|
exports.RDSBaseDriver = RDSBaseDriver;
|
|
51
135
|
//# sourceMappingURL=RDSBaseDriver.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"RDSBaseDriver.js","sourceRoot":"","sources":["../../../src/drivers/RDSBaseDriver.ts"],"names":[],"mappings":";;;;AAAA,6CAA6D;
|
|
1
|
+
{"version":3,"file":"RDSBaseDriver.js","sourceRoot":"","sources":["../../../src/drivers/RDSBaseDriver.ts"],"names":[],"mappings":";;;;AAAA,6CAA6D;AAY7D,oCAAoD;AAEpD,MAAsB,aAAc,SAAQ,uBAAU;IACpD,YAAY,MAAoB;QAC9B,KAAK,CAAC,MAAM,CAAC,CAAC;IAChB,CAAC;IAIK,IAAI,CAAC,YAAY,GAAG,KAAK;;YAC7B,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,YAAY,EAAE;gBAChB,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;aACpC;YACD,IAAI,CAAC,WAAW,EAAE;gBAChB,IAAI;oBACF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;iBACnD;gBAAC,OAAO,CAAC,EAAE;oBACV,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC;iBACzB;gBACD,IAAI,YAAY,EAAE;oBAChB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;iBACzB;aACF;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAEK,QAAQ,CACZ,SAAiB,EACjB,OAKC;;;YAED,IAAI,EAAE,GAAG,EAAE,CAAC;YACZ,IAAI,IAAI,GAAG,EAAE,CAAC;YAEd,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,UAAU,EAAE;gBACvB,EAAE,GAAG,GAAG,OAAO,CAAC,UAAU,GAAG,CAAC;aAC/B;YACD,IAAI,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,KAAI,CAAA,MAAA,OAAO,CAAC,WAAW,0CAAE,MAAM,IAAG,CAAC,EAAE;gBAC3D,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;aACtC;iBAAM;gBACL,IAAI,GAAG,GAAG,CAAC;aACZ;YACD,OAAO,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,IAAI,SAAS,EAAE,GAAG,SAAS,GAAG,EAAE;gBACrE,SAAS;gBACT,OAAO,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,OAAO;gBACzB,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;aAClC,CAAC,CAAC;;KACJ;IAEK,oBAAoB;;YACxB,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;gBAChB,OAAO,EAAE,CAAC;aACX;YACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACzD,WAAW,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACtD,OAAO,WAAW,CAAC;QACrB,CAAC;KAAA;IAID,YAAY,CAAC,GAAc;QACzB,IAAI,GAAG,EAAE;YAEP,QAAQ,GAAG,CAAC,IAAI,EAAE;gBAChB,KAAK,QAAQ;oBACX,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE;wBAC5C,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;qBAC9B;oBACD,MAAM;gBACR,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBACvB,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;gBACxB,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB;SACF;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,IAAI,CAAC,IAAI,EAAE;YACT,OAAO,SAAS,CAAC;SAClB;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAU;YAC7C,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,oBAAY,CAAC,KAAK;YAC/B,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,IAAI,IAAI,EAAE;YACR,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC;SAChB;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,sBAAsB,CACpB,GAAwB,EACxB,GAAW,EACX,YAAqB,EACrB,OAAiB,EACjB,OAA2B;QAE3B,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QACzB,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;YACb,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC;SACf;QACD,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAC;QACxC,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,CAAC;QAC5C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE;YACvB,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,YAAY,CAAC;SACnC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE;YAC/C,IAAI,OAAO,EAAE;gBACX,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;aACjD;SACF;IACH,CAAC;IAED,kBAAkB,CAAC,QAAqB;QACtC,MAAM,WAAW,GAAG,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;YACxB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;SACxC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;YACpB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;SACpC;QACD,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE3B,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,IAAI,UAAU,CAAC,CAAC;YACvE,IAAI,GAAG,IAAI,CAAC,EAAE;gBACZ,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;gBACxC,MAAM,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBACzD,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;gBACzC,OAAO;aACR;SACF;QAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE;YAC5B,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;SACvC;IACH,CAAC;CAYF;AA7JD,sCA6JC"}
|