dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-18 07:56:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

mssql bulk table writer

Browse Source
This commit is contained in:
Jan Prochazka
2020-06-11 13:58:34 +02:00
parent a86f7e96ca
commit 38b6350ef8
12 changed files with 187 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
packages/engines/mssql/MsSqlAnalyser.js
View File

@@ -3,6 +3,7 @@ const _ = require('lodash');
const sql = require('./sql');
const DatabaseAnalyser = require('../default/DatabaseAnalyser');
const { filter } = require('lodash');
function objectTypeToField(type) {
switch (type.trim()) {
@@ -159,10 +160,17 @@ function detectType(col) {
class MsSqlAnalyser extends DatabaseAnalyser {
constructor(pool, driver) {
super(pool, driver);
this.singleObjectId = null;
}
createQuery(resFileName, filterIdObjects) {
let res = sql[resFileName];
if (this.singleObjectFilter) {
const { typeField } = this.singleObjectFilter;
if (!this.singleObjectId) return null;
if (!filterIdObjects.includes(typeField)) return null;
return res.replace('=[OBJECT_ID_CONDITION]', ` = ${this.singleObjectId}`);
}
if (!this.modifications || !filterIdObjects || this.modifications.length == 0) {
res = res.replace('=[OBJECT_ID_CONDITION]', ' is not null');
} else {
@@ -177,7 +185,19 @@ class MsSqlAnalyser extends DatabaseAnalyser {
}
return res;
}
async getSingleObjectId() {
if (this.singleObjectFilter) {
const { name, typeField } = this.singleObjectFilter;
const { schemaName, pureName } = name;
const fullName = schemaName ? `[${schemaName}].[${pureName}]` : pureName;
const resId = await this.driver.query(this.pool, `SELECT OBJECT_ID('${fullName}') AS id`);
this.singleObjectId = resId.rows[0].id;
}
}
async _runAnalysis() {
await this.getSingleObjectId();
const tablesRows = await this.driver.query(this.pool, this.createQuery('tables', ['tables']));
const columnsRows = await this.driver.query(this.pool, this.createQuery('columns', ['tables']));
const pkColumnsRows = await this.driver.query(this.pool, this.createQuery('primaryKeys', ['tables']));