From 81b1cda8c5d97a13ea3e2b63e7ae84fbd7dd6609 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 28 Oct 2021 10:35:15 +0200
Subject: [PATCH] split query fix

---
 packages/query-splitter/src/splitQuery.ts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/query-splitter/src/splitQuery.ts b/packages/query-splitter/src/splitQuery.ts
index 226013a3b..1adb8b32f 100644
--- a/packages/query-splitter/src/splitQuery.ts
+++ b/packages/query-splitter/src/splitQuery.ts
@@ -2,14 +2,14 @@ import { SplitterOptions, defaultSplitterOptions } from './options';
 
 const SEMICOLON = ';';
 
-interface SplitStreamContext {
+export interface SplitStreamContext {
   options: SplitterOptions;
   currentDelimiter: string;
   pushOutput: (sql: string) => void;
   commandPart: string;
 }
 
-interface SplitLineContext extends SplitStreamContext {
+export interface SplitLineContext extends SplitStreamContext {
   source: string;
   position: number;
   // output: string[];
@@ -246,7 +246,8 @@ export function splitQuery(sql: string, options: SplitterOptions = null): string
 
   splitQueryLine(context);
 
-  pushQuery(context);
-  
+  const trimmed = context.commandPart.trim();
+  if (trimmed) context.pushOutput(trimmed);
+
   return output;
 }