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

readonly sessions on mongo (+checks on BE)

Browse Source
This commit is contained in:
Jan Prochazka
2022-03-17 13:17:30 +01:00
parent 7d196c7c62
commit 8f1d76fd2a
8 changed files with 51 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/api/src/proc/databaseConnectionProcess.js
View File

@@ -155,6 +155,7 @@ async function handleRunScript({ msgid, sql }) {
await waitConnected();
const driver = requireEngineDriver(storedConnection);
try {
ensureExecuteCustomScript(driver);
await driver.script(systemConnection, sql);
process.send({ msgtype: 'response', msgid });
} catch (err) {
@@ -166,6 +167,7 @@ async function handleQueryData({ msgid, sql }) {
await waitConnected();
const driver = requireEngineDriver(storedConnection);
try {
ensureExecuteCustomScript(driver);
const res = await driver.query(systemConnection, sql);
process.send({ msgtype: 'response', msgid, ...res });
} catch (err) {
@@ -204,7 +206,10 @@ async function handleLoadKeyInfo({ msgid, key }) {
}
async function handleCallMethod({ msgid, method, args }) {
return handleDriverDataCore(msgid, driver => driver.callMethod(systemConnection, method, args));
return handleDriverDataCore(msgid, driver => {
ensureExecuteCustomScript(driver);
return driver.callMethod(systemConnection, method, args);
});
}
async function handleLoadKeyTableRange({ msgid, key, cursor, count }) {
@@ -217,10 +222,20 @@ async function handleLoadFieldValues({ msgid, schemaName, pureName, field, searc
);
}
function ensureExecuteCustomScript(driver) {
if (driver.readOnlySessions) {
return;
}
if (storedConnection.isReadOnly) {
throw new Error('Connection is read only');
}
}
async function handleUpdateCollection({ msgid, changeSet }) {
await waitConnected();
const driver = requireEngineDriver(storedConnection);
try {
ensureExecuteCustomScript(driver);
const result = await driver.updateCollection(systemConnection, changeSet);
process.send({ msgtype: 'response', msgid, result });
} catch (err) {