mirrors/dbgate
1
0
Fork 0
mirror of https://github.com/dbgate/dbgate synced 2024-11-07 20:26:23 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

suport mongo ObjectId in data editor

Browse Source
This commit is contained in:
Jan Prochazka 2021-12-08 21:19:46 +01:00
parent 3108fb60f9
commit fca00ed248
4 changed files with 41 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/web/src/datagrid/InplaceEditor.svelte
View File

@ -1,6 +1,7 @@
<script lang="ts" context="module">
function getEditedValue(value) {
if (value?.type == 'Buffer' && _.isArray(value.data)) return arrayToHexString(value.data);
if (value?.$oid) return `ObjectId("${value?.$oid}")`;
if (_.isPlainObject(value) || _.isArray(value)) return JSON.stringify(value);
return value;
}
@ -12,6 +13,13 @@
data: hexStringToArray(newString),
};
}
if (_.isString(newString)) {
const m = newString.match(/ObjectId\("([0-9a-f]{24})"\)/);
if (m) {
return { $oid: m[1] };
}
}
return newString;
}
</script>
@ -28,7 +36,7 @@
export let onSetValue;
export let width;
export let cellValue;
export let fillParent=false;
export let fillParent = false;
let domEditor;

2
packages/web/src/jsonview/CollectionJsonRow.svelte
View File

@ -4,7 +4,7 @@
showModal(EditJsonModal, {
json: rowData,
onSave: value => {
if (rowData._id && value._id != rowData._id) {
if (grider.getRowStatus(rowIndex).status != 'inserted' && rowData._id && value._id != rowData._id) {
showModal(ErrorMessageModal, { message: '_id attribute cannot be changed' });
return false;
}

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

@ -21,7 +21,7 @@ function readCursor(cursor, options) {
});
}
function convertCondition(condition) {
function convertObjectId(condition) {
return _.cloneDeepWith(condition, (x) => {
if (x && x.$oid) return ObjectId(x.$oid);
});
@ -205,11 +205,11 @@ const driver = {
try {
const collection = pool.__getDatabase().collection(options.pureName);
if (options.countDocuments) {
const count = await collection.countDocuments(convertCondition(options.condition) || {});
const count = await collection.countDocuments(convertObjectId(options.condition) || {});
return { count };
} else {
// console.log('options.condition', JSON.stringify(options.condition, undefined, 2));
let cursor = await collection.find(convertCondition(options.condition) || {});
let cursor = await collection.find(convertObjectId(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);
@ -235,7 +235,7 @@ const driver = {
...insert.document,
...insert.fields,
};
const resdoc = await collection.insert(document);
const resdoc = await collection.insert(convertObjectId(document));
res.inserted.push(resdoc._id);
}
for (const update of changeSet.updates) {
@ -245,24 +245,24 @@ const driver = {
...update.document,
...update.fields,
};
const doc = await collection.findOne(convertCondition(update.condition));
const doc = await collection.findOne(convertObjectId(update.condition));
if (doc) {
const resdoc = await collection.replaceOne(convertCondition(update.condition), {
...document,
const resdoc = await collection.replaceOne(convertObjectId(update.condition), {
...convertObjectId(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(convertObjectId(update.condition), {
$set: convertObjectId(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(convertObjectId(del.condition));
res.deleted.push(resdoc._id);
}
return res;

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

@ -3,11 +3,11 @@ const { driverBase } = global.DBGATE_TOOLS;
const Dumper = require('./Dumper');
const { mongoSplitterOptions } = require('dbgate-query-splitter/lib/options');
function getConditionPreview(condition) {
if (condition && condition._id && condition._id.$oid) {
return `{ _id: ObjectId('${condition._id.$oid}') }`;
}
return JSON.stringify(condition);
function jsonStringifyWithObjectId(obj) {
return JSON.stringify(obj, undefined, 2).replace(
/\{\s*\"\$oid\"\s*\:\s*\"([0-9a-f]+)\"\s*\}/g,
(m, id) => `ObjectId("${id}")`
);
}
/** @type {import('dbgate-types').SqlDialect} */
@ -47,37 +47,29 @@ const driver = {
getCollectionUpdateScript(changeSet) {
let res = '';
for (const insert of changeSet.inserts) {
res += `db.${insert.pureName}.insert(${JSON.stringify(
{
...insert.document,
...insert.fields,
},
undefined,
2
)});\n`;
res += `db.${insert.pureName}.insert(${jsonStringifyWithObjectId({
...insert.document,
...insert.fields,
})});\n`;
}
for (const update of changeSet.updates) {
if (update.document) {
res += `db.${update.pureName}.replaceOne(${getConditionPreview(update.condition)}, ${JSON.stringify(
{
...update.document,
...update.fields,
},
undefined,
2
)});\n`;
res += `db.${update.pureName}.replaceOne(${jsonStringifyWithObjectId(
update.condition
)}, ${jsonStringifyWithObjectId({
...update.document,
...update.fields,
})});\n`;
} else {
res += `db.${update.pureName}.updateOne(${getConditionPreview(update.condition)}, ${JSON.stringify(
{
$set: update.fields,
},
undefined,
2
)});\n`;
res += `db.${update.pureName}.updateOne(${jsonStringifyWithObjectId(
update.condition
)}, ${jsonStringifyWithObjectId({
$set: update.fields,
})});\n`;
}
}
for (const del of changeSet.deletes) {
res += `db.${del.pureName}.deleteOne(${getConditionPreview(del.condition)});\n`;
res += `db.${del.pureName}.deleteOne(${jsonStringifyWithObjectId(del.condition)});\n`;
}
return res;
},