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

quick export - works for DuckDB

Browse Source
This commit is contained in:
SPRINX0\prochazka
2025-04-28 09:37:44 +02:00
parent edf17b8100
commit 156e1b928c
5 changed files with 75 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,6 +11,7 @@ import resolveApi, { resolveApiHeaders } from './resolveApi';
import { apiCall, apiOff, apiOn } from './api';
import { normalizeExportColumnMap } from '../impexp/createImpExpScript';
import { QuickExportDefinition } from 'dbgate-types';
import uuidv1 from 'uuid/v1';
// export async function importSqlDump(inputFile, connection) {
// const script = getCurrentConfig().allowShellScripting ? new ScriptWriterJavaScript() : new ScriptWriterJson();
@@ -52,13 +53,32 @@ import { QuickExportDefinition } from 'dbgate-types';
// });
// }
async function runImportExportScript({ script, runningMessage, canceledMessage, finishedMessage, afterFinish = null }) {
async function runImportExportScript({
script,
runningMessage,
canceledMessage,
finishedMessage,
afterFinish = null,
hostConnection = null,
}) {
const electron = getElectron();
const resp = await apiCall('runners/start', { script });
const runid = resp.runid;
let runid;
let isCanceled = false;
if (hostConnection) {
runid = uuidv1();
await apiCall('database-connections/eval-json-script', {
runid,
conid: hostConnection.conid,
database: hostConnection.database,
script,
});
} else {
const resp = await apiCall('runners/start', { script });
runid = resp.runid;
}
const snackId = showSnackbar({
message: runningMessage,
icon: 'icon loading',
@@ -96,7 +116,14 @@ async function runImportExportScript({ script, runningMessage, canceledMessage,
apiOn(`runner-progress-${runid}`, handleRunnerProgress);
}
export async function saveExportedFile(filters, defaultPath, extension, dataName, getScript: (filaPath: string) => {}) {
export async function saveExportedFile(
filters,
defaultPath,
extension,
dataName,
getScript: (filaPath: string) => {},
hostConnection = null
) {
const electron = getElectron();
let filePath;
@@ -127,6 +154,7 @@ export async function saveExportedFile(filters, defaultPath, extension, dataName
downloadFromApi(`uploads/get?file=${pureFileName}`, defaultPath);
}
},
hostConnection,
});
}
@@ -167,6 +195,7 @@ export async function exportQuickExportFile(dataName, reader, format: QuickExpor
runningMessage: `Exporting ${dataName}`,
canceledMessage: `Export ${dataName} canceled`,
finishedMessage: `Export ${dataName} finished`,
hostConnection: reader.hostConnection,
});
} else {
await saveExportedFile(
@@ -174,7 +203,8 @@ export async function exportQuickExportFile(dataName, reader, format: QuickExpor
`${dataName}.${format.extension}`,
format.extension,
dataName,
filePath => generateQuickExportScript(reader, format, filePath, dataName, columnMap)
filePath => generateQuickExportScript(reader, format, filePath, dataName, columnMap),
reader.hostConnection
);
}
}