Merge branch 'master' of https://github.com/dbgate/dbgate-query-splitter

Author: SPRINX0\prochazka

package.json

@@ -1,5 +1,5 @@
 {
-  "version": "4.10.4",
+  "version": "4.10.5",
   "name": "dbgate-query-splitter",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",

sql/pgdump.sql

@@ -0,0 +1,13 @@
+ALTER TABLE public.testxxx OWNER TO postgres;
+
+--
+-- Data for Name: Album; Type: TABLE DATA; Schema: public; Owner: postgres
+--
+
+COPY public."Album" ("AlbumId", "Title", "ArtistId") FROM stdin;
+3	Restless and Wild	2
+4	Let There Be Rock	1
+5	Big Ones	3
+1	For Those About To Rock We Salute You	2
+\.
+

src/splitQueryStream.ts

@@ -68,6 +68,8 @@ export class SplitQueryStream extends stream.Transform {
       trimCommandStartColumn: this.context.trimCommandStartColumn,
 
       wasDataInCommand: this.context.wasDataInCommand,
+      isCopyFromStdin: this.context.isCopyFromStdin,
+      isCopyFromStdinCandidate: this.context.isCopyFromStdinCandidate,
     };
 
     splitQueryLine(lineContext);
@@ -92,6 +94,8 @@ export class SplitQueryStream extends stream.Transform {
     this.context.trimCommandStartColumn = lineContext.trimCommandStartColumn;
 
     this.context.wasDataInCommand = lineContext.wasDataInCommand;
+    this.context.isCopyFromStdin = lineContext.isCopyFromStdin;
+    this.context.isCopyFromStdinCandidate = lineContext.isCopyFromStdinCandidate;
 
     this.lineBuffer = '';
   }

src/splitterStream.test.ts

@@ -73,3 +73,16 @@ test('northwind test', async () => {
   const output = await streamToArray(splitQueryStream(fileStream, mysqlSplitterOptions));
   expect(output['length']).toEqual(29);
 });
+
+test('pgdump test', async () => {
+  console.log(process.cwd());
+  const fileStream = fs.createReadStream('sql/pgdump.sql', 'utf-8');
+  const output = await streamToArray(
+    splitQueryStream(fileStream, {
+      ...postgreSplitterOptions,
+      copyFromStdin: true,
+      returnRichInfo: true,
+    })
+  );
+  expect(output['length']).toEqual(7);
+});