#187 split query stream

packages/query-splitter/src/splitQueryStream.ts

@@ -0,0 +1,41 @@
+import stream from 'stream';
+import { SplitStreamContext, getInitialDelimiter, SplitLineContext, splitQueryLine } from './splitQuery';
+import { SplitterOptions } from './options';
+
+export class SplitQueryStream extends stream.Transform {
+  context: SplitStreamContext;
+
+  constructor(options: SplitterOptions) {
+    super({ objectMode: true });
+    this.context = {
+      commandPart: '',
+      options,
+      currentDelimiter: getInitialDelimiter(options),
+      pushOutput: cmd => this.push(cmd),
+    };
+  }
+  _transform(chunk, encoding, done) {
+    const lineContext: SplitLineContext = {
+      ...this.context,
+      position: 0,
+      currentCommandStart: 0,
+      wasDataOnLine: false,
+      source: chunk,
+      end: chunk.length,
+    };
+    splitQueryLine(lineContext);
+    this.context.commandPart = lineContext.commandPart;
+    done();
+  }
+  _flush(done) {
+    const trimmed = this.context.commandPart;
+    if (trimmed) this.push(trimmed);
+    done();
+  }
+}
+
+export function splitQueryStream(sourceStream, options: SplitterOptions) {
+  const splitter = new SplitQueryStream(options);
+  sourceStream.pipe(splitter);
+  return splitter;
+}