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

#166 filter by mongo ObjectID

Browse Source
This commit is contained in:
Jan Prochazka
2021-09-16 12:43:51 +02:00
parent 90262a0318
commit 81fbed7a7f
3 changed files with 36 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,7 +18,7 @@ function readCursor(cursor, options) {
}
const mongoIdRegex = /^[0-9a-f]{24}$/;
function convertCondition(condition) {
function convertConditionInternal(condition) {
if (condition && _.isString(condition._id) && condition._id.match(mongoIdRegex)) {
return {
_id: ObjectId(condition._id),
@@ -27,6 +27,12 @@ function convertCondition(condition) {
return condition;
}
function convertConditionUser(condition) {
return _.cloneDeepWith(condition, (x) => {
if (x && x.$oid) return ObjectId(x.$oid);
});
}
function findArrayResult(resValue) {
if (!_.isPlainObject(resValue)) return null;
const arrays = _.values(resValue).filter((x) => _.isArray(x));
@@ -191,10 +197,11 @@ const driver = {
try {
const collection = pool.__getDatabase().collection(options.pureName);
if (options.countDocuments) {
const count = await collection.countDocuments(options.condition || {});
const count = await collection.countDocuments(convertConditionUser(options.condition) || {});
return { count };
} else {
let cursor = await collection.find(options.condition || {});
// console.log('options.condition', JSON.stringify(options.condition, undefined, 2));
let cursor = await collection.find(convertConditionUser(options.condition) || {});
if (options.sort) cursor = cursor.sort(options.sort);
if (options.skip) cursor = cursor.skip(options.skip);
if (options.limit) cursor = cursor.limit(options.limit);
@@ -230,22 +237,24 @@ const driver = {
...update.document,
...update.fields,
};
const doc = await collection.findOne(convertCondition(update.condition));
const doc = await collection.findOne(convertConditionInternal(update.condition));
if (doc) {
const resdoc = await collection.replaceOne(convertCondition(update.condition), {
const resdoc = await collection.replaceOne(convertConditionInternal(update.condition), {
...document,
_id: doc._id,
});
res.replaced.push(resdoc._id);
}
} else {
const resdoc = await collection.updateOne(convertCondition(update.condition), { $set: update.fields });
const resdoc = await collection.updateOne(convertConditionInternal(update.condition), {
$set: update.fields,
});
res.updated.push(resdoc._id);
}
}
for (const del of changeSet.deletes) {
const collection = db.collection(del.pureName);
const resdoc = await collection.deleteOne(convertCondition(del.condition));
const resdoc = await collection.deleteOne(convertConditionInternal(del.condition));
res.deleted.push(resdoc._id);
}
return res;