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

import SQL dump POC

Browse Source
This commit is contained in:
Jan Prochazka
2022-04-18 14:22:16 +02:00
parent c7c667bbe0
commit a75e931fd5
6 changed files with 207 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
packages/web/src/utility/exportFileTools.ts
View File

@@ -6,6 +6,57 @@ import { apiCall, apiOff, apiOn } from './api';
import { normalizeExportColumnMap } from '../impexp/createImpExpScript';
import { getCurrentConfig } from '../stores';
export async function importSqlDump(inputFile, connection) {
const script = getCurrentConfig().allowShellScripting ? new ScriptWriter() : new ScriptWriterJson();
script.importDatabase({
inputFile,
connection,
});
await runImportExportScript({
script: script.getScript(),
runningMessage: 'Importing database',
canceledMessage: 'Database import canceled',
finishedMessage: 'Database import finished',
});
}
async function runImportExportScript({ script, runningMessage, canceledMessage, finishedMessage, afterFinish = null }) {
const electron = getElectron();
const resp = await apiCall('runners/start', { script });
const runid = resp.runid;
let isCanceled = false;
const snackId = showSnackbar({
message: runningMessage,
icon: 'icon loading',
buttons: [
{
label: 'Cancel',
onClick: () => {
isCanceled = true;
apiCall('runners/cancel', { runid });
},
},
],
});
function handleRunnerDone() {
closeSnackbar(snackId);
apiOff(`runner-done-${runid}`, handleRunnerDone);
if (isCanceled) {
showSnackbarError(canceledMessage);
} else {
showSnackbarInfo(finishedMessage);
if (afterFinish) afterFinish();
}
}
apiOn(`runner-done-${runid}`, handleRunnerDone);
}
export async function saveExportedFile(filters, defaultPath, extension, dataName, getScript: (filaPath: string) => {}) {
const electron = getElectron();
@@ -27,36 +78,17 @@ export async function saveExportedFile(filters, defaultPath, extension, dataName
const script = getScript(filePath);
const resp = await apiCall('runners/start', { script });
const runid = resp.runid;
let isCanceled = false;
const snackId = showSnackbar({
message: `Exporting ${dataName}`,
icon: 'icon loading',
buttons: [
{
label: 'Cancel',
onClick: () => {
isCanceled = true;
apiCall('runners/cancel', { runid });
},
},
],
runImportExportScript({
script,
runningMessage: `Exporting ${dataName}`,
canceledMessage: `Export ${dataName} canceled`,
finishedMessage: `Export ${dataName} finished`,
afterFinish: () => {
if (!electron) {
window.open(`${resolveApi()}/uploads/get?file=${pureFileName}`, '_blank');
}
},
});
function handleRunnerDone() {
closeSnackbar(snackId);
apiOff(`runner-done-${runid}`, handleRunnerDone);
if (isCanceled) showSnackbarError(`Export ${dataName} canceled`);
else showSnackbarInfo(`Export ${dataName} finished`);
if (!electron) {
window.open(`${resolveApi()}/uploads/get?file=${pureFileName}`, '_blank');
}
}
apiOn(`runner-done-${runid}`, handleRunnerDone);
}
export async function exportQuickExportFile(dataName, reader, format, columnMap = null) {