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

fix: use correct dbName in connection string

Browse Source
This commit is contained in:
Pavel
2025-07-29 16:12:56 +02:00
parent 4892dbce5e
commit 461f1e39fa
1 changed files with 24 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -42,6 +42,28 @@ async function getScriptableDb(dbhan) {
return db;
}
/**
* @param {string} uri
* @param {string} dbName
* @returns {string}
*/
function ensureDatabaseInMongoURI(uri, dbName) {
if (!dbName) return uri;
try {
const url = new URL(uri);
const hasDatabase = url.pathname && url.pathname !== '/' && url.pathname.length > 1;
if (hasDatabase) return uri;
url.pathname = `/${dbName}`;
return url.toString();
} catch (error) {
logger.error('Invalid URI format:', error.message);
return uri;
}
}
/** @type {import('dbgate-types').EngineDriver<MongoClient>} */
const driver = {
...driverBase,
@@ -92,7 +114,7 @@ const driver = {
};
},
async script(dbhan, sql) {
const connectionString = dbhan.client.s.url;
const connectionString = ensureDatabaseInMongoURI(dbhan.client.s.url, dbhan.database);
const serviceProvider = await NodeDriverServiceProvider.connect(connectionString);
const runtime = new ElectronRuntime(serviceProvider);
const exprValue = await runtime.evaluate(sql);
@@ -136,7 +158,7 @@ const driver = {
let exprValue;
try {
const connectionString = dbhan.client.s.url;
const connectionString = ensureDatabaseInMongoURI(dbhan.client.s.url, dbhan.database);
const serviceProvider = await NodeDriverServiceProvider.connect(connectionString);
const runtime = new ElectronRuntime(serviceProvider);
exprValue = await runtime.evaluate(sql);