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

native module refactor POC

Browse Source
This commit is contained in:
SPRINX0\prochazka
2024-11-13 09:06:51 +01:00
parent c6dab85fc2
commit 9f0e06e663
4 changed files with 16 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
plugins/dbgate-plugin-mssql/src/backend/driver.js
View File

@@ -5,15 +5,13 @@ const MsSqlAnalyser = require('./MsSqlAnalyser');
const createTediousBulkInsertStream = require('./createTediousBulkInsertStream');
const createNativeBulkInsertStream = require('./createNativeBulkInsertStream');
const AsyncLock = require('async-lock');
const nativeDriver = require('./nativeDriver');
const lock = new AsyncLock();
const { tediousConnect, tediousQueryCore, tediousReadQuery, tediousStream } = require('./tediousDriver');
const { nativeConnect, nativeQueryCore, nativeReadQuery, nativeStream } = nativeDriver;
const { nativeConnect, nativeQueryCore, nativeReadQuery, nativeStream } = require('./nativeDriver');
const { getLogger } = global.DBGATE_PACKAGES['dbgate-tools'];
const logger = getLogger('mssqlDriver');
let requireMsnodesqlv8;
let platformInfo;
let authProxy;
@@ -59,7 +57,7 @@ const driver = {
getAuthTypes() {
const res = [];
if (requireMsnodesqlv8) res.push(...windowsAuthTypes);
if (platformInfo?.isWindows) res.push(...windowsAuthTypes);
if (authProxy.isAuthProxySupported()) {
res.push(
@@ -82,7 +80,7 @@ const driver = {
async connect(conn) {
const { authType } = conn;
const connectionType = requireMsnodesqlv8 && (authType == 'sspi' || authType == 'sql') ? 'msnodesqlv8' : 'tedious';
const connectionType = platformInfo?.isWindows && (authType == 'sspi' || authType == 'sql') ? 'msnodesqlv8' : 'tedious';
const client = connectionType == 'msnodesqlv8' ? await nativeConnect(conn) : await tediousConnect(conn);
return {
@@ -172,12 +170,8 @@ const driver = {
};
driver.initialize = dbgateEnv => {
if (dbgateEnv.nativeModules && dbgateEnv.nativeModules.msnodesqlv8) {
requireMsnodesqlv8 = dbgateEnv.nativeModules.msnodesqlv8;
}
platformInfo = dbgateEnv.platformInfo;
authProxy = dbgateEnv.authProxy;
nativeDriver.initialize(dbgateEnv);
};
module.exports = driver;

10
plugins/dbgate-plugin-mssql/src/backend/nativeDriver.js
View File

@@ -1,7 +1,6 @@
const _ = require('lodash');
const stream = require('stream');
const makeUniqueColumnNames = require('./makeUniqueColumnNames');
let requireMsnodesqlv8;
const { extractDbNameFromComposite } = global.DBGATE_PACKAGES['dbgate-tools'];
// async function nativeQueryCore(pool, sql, options) {
@@ -24,7 +23,7 @@ const { extractDbNameFromComposite } = global.DBGATE_PACKAGES['dbgate-tools'];
let msnodesqlv8Value;
function getMsnodesqlv8() {
if (!msnodesqlv8Value) {
msnodesqlv8Value = requireMsnodesqlv8();
msnodesqlv8Value = require('msnodesqlv8');
}
return msnodesqlv8Value;
}
@@ -225,16 +224,9 @@ async function nativeStream(dbhan, sql, options) {
});
}
const initialize = dbgateEnv => {
if (dbgateEnv.nativeModules && dbgateEnv.nativeModules.msnodesqlv8) {
requireMsnodesqlv8 = dbgateEnv.nativeModules.msnodesqlv8;
}
};
module.exports = {
nativeConnect,
nativeQueryCore,
nativeReadQuery,
nativeStream,
initialize,
};