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

generate script fixes

Browse Source
This commit is contained in:
Jan Prochazka
2020-06-06 22:17:17 +02:00
parent 74aa90fd2a
commit 617548cd50
6 changed files with 40 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/web/src/impexp/ScriptWriter.js
View File

@@ -2,25 +2,26 @@ export default class ScriptWriter {
constructor() {
this.s = '';
this.put('const dbgateApi = require("@dbgate/api");');
this.put();
this.put('async function run() {');
}
put(s) {
put(s = '') {
this.s += s;
this.s += '\n';
}
finish() {
this.put('await dbgateApi.copyStream(queryReader, csvWriter);');
this.put('dbgateApi.runScript(run);');
this.put('}');
this.put();
this.put('dbgateApi.runScript(run);');
}
assign(variableName, functionName, props) {
this.put(`const ${variableName} = await dbgateApi.${functionName}(${JSON.stringify(props)});`);
this.put(` const ${variableName} = await dbgateApi.${functionName}(${JSON.stringify(props)});`);
}
copyStream(sourceVar, targetVar) {
this.put(`await dbgateApi.copyStream(${sourceVar}, ${targetVar});`);
this.put(` await dbgateApi.copyStream(${sourceVar}, ${targetVar});`);
}
}