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

basic XML import works + small fixes

Browse Source
This commit is contained in:
Jan Prochazka
2022-01-30 11:26:10 +01:00
parent 91a77765b6
commit 8bad6da348
7 changed files with 75 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
plugins/dbgate-plugin-xml/src/backend/reader.js
View File

@@ -3,7 +3,7 @@ const stream = require('stream');
const NodeXmlStream = require('node-xml-stream');
class ParseStream extends stream.Transform {
constructor({ elementName }) {
constructor({ itemElementName }) {
super({ objectMode: true });
this.rowsWritten = 0;
this.parser = new NodeXmlStream();
@@ -17,7 +17,7 @@ class ParseStream extends stream.Transform {
}
});
this.parser.on('closetag', (name, attrs) => {
if (name == elementName) {
if (name == itemElementName) {
this.rowsWritten += 1;
this.push({ ...this.stack[this.stack.length - 1].attrs, ...this.stack[this.stack.length - 1].nodes });
}
@@ -30,11 +30,11 @@ class ParseStream extends stream.Transform {
}
}
async function reader({ fileName, encoding = 'utf-8', elementName }) {
async function reader({ fileName, encoding = 'utf-8', itemElementName }) {
console.log(`Reading file ${fileName}`);
const fileStream = fs.createReadStream(fileName, encoding);
const parser = new ParseStream({ elementName });
const parser = new ParseStream({ itemElementName });
fileStream.pipe(parser);
return parser;
}