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

import SQL dump fixed

Browse Source
This commit is contained in:
Jan Prochazka
2022-04-21 17:30:16 +02:00
parent 82304f13e7
commit 5315a549b0
11 changed files with 27 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/api/src/shell/importDatabase.js
View File

@@ -1,5 +1,4 @@
const fs = require('fs');
const byline = require('byline');
const requireEngineDriver = require('../utility/requireEngineDriver');
const connectUtility = require('../utility/connectUtility');
const { splitQueryStream } = require('dbgate-query-splitter/lib/splitQueryStream');
@@ -20,6 +19,11 @@ class ImportStream extends stream.Transform {
}
cb();
}
_flush(cb) {
this.push('finish');
cb();
this.emit('end');
}
}
function awaitStreamEnd(stream) {
@@ -43,12 +47,11 @@ async function importDatabase({ connection = undefined, systemConnection = undef
const downloadedFile = await download(inputFile);
const fileStream = fs.createReadStream(downloadedFile, 'utf-8');
const lineStream = byline(fileStream);
const splittedStream = splitQueryStream(lineStream, driver.getQuerySplitterOptions());
const splittedStream = splitQueryStream(fileStream, driver.getQuerySplitterOptions());
const importStream = new ImportStream(pool, driver);
// @ts-ignore
splittedStream.pipe(importStream);
await awaitStreamEnd(splittedStream);
await awaitStreamEnd(importStream);
}
module.exports = importDatabase;