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

feat: mongosh for script and stream methods

Browse Source
This commit is contained in:
Pavel
2025-07-24 15:52:59 +02:00
parent b8e50737d2
commit 06a845697a
1 changed files with 38 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
plugins/dbgate-plugin-mongo/src/backend/driver.js
View File

@@ -1,9 +1,8 @@
const _ = require('lodash');
const stream = require('stream');
const isPromise = require('is-promise');
const driverBase = require('../frontend/driver');
const Analyser = require('./Analyser');
const { MongoClient, ObjectId, AbstractCursor, Long } = require('mongodb');
const { MongoClient, ObjectId, Long } = require('mongodb');
const { EJSON } = require('bson');
const { NodeDriverServiceProvider } = require('@mongosh/service-provider-node-driver');
const { ElectronRuntime } = require('@mongosh/browser-runtime-electron');
@@ -30,24 +29,10 @@ function serializeMongoData(row) {
);
}
async function readCursor(cursor, options) {
options.recordset({ __isDynamicStructure: true });
await cursor.forEach((row) => {
options.row(serializeMongoData(row));
});
}
function deserializeMongoData(value) {
return deserializeJsTypesFromJsonParse(EJSON.deserialize(value));
}
function findArrayResult(resValue) {
if (!_.isPlainObject(resValue)) return null;
const arrays = _.values(resValue).filter((x) => _.isArray(x));
if (arrays.length == 1) return arrays[0];
return null;
}
async function getScriptableDb(dbhan) {
const db = dbhan.getDatabase();
const collections = await db.listCollections().toArray();
@@ -107,11 +92,20 @@ const driver = {
};
},
async script(dbhan, sql) {
let func;
func = eval(`(db,ObjectId) => ${sql}`);
const db = await getScriptableDb(dbhan);
const res = func(db, ObjectId.createFromHexString);
if (isPromise(res)) await res;
const connectionString = dbhan.client.s.url;
const serviceProvider = await NodeDriverServiceProvider.connect(connectionString);
const runtime = new ElectronRuntime(serviceProvider);
const exprValue = await runtime.evaluate(sql);
const { printable } = exprValue;
if (Array.isArray(printable)) {
return printable;
} else if ('documents' in printable) {
return printable.documents;
}
return printable;
},
async operation(dbhan, operation, options) {
const { type } = operation;
@@ -154,19 +148,35 @@ const driver = {
return;
}
console.log(exprValue);
const { printable, type } = exprValue;
if (exprValue.type === 'Document') {
if (type === 'Document') {
options.recordset({ __isDynamicStructure: true });
options.row(exprValue.printable);
} else if (exprValue.type === 'Cursor' || exprValue.type === 'AggregationCursor') {
options.row(printable);
} else if (type === 'Cursor' || exprValue.type === 'AggregationCursor') {
options.recordset({ __isDynamicStructure: true });
for (const doc of exprValue.printable.documents) {
for (const doc of printable.documents) {
options.row(doc);
}
} else {
options.recordset({ __isDynamicStructure: true });
options.row(exprValue.printable);
if (Array.isArray(printable)) {
options.recordset({ __isDynamicStructure: true });
for (const row of printable) {
options.row(row);
}
} else if ('documents' in printable) {
options.recordset({ __isDynamicStructure: true });
for (const row of printable.documents) {
options.row(row);
}
} else {
options.info({
printable: printable,
time: new Date(),
severity: 'info',
message: 'Query returned not supported value.',
});
}
}
options.done();