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: use getColleciton syntax for mongo

Browse Source
This commit is contained in:
Pavel
2025-07-29 16:33:39 +02:00
parent 461f1e39fa
commit 6e8cdc24a3
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -138,15 +138,18 @@ const driver = {
await this.script(dbhan, `db.createCollection('${operation.collection.name}')`);
break;
case 'dropCollection':
await this.script(dbhan, `db.dropCollection('${operation.collection}')`);
await this.script(dbhan, `db.getCollection('${operation.collection}').drop()`);
break;
case 'renameCollection':
await this.script(dbhan, `db.renameCollection('${operation.collection}', '${operation.newName}')`);
await this.script(
dbhan,
`db.getCollection('${operation.collection}').renameCollection('${operation.newName}')`
);
break;
case 'cloneCollection':
await this.script(
dbhan,
`db.collection('${operation.collection}').aggregate([{$out: '${operation.newName}'}]).toArray()`
`db.getCollection('${operation.collection}').aggregate([{$out: '${operation.newName}'}]).toArray()`
);
break;
default:

14
plugins/dbgate-plugin-mongo/src/frontend/driver.js
View File

@@ -88,14 +88,14 @@ const driver = {
getCollectionUpdateScript(changeSet, collectionInfo) {
let res = '';
for (const insert of changeSet.inserts) {
res += `db['${insert.pureName}'].insertOne(${jsonStringifyWithObjectId({
res += `db.getCollection('${insert.pureName}').insertOne(${jsonStringifyWithObjectId({
...insert.document,
...insert.fields,
})});\n`;
}
for (const update of changeSet.updates) {
if (update.document) {
res += `db['${update.pureName}'].replaceOne(${jsonStringifyWithObjectId(
res += `db.getCollection('${update.pureName}').replaceOne(${jsonStringifyWithObjectId(
update.condition
)}, ${jsonStringifyWithObjectId({
...update.document,
@@ -112,13 +112,13 @@ const driver = {
if (!_.isEmpty(set)) updates.$set = set;
if (!_.isEmpty(unset)) updates.$unset = unset;
res += `db['${update.pureName}'].updateOne(${jsonStringifyWithObjectId(
res += `db.getCollection('${update.pureName}').updateOne(${jsonStringifyWithObjectId(
update.condition
)}, ${jsonStringifyWithObjectId(updates)});\n`;
}
}
for (const del of changeSet.deletes) {
res += `db['${del.pureName}'].deleteOne(${jsonStringifyWithObjectId(del.condition)});\n`;
res += `db.getCollection('${del.pureName}').deleteOne(${jsonStringifyWithObjectId(del.condition)});\n`;
}
return res;
},
@@ -128,7 +128,7 @@ const driver = {
},
getCollectionExportQueryScript(collection, condition, sort) {
return `db['${collection}']
return `db.getCollection('${collection}')
.find(${JSON.stringify(convertToMongoCondition(condition) || {})})
.sort(${JSON.stringify(convertToMongoSort(sort) || {})})`;
},
@@ -182,9 +182,9 @@ const driver = {
async getScriptTemplateContent(scriptTemplate, props) {
switch (scriptTemplate) {
case 'dropCollection':
return `db['${props.pureName}'].drop();`;
return `db.getCollection('${props.pureName}').drop();`;
case 'findCollection':
return `db['${props.pureName}'].find();`;
return `db.getCollection('${props.pureName}').find();`;
}
},
};