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

xml feed reader - auto detect item name

Browse Source
This commit is contained in:
Jan Prochazka
2022-01-30 11:41:11 +01:00
parent 3adc8119df
commit 49222bef25
2 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,10 +5,21 @@ const NodeXmlStream = require('node-xml-stream');
class ParseStream extends stream.Transform {
constructor({ itemElementName }) {
super({ objectMode: true });
let element = itemElementName;
this.push({
__isStreamHeader: true,
__isDynamicStructure: true,
});
this.rowsWritten = 0;
this.parser = new NodeXmlStream();
this.stack = [];
this.parser.on('opentag', (name, attrs) => {
if (!element && this.stack.length == 1) {
element = name;
}
this.stack.push({ name, attrs, nodes: {} });
});
this.parser.on('text', (text) => {
@@ -17,9 +28,12 @@ class ParseStream extends stream.Transform {
}
});
this.parser.on('closetag', (name, attrs) => {
if (name == itemElementName) {
this.rowsWritten += 1;
this.push({ ...this.stack[this.stack.length - 1].attrs, ...this.stack[this.stack.length - 1].nodes });
if (name == element) {
const obj = { ...this.stack[this.stack.length - 1].attrs, ...this.stack[this.stack.length - 1].nodes };
if (Object.keys(obj).length > 0) {
this.rowsWritten += 1;
this.push(obj);
}
}
this.stack.splice(-1);
});